store

package
v0.1.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Nov 14, 2024 License: Apache-2.0 Imports: 5 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type ExperimentTrackingStore 

type ExperimentTrackingStore interface {
	// GetExperiment returns experiment by the experiment ID.
	// The experiment should contain the linked tags.
	GetExperiment(ctx context.Context, id string) (*entities.Experiment, *contract.Error)
	GetExperimentByName(ctx context.Context, name string) (*entities.Experiment, *contract.Error)
	SearchExperiments(
		ctx context.Context,
		experimentViewType protos.ViewType,
		maxResults int64,
		filter string,
		orderBy []string,
		pageToken string,
	) ([]*entities.Experiment, string, *contract.Error)

	CreateExperiment(
		ctx context.Context,
		name string,
		artifactLocation string,
		tags []*entities.ExperimentTag,
	) (string, *contract.Error)
	RestoreExperiment(ctx context.Context, id string) *contract.Error
	RenameExperiment(ctx context.Context, experimentID, name string) *contract.Error

	SearchRuns(
		ctx context.Context,
		experimentIDs []string,
		filter string,
		runViewType protos.ViewType,
		maxResults int,
		orderBy []string,
		pageToken string,
	) ([]*entities.Run, string, *contract.Error)

	DeleteExperiment(ctx context.Context, id string) *contract.Error
	SetExperimentTag(ctx context.Context, experimentID, key, value string) *contract.Error
}

type InputTrackingStore 

type InputTrackingStore interface {
	LogInputs(ctx context.Context, runID string, datasets []*entities.DatasetInput) *contract.Error
}

type MetricTrackingStore 

type MetricTrackingStore interface {
	LogBatch(
		ctx context.Context,
		runID string,
		metrics []*entities.Metric,
		params []*entities.Param,
		tags []*entities.RunTag) *contract.Error

	LogMetric(ctx context.Context, runID string, metric *entities.Metric) *contract.Error
	LogParam(ctx context.Context, runID string, metric *entities.Param) *contract.Error
	GetMetricHistory(ctx context.Context, runID, metricKey string) ([]*entities.Metric, *contract.Error)
}

type MockTrackingStore 

type MockTrackingStore struct {
	mock.Mock
}

MockTrackingStore is an autogenerated mock type for the TrackingStore type

func NewMockTrackingStore 

func NewMockTrackingStore(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockTrackingStore

NewMockTrackingStore creates a new instance of MockTrackingStore. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. The first argument is typically a *testing.T value.

func (*MockTrackingStore) CreateExperiment 

func (_m *MockTrackingStore) CreateExperiment(ctx context.Context, name string, artifactLocation string, tags []*entities.ExperimentTag) (string, *contract.Error)

CreateExperiment provides a mock function with given fields: ctx, name, artifactLocation, tags

func (*MockTrackingStore) CreateRun 

func (_m *MockTrackingStore) CreateRun(ctx context.Context, experimentID string, userID string, startTime int64, tags []*entities.RunTag, runName string) (*entities.Run, *contract.Error)

CreateRun provides a mock function with given fields: ctx, experimentID, userID, startTime, tags, runName

func (*MockTrackingStore) DeleteExperiment 

func (_m *MockTrackingStore) DeleteExperiment(ctx context.Context, id string) *contract.Error

DeleteExperiment provides a mock function with given fields: ctx, id

func (*MockTrackingStore) DeleteRun 

func (_m *MockTrackingStore) DeleteRun(ctx context.Context, runID string) *contract.Error

DeleteRun provides a mock function with given fields: ctx, runID

func (*MockTrackingStore) DeleteTag 

func (_m *MockTrackingStore) DeleteTag(ctx context.Context, runID string, key string) *contract.Error

DeleteTag provides a mock function with given fields: ctx, runID, key

func (*MockTrackingStore) DeleteTraceTag 

func (_m *MockTrackingStore) DeleteTraceTag(ctx context.Context, tag *entities.TraceTag) *contract.Error

DeleteTraceTag provides a mock function with given fields: ctx, tag

func (*MockTrackingStore) DeleteTraces 

func (_m *MockTrackingStore) DeleteTraces(ctx context.Context, experimentID string, maxTimestampMillis int64, maxTraces int32, requestIDs []string) (int32, *contract.Error)

DeleteTraces provides a mock function with given fields: ctx, experimentID, maxTimestampMillis, maxTraces, requestIDs

func (*MockTrackingStore) Destroy 

func (_m *MockTrackingStore) Destroy() error

Destroy provides a mock function with given fields:

func (*MockTrackingStore) EXPECT 

func (_m *MockTrackingStore) EXPECT() *MockTrackingStore_Expecter

func (*MockTrackingStore) EndTrace 

func (_m *MockTrackingStore) EndTrace(ctx context.Context, reqeustID string, timestampMS int64, status string, metadata []*entities.TraceRequestMetadata, tags []*entities.TraceTag) (*entities.TraceInfo, error)

EndTrace provides a mock function with given fields: ctx, reqeustID, timestampMS, status, metadata, tags

func (*MockTrackingStore) GetExperiment 

func (_m *MockTrackingStore) GetExperiment(ctx context.Context, id string) (*entities.Experiment, *contract.Error)

GetExperiment provides a mock function with given fields: ctx, id

func (*MockTrackingStore) GetExperimentByName 

func (_m *MockTrackingStore) GetExperimentByName(ctx context.Context, name string) (*entities.Experiment, *contract.Error)

GetExperimentByName provides a mock function with given fields: ctx, name

func (*MockTrackingStore) GetMetricHistory 

func (_m *MockTrackingStore) GetMetricHistory(ctx context.Context, runID string, metricKey string) ([]*entities.Metric, *contract.Error)

GetMetricHistory provides a mock function with given fields: ctx, runID, metricKey

func (*MockTrackingStore) GetRun 

func (_m *MockTrackingStore) GetRun(ctx context.Context, runID string) (*entities.Run, *contract.Error)

GetRun provides a mock function with given fields: ctx, runID

func (*MockTrackingStore) GetRunTag 

func (_m *MockTrackingStore) GetRunTag(ctx context.Context, runID string, tagKey string) (*entities.RunTag, *contract.Error)

GetRunTag provides a mock function with given fields: ctx, runID, tagKey

func (*MockTrackingStore) GetTraceInfo 

func (_m *MockTrackingStore) GetTraceInfo(ctx context.Context, reqeustID string) (*entities.TraceInfo, *contract.Error)

GetTraceInfo provides a mock function with given fields: ctx, reqeustID

func (*MockTrackingStore) GetTraceTag 

func (_m *MockTrackingStore) GetTraceTag(ctx context.Context, requestID string, key string) (*entities.TraceTag, *contract.Error)

GetTraceTag provides a mock function with given fields: ctx, requestID, key

func (*MockTrackingStore) LogBatch 

func (_m *MockTrackingStore) LogBatch(ctx context.Context, runID string, metrics []*entities.Metric, params []*entities.Param, tags []*entities.RunTag) *contract.Error

LogBatch provides a mock function with given fields: ctx, runID, metrics, params, tags

func (*MockTrackingStore) LogInputs 

func (_m *MockTrackingStore) LogInputs(ctx context.Context, runID string, datasets []*entities.DatasetInput) *contract.Error

LogInputs provides a mock function with given fields: ctx, runID, datasets

func (*MockTrackingStore) LogMetric 

func (_m *MockTrackingStore) LogMetric(ctx context.Context, runID string, metric *entities.Metric) *contract.Error

LogMetric provides a mock function with given fields: ctx, runID, metric

func (*MockTrackingStore) LogParam 

func (_m *MockTrackingStore) LogParam(ctx context.Context, runID string, metric *entities.Param) *contract.Error

LogParam provides a mock function with given fields: ctx, runID, metric

func (*MockTrackingStore) RenameExperiment 

func (_m *MockTrackingStore) RenameExperiment(ctx context.Context, experimentID string, name string) *contract.Error

RenameExperiment provides a mock function with given fields: ctx, experimentID, name

func (*MockTrackingStore) RestoreExperiment 

func (_m *MockTrackingStore) RestoreExperiment(ctx context.Context, id string) *contract.Error

RestoreExperiment provides a mock function with given fields: ctx, id

func (*MockTrackingStore) RestoreRun 

func (_m *MockTrackingStore) RestoreRun(ctx context.Context, runID string) *contract.Error

RestoreRun provides a mock function with given fields: ctx, runID

func (*MockTrackingStore) SearchExperiments 

func (_m *MockTrackingStore) SearchExperiments(ctx context.Context, experimentViewType protos.ViewType, maxResults int64, filter string, orderBy []string, pageToken string) ([]*entities.Experiment, string, *contract.Error)

SearchExperiments provides a mock function with given fields: ctx, experimentViewType, maxResults, filter, orderBy, pageToken

func (*MockTrackingStore) SearchRuns 

func (_m *MockTrackingStore) SearchRuns(ctx context.Context, experimentIDs []string, filter string, runViewType protos.ViewType, maxResults int, orderBy []string, pageToken string) ([]*entities.Run, string, *contract.Error)

SearchRuns provides a mock function with given fields: ctx, experimentIDs, filter, runViewType, maxResults, orderBy, pageToken

func (*MockTrackingStore) SetExperimentTag 

func (_m *MockTrackingStore) SetExperimentTag(ctx context.Context, experimentID string, key string, value string) *contract.Error

SetExperimentTag provides a mock function with given fields: ctx, experimentID, key, value

func (*MockTrackingStore) SetTag 

func (_m *MockTrackingStore) SetTag(ctx context.Context, runID string, key string, value string) *contract.Error

SetTag provides a mock function with given fields: ctx, runID, key, value

func (*MockTrackingStore) SetTrace 

func (_m *MockTrackingStore) SetTrace(ctx context.Context, experimentID string, timestampMS int64, metadata []*entities.TraceRequestMetadata, tags []*entities.TraceTag) (*entities.TraceInfo, error)

SetTrace provides a mock function with given fields: ctx, experimentID, timestampMS, metadata, tags

func (*MockTrackingStore) SetTraceTag 

func (_m *MockTrackingStore) SetTraceTag(ctx context.Context, requestID string, key string, value string) error

SetTraceTag provides a mock function with given fields: ctx, requestID, key, value

func (*MockTrackingStore) UpdateRun 

func (_m *MockTrackingStore) UpdateRun(ctx context.Context, runID string, runStatus string, endTime *int64, runName string) *contract.Error

UpdateRun provides a mock function with given fields: ctx, runID, runStatus, endTime, runName

type MockTrackingStore_CreateExperiment_Call 

type MockTrackingStore_CreateExperiment_Call struct {
	*mock.Call
}

MockTrackingStore_CreateExperiment_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateExperiment'

func (*MockTrackingStore_CreateExperiment_Call) Return 

func (_c *MockTrackingStore_CreateExperiment_Call) Return(_a0 string, _a1 *contract.Error) *MockTrackingStore_CreateExperiment_Call

func (*MockTrackingStore_CreateExperiment_Call) Run 

func (_c *MockTrackingStore_CreateExperiment_Call) Run(run func(ctx context.Context, name string, artifactLocation string, tags []*entities.ExperimentTag)) *MockTrackingStore_CreateExperiment_Call

func (*MockTrackingStore_CreateExperiment_Call) RunAndReturn 

func (_c *MockTrackingStore_CreateExperiment_Call) RunAndReturn(run func(context.Context, string, string, []*entities.ExperimentTag) (string, *contract.Error)) *MockTrackingStore_CreateExperiment_Call

type MockTrackingStore_CreateRun_Call 

type MockTrackingStore_CreateRun_Call struct {
	*mock.Call
}

MockTrackingStore_CreateRun_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateRun'

func (*MockTrackingStore_CreateRun_Call) Return 

func (_c *MockTrackingStore_CreateRun_Call) Return(_a0 *entities.Run, _a1 *contract.Error) *MockTrackingStore_CreateRun_Call

func (*MockTrackingStore_CreateRun_Call) Run 

func (_c *MockTrackingStore_CreateRun_Call) Run(run func(ctx context.Context, experimentID string, userID string, startTime int64, tags []*entities.RunTag, runName string)) *MockTrackingStore_CreateRun_Call

func (*MockTrackingStore_CreateRun_Call) RunAndReturn 

func (_c *MockTrackingStore_CreateRun_Call) RunAndReturn(run func(context.Context, string, string, int64, []*entities.RunTag, string) (*entities.Run, *contract.Error)) *MockTrackingStore_CreateRun_Call

type MockTrackingStore_DeleteExperiment_Call 

type MockTrackingStore_DeleteExperiment_Call struct {
	*mock.Call
}

MockTrackingStore_DeleteExperiment_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteExperiment'

func (*MockTrackingStore_DeleteExperiment_Call) Return 

func (_c *MockTrackingStore_DeleteExperiment_Call) Return(_a0 *contract.Error) *MockTrackingStore_DeleteExperiment_Call

func (*MockTrackingStore_DeleteExperiment_Call) Run 

func (_c *MockTrackingStore_DeleteExperiment_Call) Run(run func(ctx context.Context, id string)) *MockTrackingStore_DeleteExperiment_Call

func (*MockTrackingStore_DeleteExperiment_Call) RunAndReturn 

func (_c *MockTrackingStore_DeleteExperiment_Call) RunAndReturn(run func(context.Context, string) *contract.Error) *MockTrackingStore_DeleteExperiment_Call

type MockTrackingStore_DeleteRun_Call 

type MockTrackingStore_DeleteRun_Call struct {
	*mock.Call
}

MockTrackingStore_DeleteRun_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteRun'

func (*MockTrackingStore_DeleteRun_Call) Return 

func (_c *MockTrackingStore_DeleteRun_Call) Return(_a0 *contract.Error) *MockTrackingStore_DeleteRun_Call

func (*MockTrackingStore_DeleteRun_Call) Run 

func (_c *MockTrackingStore_DeleteRun_Call) Run(run func(ctx context.Context, runID string)) *MockTrackingStore_DeleteRun_Call

func (*MockTrackingStore_DeleteRun_Call) RunAndReturn 

func (_c *MockTrackingStore_DeleteRun_Call) RunAndReturn(run func(context.Context, string) *contract.Error) *MockTrackingStore_DeleteRun_Call

type MockTrackingStore_DeleteTag_Call 

type MockTrackingStore_DeleteTag_Call struct {
	*mock.Call
}

MockTrackingStore_DeleteTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteTag'

func (*MockTrackingStore_DeleteTag_Call) Return 

func (_c *MockTrackingStore_DeleteTag_Call) Return(_a0 *contract.Error) *MockTrackingStore_DeleteTag_Call

func (*MockTrackingStore_DeleteTag_Call) Run 

func (_c *MockTrackingStore_DeleteTag_Call) Run(run func(ctx context.Context, runID string, key string)) *MockTrackingStore_DeleteTag_Call

func (*MockTrackingStore_DeleteTag_Call) RunAndReturn 

func (_c *MockTrackingStore_DeleteTag_Call) RunAndReturn(run func(context.Context, string, string) *contract.Error) *MockTrackingStore_DeleteTag_Call

type MockTrackingStore_DeleteTraceTag_Call 

type MockTrackingStore_DeleteTraceTag_Call struct {
	*mock.Call
}

MockTrackingStore_DeleteTraceTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteTraceTag'

func (*MockTrackingStore_DeleteTraceTag_Call) Return 

func (_c *MockTrackingStore_DeleteTraceTag_Call) Return(_a0 *contract.Error) *MockTrackingStore_DeleteTraceTag_Call

func (*MockTrackingStore_DeleteTraceTag_Call) Run 

func (_c *MockTrackingStore_DeleteTraceTag_Call) Run(run func(ctx context.Context, tag *entities.TraceTag)) *MockTrackingStore_DeleteTraceTag_Call

func (*MockTrackingStore_DeleteTraceTag_Call) RunAndReturn 

func (_c *MockTrackingStore_DeleteTraceTag_Call) RunAndReturn(run func(context.Context, *entities.TraceTag) *contract.Error) *MockTrackingStore_DeleteTraceTag_Call

type MockTrackingStore_DeleteTraces_Call 

type MockTrackingStore_DeleteTraces_Call struct {
	*mock.Call
}

MockTrackingStore_DeleteTraces_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteTraces'

func (*MockTrackingStore_DeleteTraces_Call) Return 

func (_c *MockTrackingStore_DeleteTraces_Call) Return(_a0 int32, _a1 *contract.Error) *MockTrackingStore_DeleteTraces_Call

func (*MockTrackingStore_DeleteTraces_Call) Run 

func (_c *MockTrackingStore_DeleteTraces_Call) Run(run func(ctx context.Context, experimentID string, maxTimestampMillis int64, maxTraces int32, requestIDs []string)) *MockTrackingStore_DeleteTraces_Call

func (*MockTrackingStore_DeleteTraces_Call) RunAndReturn 

func (_c *MockTrackingStore_DeleteTraces_Call) RunAndReturn(run func(context.Context, string, int64, int32, []string) (int32, *contract.Error)) *MockTrackingStore_DeleteTraces_Call

type MockTrackingStore_Destroy_Call 

type MockTrackingStore_Destroy_Call struct {
	*mock.Call
}

MockTrackingStore_Destroy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Destroy'

func (*MockTrackingStore_Destroy_Call) Return 

func (_c *MockTrackingStore_Destroy_Call) Return(_a0 error) *MockTrackingStore_Destroy_Call

func (*MockTrackingStore_Destroy_Call) Run 

func (_c *MockTrackingStore_Destroy_Call) Run(run func()) *MockTrackingStore_Destroy_Call

func (*MockTrackingStore_Destroy_Call) RunAndReturn 

func (_c *MockTrackingStore_Destroy_Call) RunAndReturn(run func() error) *MockTrackingStore_Destroy_Call

type MockTrackingStore_EndTrace_Call 

type MockTrackingStore_EndTrace_Call struct {
	*mock.Call
}

MockTrackingStore_EndTrace_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'EndTrace'

func (*MockTrackingStore_EndTrace_Call) Return 

func (_c *MockTrackingStore_EndTrace_Call) Return(_a0 *entities.TraceInfo, _a1 error) *MockTrackingStore_EndTrace_Call

func (*MockTrackingStore_EndTrace_Call) Run 

func (_c *MockTrackingStore_EndTrace_Call) Run(run func(ctx context.Context, reqeustID string, timestampMS int64, status string, metadata []*entities.TraceRequestMetadata, tags []*entities.TraceTag)) *MockTrackingStore_EndTrace_Call

func (*MockTrackingStore_EndTrace_Call) RunAndReturn 

func (_c *MockTrackingStore_EndTrace_Call) RunAndReturn(run func(context.Context, string, int64, string, []*entities.TraceRequestMetadata, []*entities.TraceTag) (*entities.TraceInfo, error)) *MockTrackingStore_EndTrace_Call

type MockTrackingStore_Expecter 

type MockTrackingStore_Expecter struct {
	// contains filtered or unexported fields
}

func (*MockTrackingStore_Expecter) CreateExperiment 

func (_e *MockTrackingStore_Expecter) CreateExperiment(ctx interface{}, name interface{}, artifactLocation interface{}, tags interface{}) *MockTrackingStore_CreateExperiment_Call

CreateExperiment is a helper method to define mock.On call

  • ctx context.Context
  • name string
  • artifactLocation string
  • tags []*entities.ExperimentTag

func (*MockTrackingStore_Expecter) CreateRun 

func (_e *MockTrackingStore_Expecter) CreateRun(ctx interface{}, experimentID interface{}, userID interface{}, startTime interface{}, tags interface{}, runName interface{}) *MockTrackingStore_CreateRun_Call

CreateRun is a helper method to define mock.On call

  • ctx context.Context
  • experimentID string
  • userID string
  • startTime int64
  • tags []*entities.RunTag
  • runName string

func (*MockTrackingStore_Expecter) DeleteExperiment 

func (_e *MockTrackingStore_Expecter) DeleteExperiment(ctx interface{}, id interface{}) *MockTrackingStore_DeleteExperiment_Call

DeleteExperiment is a helper method to define mock.On call

  • ctx context.Context
  • id string

func (*MockTrackingStore_Expecter) DeleteRun 

func (_e *MockTrackingStore_Expecter) DeleteRun(ctx interface{}, runID interface{}) *MockTrackingStore_DeleteRun_Call

DeleteRun is a helper method to define mock.On call

  • ctx context.Context
  • runID string

func (*MockTrackingStore_Expecter) DeleteTag 

func (_e *MockTrackingStore_Expecter) DeleteTag(ctx interface{}, runID interface{}, key interface{}) *MockTrackingStore_DeleteTag_Call

DeleteTag is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • key string

func (*MockTrackingStore_Expecter) DeleteTraceTag 

func (_e *MockTrackingStore_Expecter) DeleteTraceTag(ctx interface{}, tag interface{}) *MockTrackingStore_DeleteTraceTag_Call

DeleteTraceTag is a helper method to define mock.On call

  • ctx context.Context
  • tag *entities.TraceTag

func (*MockTrackingStore_Expecter) DeleteTraces 

func (_e *MockTrackingStore_Expecter) DeleteTraces(ctx interface{}, experimentID interface{}, maxTimestampMillis interface{}, maxTraces interface{}, requestIDs interface{}) *MockTrackingStore_DeleteTraces_Call

DeleteTraces is a helper method to define mock.On call

  • ctx context.Context
  • experimentID string
  • maxTimestampMillis int64
  • maxTraces int32
  • requestIDs []string

func (*MockTrackingStore_Expecter) Destroy 

func (_e *MockTrackingStore_Expecter) Destroy() *MockTrackingStore_Destroy_Call

Destroy is a helper method to define mock.On call

func (*MockTrackingStore_Expecter) EndTrace 

func (_e *MockTrackingStore_Expecter) EndTrace(ctx interface{}, reqeustID interface{}, timestampMS interface{}, status interface{}, metadata interface{}, tags interface{}) *MockTrackingStore_EndTrace_Call

EndTrace is a helper method to define mock.On call

  • ctx context.Context
  • reqeustID string
  • timestampMS int64
  • status string
  • metadata []*entities.TraceRequestMetadata
  • tags []*entities.TraceTag

func (*MockTrackingStore_Expecter) GetExperiment 

func (_e *MockTrackingStore_Expecter) GetExperiment(ctx interface{}, id interface{}) *MockTrackingStore_GetExperiment_Call

GetExperiment is a helper method to define mock.On call

  • ctx context.Context
  • id string

func (*MockTrackingStore_Expecter) GetExperimentByName 

func (_e *MockTrackingStore_Expecter) GetExperimentByName(ctx interface{}, name interface{}) *MockTrackingStore_GetExperimentByName_Call

GetExperimentByName is a helper method to define mock.On call

  • ctx context.Context
  • name string

func (*MockTrackingStore_Expecter) GetMetricHistory 

func (_e *MockTrackingStore_Expecter) GetMetricHistory(ctx interface{}, runID interface{}, metricKey interface{}) *MockTrackingStore_GetMetricHistory_Call

GetMetricHistory is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • metricKey string

func (*MockTrackingStore_Expecter) GetRun 

func (_e *MockTrackingStore_Expecter) GetRun(ctx interface{}, runID interface{}) *MockTrackingStore_GetRun_Call

GetRun is a helper method to define mock.On call

  • ctx context.Context
  • runID string

func (*MockTrackingStore_Expecter) GetRunTag 

func (_e *MockTrackingStore_Expecter) GetRunTag(ctx interface{}, runID interface{}, tagKey interface{}) *MockTrackingStore_GetRunTag_Call

GetRunTag is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • tagKey string

func (*MockTrackingStore_Expecter) GetTraceInfo 

func (_e *MockTrackingStore_Expecter) GetTraceInfo(ctx interface{}, reqeustID interface{}) *MockTrackingStore_GetTraceInfo_Call

GetTraceInfo is a helper method to define mock.On call

  • ctx context.Context
  • reqeustID string

func (*MockTrackingStore_Expecter) GetTraceTag 

func (_e *MockTrackingStore_Expecter) GetTraceTag(ctx interface{}, requestID interface{}, key interface{}) *MockTrackingStore_GetTraceTag_Call

GetTraceTag is a helper method to define mock.On call

  • ctx context.Context
  • requestID string
  • key string

func (*MockTrackingStore_Expecter) LogBatch 

func (_e *MockTrackingStore_Expecter) LogBatch(ctx interface{}, runID interface{}, metrics interface{}, params interface{}, tags interface{}) *MockTrackingStore_LogBatch_Call

LogBatch is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • metrics []*entities.Metric
  • params []*entities.Param
  • tags []*entities.RunTag

func (*MockTrackingStore_Expecter) LogInputs 

func (_e *MockTrackingStore_Expecter) LogInputs(ctx interface{}, runID interface{}, datasets interface{}) *MockTrackingStore_LogInputs_Call

LogInputs is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • datasets []*entities.DatasetInput

func (*MockTrackingStore_Expecter) LogMetric 

func (_e *MockTrackingStore_Expecter) LogMetric(ctx interface{}, runID interface{}, metric interface{}) *MockTrackingStore_LogMetric_Call

LogMetric is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • metric *entities.Metric

func (*MockTrackingStore_Expecter) LogParam 

func (_e *MockTrackingStore_Expecter) LogParam(ctx interface{}, runID interface{}, metric interface{}) *MockTrackingStore_LogParam_Call

LogParam is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • metric *entities.Param

func (*MockTrackingStore_Expecter) RenameExperiment 

func (_e *MockTrackingStore_Expecter) RenameExperiment(ctx interface{}, experimentID interface{}, name interface{}) *MockTrackingStore_RenameExperiment_Call

RenameExperiment is a helper method to define mock.On call

  • ctx context.Context
  • experimentID string
  • name string

func (*MockTrackingStore_Expecter) RestoreExperiment 

func (_e *MockTrackingStore_Expecter) RestoreExperiment(ctx interface{}, id interface{}) *MockTrackingStore_RestoreExperiment_Call

RestoreExperiment is a helper method to define mock.On call

  • ctx context.Context
  • id string

func (*MockTrackingStore_Expecter) RestoreRun 

func (_e *MockTrackingStore_Expecter) RestoreRun(ctx interface{}, runID interface{}) *MockTrackingStore_RestoreRun_Call

RestoreRun is a helper method to define mock.On call

  • ctx context.Context
  • runID string

func (*MockTrackingStore_Expecter) SearchExperiments 

func (_e *MockTrackingStore_Expecter) SearchExperiments(ctx interface{}, experimentViewType interface{}, maxResults interface{}, filter interface{}, orderBy interface{}, pageToken interface{}) *MockTrackingStore_SearchExperiments_Call

SearchExperiments is a helper method to define mock.On call

  • ctx context.Context
  • experimentViewType protos.ViewType
  • maxResults int64
  • filter string
  • orderBy []string
  • pageToken string

func (*MockTrackingStore_Expecter) SearchRuns 

func (_e *MockTrackingStore_Expecter) SearchRuns(ctx interface{}, experimentIDs interface{}, filter interface{}, runViewType interface{}, maxResults interface{}, orderBy interface{}, pageToken interface{}) *MockTrackingStore_SearchRuns_Call

SearchRuns is a helper method to define mock.On call

  • ctx context.Context
  • experimentIDs []string
  • filter string
  • runViewType protos.ViewType
  • maxResults int
  • orderBy []string
  • pageToken string

func (*MockTrackingStore_Expecter) SetExperimentTag 

func (_e *MockTrackingStore_Expecter) SetExperimentTag(ctx interface{}, experimentID interface{}, key interface{}, value interface{}) *MockTrackingStore_SetExperimentTag_Call

SetExperimentTag is a helper method to define mock.On call

  • ctx context.Context
  • experimentID string
  • key string
  • value string

func (*MockTrackingStore_Expecter) SetTag 

func (_e *MockTrackingStore_Expecter) SetTag(ctx interface{}, runID interface{}, key interface{}, value interface{}) *MockTrackingStore_SetTag_Call

SetTag is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • key string
  • value string

func (*MockTrackingStore_Expecter) SetTrace 

func (_e *MockTrackingStore_Expecter) SetTrace(ctx interface{}, experimentID interface{}, timestampMS interface{}, metadata interface{}, tags interface{}) *MockTrackingStore_SetTrace_Call

SetTrace is a helper method to define mock.On call

  • ctx context.Context
  • experimentID string
  • timestampMS int64
  • metadata []*entities.TraceRequestMetadata
  • tags []*entities.TraceTag

func (*MockTrackingStore_Expecter) SetTraceTag 

func (_e *MockTrackingStore_Expecter) SetTraceTag(ctx interface{}, requestID interface{}, key interface{}, value interface{}) *MockTrackingStore_SetTraceTag_Call

SetTraceTag is a helper method to define mock.On call

  • ctx context.Context
  • requestID string
  • key string
  • value string

func (*MockTrackingStore_Expecter) UpdateRun 

func (_e *MockTrackingStore_Expecter) UpdateRun(ctx interface{}, runID interface{}, runStatus interface{}, endTime interface{}, runName interface{}) *MockTrackingStore_UpdateRun_Call

UpdateRun is a helper method to define mock.On call

  • ctx context.Context
  • runID string
  • runStatus string
  • endTime *int64
  • runName string

type MockTrackingStore_GetExperimentByName_Call 

type MockTrackingStore_GetExperimentByName_Call struct {
	*mock.Call
}

MockTrackingStore_GetExperimentByName_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetExperimentByName'

func (*MockTrackingStore_GetExperimentByName_Call) Return 

func (_c *MockTrackingStore_GetExperimentByName_Call) Return(_a0 *entities.Experiment, _a1 *contract.Error) *MockTrackingStore_GetExperimentByName_Call

func (*MockTrackingStore_GetExperimentByName_Call) Run 

func (_c *MockTrackingStore_GetExperimentByName_Call) Run(run func(ctx context.Context, name string)) *MockTrackingStore_GetExperimentByName_Call

func (*MockTrackingStore_GetExperimentByName_Call) RunAndReturn 

func (_c *MockTrackingStore_GetExperimentByName_Call) RunAndReturn(run func(context.Context, string) (*entities.Experiment, *contract.Error)) *MockTrackingStore_GetExperimentByName_Call

type MockTrackingStore_GetExperiment_Call 

type MockTrackingStore_GetExperiment_Call struct {
	*mock.Call
}

MockTrackingStore_GetExperiment_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetExperiment'

func (*MockTrackingStore_GetExperiment_Call) Return 

func (_c *MockTrackingStore_GetExperiment_Call) Return(_a0 *entities.Experiment, _a1 *contract.Error) *MockTrackingStore_GetExperiment_Call

func (*MockTrackingStore_GetExperiment_Call) Run 

func (_c *MockTrackingStore_GetExperiment_Call) Run(run func(ctx context.Context, id string)) *MockTrackingStore_GetExperiment_Call

func (*MockTrackingStore_GetExperiment_Call) RunAndReturn 

func (_c *MockTrackingStore_GetExperiment_Call) RunAndReturn(run func(context.Context, string) (*entities.Experiment, *contract.Error)) *MockTrackingStore_GetExperiment_Call

type MockTrackingStore_GetMetricHistory_Call 

type MockTrackingStore_GetMetricHistory_Call struct {
	*mock.Call
}

MockTrackingStore_GetMetricHistory_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetMetricHistory'

func (*MockTrackingStore_GetMetricHistory_Call) Return 

func (_c *MockTrackingStore_GetMetricHistory_Call) Return(_a0 []*entities.Metric, _a1 *contract.Error) *MockTrackingStore_GetMetricHistory_Call

func (*MockTrackingStore_GetMetricHistory_Call) Run 

func (_c *MockTrackingStore_GetMetricHistory_Call) Run(run func(ctx context.Context, runID string, metricKey string)) *MockTrackingStore_GetMetricHistory_Call

func (*MockTrackingStore_GetMetricHistory_Call) RunAndReturn 

func (_c *MockTrackingStore_GetMetricHistory_Call) RunAndReturn(run func(context.Context, string, string) ([]*entities.Metric, *contract.Error)) *MockTrackingStore_GetMetricHistory_Call

type MockTrackingStore_GetRunTag_Call 

type MockTrackingStore_GetRunTag_Call struct {
	*mock.Call
}

MockTrackingStore_GetRunTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetRunTag'

func (*MockTrackingStore_GetRunTag_Call) Return 

func (_c *MockTrackingStore_GetRunTag_Call) Return(_a0 *entities.RunTag, _a1 *contract.Error) *MockTrackingStore_GetRunTag_Call

func (*MockTrackingStore_GetRunTag_Call) Run 

func (_c *MockTrackingStore_GetRunTag_Call) Run(run func(ctx context.Context, runID string, tagKey string)) *MockTrackingStore_GetRunTag_Call

func (*MockTrackingStore_GetRunTag_Call) RunAndReturn 

func (_c *MockTrackingStore_GetRunTag_Call) RunAndReturn(run func(context.Context, string, string) (*entities.RunTag, *contract.Error)) *MockTrackingStore_GetRunTag_Call

type MockTrackingStore_GetRun_Call 

type MockTrackingStore_GetRun_Call struct {
	*mock.Call
}

MockTrackingStore_GetRun_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetRun'

func (*MockTrackingStore_GetRun_Call) Return 

func (_c *MockTrackingStore_GetRun_Call) Return(_a0 *entities.Run, _a1 *contract.Error) *MockTrackingStore_GetRun_Call

func (*MockTrackingStore_GetRun_Call) Run 

func (_c *MockTrackingStore_GetRun_Call) Run(run func(ctx context.Context, runID string)) *MockTrackingStore_GetRun_Call

func (*MockTrackingStore_GetRun_Call) RunAndReturn 

func (_c *MockTrackingStore_GetRun_Call) RunAndReturn(run func(context.Context, string) (*entities.Run, *contract.Error)) *MockTrackingStore_GetRun_Call

type MockTrackingStore_GetTraceInfo_Call 

type MockTrackingStore_GetTraceInfo_Call struct {
	*mock.Call
}

MockTrackingStore_GetTraceInfo_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetTraceInfo'

func (*MockTrackingStore_GetTraceInfo_Call) Return 

func (_c *MockTrackingStore_GetTraceInfo_Call) Return(_a0 *entities.TraceInfo, _a1 *contract.Error) *MockTrackingStore_GetTraceInfo_Call

func (*MockTrackingStore_GetTraceInfo_Call) Run 

func (_c *MockTrackingStore_GetTraceInfo_Call) Run(run func(ctx context.Context, reqeustID string)) *MockTrackingStore_GetTraceInfo_Call

func (*MockTrackingStore_GetTraceInfo_Call) RunAndReturn 

func (_c *MockTrackingStore_GetTraceInfo_Call) RunAndReturn(run func(context.Context, string) (*entities.TraceInfo, *contract.Error)) *MockTrackingStore_GetTraceInfo_Call

type MockTrackingStore_GetTraceTag_Call 

type MockTrackingStore_GetTraceTag_Call struct {
	*mock.Call
}

MockTrackingStore_GetTraceTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetTraceTag'

func (*MockTrackingStore_GetTraceTag_Call) Return 

func (_c *MockTrackingStore_GetTraceTag_Call) Return(_a0 *entities.TraceTag, _a1 *contract.Error) *MockTrackingStore_GetTraceTag_Call

func (*MockTrackingStore_GetTraceTag_Call) Run 

func (_c *MockTrackingStore_GetTraceTag_Call) Run(run func(ctx context.Context, requestID string, key string)) *MockTrackingStore_GetTraceTag_Call

func (*MockTrackingStore_GetTraceTag_Call) RunAndReturn 

func (_c *MockTrackingStore_GetTraceTag_Call) RunAndReturn(run func(context.Context, string, string) (*entities.TraceTag, *contract.Error)) *MockTrackingStore_GetTraceTag_Call

type MockTrackingStore_LogBatch_Call 

type MockTrackingStore_LogBatch_Call struct {
	*mock.Call
}

MockTrackingStore_LogBatch_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LogBatch'

func (*MockTrackingStore_LogBatch_Call) Return 

func (_c *MockTrackingStore_LogBatch_Call) Return(_a0 *contract.Error) *MockTrackingStore_LogBatch_Call

func (*MockTrackingStore_LogBatch_Call) Run 

func (_c *MockTrackingStore_LogBatch_Call) Run(run func(ctx context.Context, runID string, metrics []*entities.Metric, params []*entities.Param, tags []*entities.RunTag)) *MockTrackingStore_LogBatch_Call

func (*MockTrackingStore_LogBatch_Call) RunAndReturn 

func (_c *MockTrackingStore_LogBatch_Call) RunAndReturn(run func(context.Context, string, []*entities.Metric, []*entities.Param, []*entities.RunTag) *contract.Error) *MockTrackingStore_LogBatch_Call

type MockTrackingStore_LogInputs_Call 

type MockTrackingStore_LogInputs_Call struct {
	*mock.Call
}

MockTrackingStore_LogInputs_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LogInputs'

func (*MockTrackingStore_LogInputs_Call) Return 

func (_c *MockTrackingStore_LogInputs_Call) Return(_a0 *contract.Error) *MockTrackingStore_LogInputs_Call

func (*MockTrackingStore_LogInputs_Call) Run 

func (_c *MockTrackingStore_LogInputs_Call) Run(run func(ctx context.Context, runID string, datasets []*entities.DatasetInput)) *MockTrackingStore_LogInputs_Call

func (*MockTrackingStore_LogInputs_Call) RunAndReturn 

func (_c *MockTrackingStore_LogInputs_Call) RunAndReturn(run func(context.Context, string, []*entities.DatasetInput) *contract.Error) *MockTrackingStore_LogInputs_Call

type MockTrackingStore_LogMetric_Call 

type MockTrackingStore_LogMetric_Call struct {
	*mock.Call
}

MockTrackingStore_LogMetric_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LogMetric'

func (*MockTrackingStore_LogMetric_Call) Return 

func (_c *MockTrackingStore_LogMetric_Call) Return(_a0 *contract.Error) *MockTrackingStore_LogMetric_Call

func (*MockTrackingStore_LogMetric_Call) Run 

func (_c *MockTrackingStore_LogMetric_Call) Run(run func(ctx context.Context, runID string, metric *entities.Metric)) *MockTrackingStore_LogMetric_Call

func (*MockTrackingStore_LogMetric_Call) RunAndReturn 

func (_c *MockTrackingStore_LogMetric_Call) RunAndReturn(run func(context.Context, string, *entities.Metric) *contract.Error) *MockTrackingStore_LogMetric_Call

type MockTrackingStore_LogParam_Call 

type MockTrackingStore_LogParam_Call struct {
	*mock.Call
}

MockTrackingStore_LogParam_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LogParam'

func (*MockTrackingStore_LogParam_Call) Return 

func (_c *MockTrackingStore_LogParam_Call) Return(_a0 *contract.Error) *MockTrackingStore_LogParam_Call

func (*MockTrackingStore_LogParam_Call) Run 

func (_c *MockTrackingStore_LogParam_Call) Run(run func(ctx context.Context, runID string, metric *entities.Param)) *MockTrackingStore_LogParam_Call

func (*MockTrackingStore_LogParam_Call) RunAndReturn 

func (_c *MockTrackingStore_LogParam_Call) RunAndReturn(run func(context.Context, string, *entities.Param) *contract.Error) *MockTrackingStore_LogParam_Call

type MockTrackingStore_RenameExperiment_Call 

type MockTrackingStore_RenameExperiment_Call struct {
	*mock.Call
}

MockTrackingStore_RenameExperiment_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RenameExperiment'

func (*MockTrackingStore_RenameExperiment_Call) Return 

func (_c *MockTrackingStore_RenameExperiment_Call) Return(_a0 *contract.Error) *MockTrackingStore_RenameExperiment_Call

func (*MockTrackingStore_RenameExperiment_Call) Run 

func (_c *MockTrackingStore_RenameExperiment_Call) Run(run func(ctx context.Context, experimentID string, name string)) *MockTrackingStore_RenameExperiment_Call

func (*MockTrackingStore_RenameExperiment_Call) RunAndReturn 

func (_c *MockTrackingStore_RenameExperiment_Call) RunAndReturn(run func(context.Context, string, string) *contract.Error) *MockTrackingStore_RenameExperiment_Call

type MockTrackingStore_RestoreExperiment_Call 

type MockTrackingStore_RestoreExperiment_Call struct {
	*mock.Call
}

MockTrackingStore_RestoreExperiment_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RestoreExperiment'

func (*MockTrackingStore_RestoreExperiment_Call) Return 

func (_c *MockTrackingStore_RestoreExperiment_Call) Return(_a0 *contract.Error) *MockTrackingStore_RestoreExperiment_Call

func (*MockTrackingStore_RestoreExperiment_Call) Run 

func (_c *MockTrackingStore_RestoreExperiment_Call) Run(run func(ctx context.Context, id string)) *MockTrackingStore_RestoreExperiment_Call

func (*MockTrackingStore_RestoreExperiment_Call) RunAndReturn 

func (_c *MockTrackingStore_RestoreExperiment_Call) RunAndReturn(run func(context.Context, string) *contract.Error) *MockTrackingStore_RestoreExperiment_Call

type MockTrackingStore_RestoreRun_Call 

type MockTrackingStore_RestoreRun_Call struct {
	*mock.Call
}

MockTrackingStore_RestoreRun_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RestoreRun'

func (*MockTrackingStore_RestoreRun_Call) Return 

func (_c *MockTrackingStore_RestoreRun_Call) Return(_a0 *contract.Error) *MockTrackingStore_RestoreRun_Call

func (*MockTrackingStore_RestoreRun_Call) Run 

func (_c *MockTrackingStore_RestoreRun_Call) Run(run func(ctx context.Context, runID string)) *MockTrackingStore_RestoreRun_Call

func (*MockTrackingStore_RestoreRun_Call) RunAndReturn 

func (_c *MockTrackingStore_RestoreRun_Call) RunAndReturn(run func(context.Context, string) *contract.Error) *MockTrackingStore_RestoreRun_Call

type MockTrackingStore_SearchExperiments_Call 

type MockTrackingStore_SearchExperiments_Call struct {
	*mock.Call
}

MockTrackingStore_SearchExperiments_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SearchExperiments'

func (*MockTrackingStore_SearchExperiments_Call) Return 

func (_c *MockTrackingStore_SearchExperiments_Call) Return(_a0 []*entities.Experiment, _a1 string, _a2 *contract.Error) *MockTrackingStore_SearchExperiments_Call

func (*MockTrackingStore_SearchExperiments_Call) Run 

func (_c *MockTrackingStore_SearchExperiments_Call) Run(run func(ctx context.Context, experimentViewType protos.ViewType, maxResults int64, filter string, orderBy []string, pageToken string)) *MockTrackingStore_SearchExperiments_Call

func (*MockTrackingStore_SearchExperiments_Call) RunAndReturn 

func (_c *MockTrackingStore_SearchExperiments_Call) RunAndReturn(run func(context.Context, protos.ViewType, int64, string, []string, string) ([]*entities.Experiment, string, *contract.Error)) *MockTrackingStore_SearchExperiments_Call

type MockTrackingStore_SearchRuns_Call 

type MockTrackingStore_SearchRuns_Call struct {
	*mock.Call
}

MockTrackingStore_SearchRuns_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SearchRuns'

func (*MockTrackingStore_SearchRuns_Call) Return 

func (_c *MockTrackingStore_SearchRuns_Call) Return(_a0 []*entities.Run, _a1 string, _a2 *contract.Error) *MockTrackingStore_SearchRuns_Call

func (*MockTrackingStore_SearchRuns_Call) Run 

func (_c *MockTrackingStore_SearchRuns_Call) Run(run func(ctx context.Context, experimentIDs []string, filter string, runViewType protos.ViewType, maxResults int, orderBy []string, pageToken string)) *MockTrackingStore_SearchRuns_Call

func (*MockTrackingStore_SearchRuns_Call) RunAndReturn 

func (_c *MockTrackingStore_SearchRuns_Call) RunAndReturn(run func(context.Context, []string, string, protos.ViewType, int, []string, string) ([]*entities.Run, string, *contract.Error)) *MockTrackingStore_SearchRuns_Call

type MockTrackingStore_SetExperimentTag_Call 

type MockTrackingStore_SetExperimentTag_Call struct {
	*mock.Call
}

MockTrackingStore_SetExperimentTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetExperimentTag'

func (*MockTrackingStore_SetExperimentTag_Call) Return 

func (_c *MockTrackingStore_SetExperimentTag_Call) Return(_a0 *contract.Error) *MockTrackingStore_SetExperimentTag_Call

func (*MockTrackingStore_SetExperimentTag_Call) Run 

func (_c *MockTrackingStore_SetExperimentTag_Call) Run(run func(ctx context.Context, experimentID string, key string, value string)) *MockTrackingStore_SetExperimentTag_Call

func (*MockTrackingStore_SetExperimentTag_Call) RunAndReturn 

func (_c *MockTrackingStore_SetExperimentTag_Call) RunAndReturn(run func(context.Context, string, string, string) *contract.Error) *MockTrackingStore_SetExperimentTag_Call

type MockTrackingStore_SetTag_Call 

type MockTrackingStore_SetTag_Call struct {
	*mock.Call
}

MockTrackingStore_SetTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetTag'

func (*MockTrackingStore_SetTag_Call) Return 

func (_c *MockTrackingStore_SetTag_Call) Return(_a0 *contract.Error) *MockTrackingStore_SetTag_Call

func (*MockTrackingStore_SetTag_Call) Run 

func (_c *MockTrackingStore_SetTag_Call) Run(run func(ctx context.Context, runID string, key string, value string)) *MockTrackingStore_SetTag_Call

func (*MockTrackingStore_SetTag_Call) RunAndReturn 

func (_c *MockTrackingStore_SetTag_Call) RunAndReturn(run func(context.Context, string, string, string) *contract.Error) *MockTrackingStore_SetTag_Call

type MockTrackingStore_SetTraceTag_Call 

type MockTrackingStore_SetTraceTag_Call struct {
	*mock.Call
}

MockTrackingStore_SetTraceTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetTraceTag'

func (*MockTrackingStore_SetTraceTag_Call) Return 

func (_c *MockTrackingStore_SetTraceTag_Call) Return(_a0 error) *MockTrackingStore_SetTraceTag_Call

func (*MockTrackingStore_SetTraceTag_Call) Run 

func (_c *MockTrackingStore_SetTraceTag_Call) Run(run func(ctx context.Context, requestID string, key string, value string)) *MockTrackingStore_SetTraceTag_Call

func (*MockTrackingStore_SetTraceTag_Call) RunAndReturn 

func (_c *MockTrackingStore_SetTraceTag_Call) RunAndReturn(run func(context.Context, string, string, string) error) *MockTrackingStore_SetTraceTag_Call

type MockTrackingStore_SetTrace_Call 

type MockTrackingStore_SetTrace_Call struct {
	*mock.Call
}

MockTrackingStore_SetTrace_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetTrace'

func (*MockTrackingStore_SetTrace_Call) Return 

func (_c *MockTrackingStore_SetTrace_Call) Return(_a0 *entities.TraceInfo, _a1 error) *MockTrackingStore_SetTrace_Call

func (*MockTrackingStore_SetTrace_Call) Run 

func (_c *MockTrackingStore_SetTrace_Call) Run(run func(ctx context.Context, experimentID string, timestampMS int64, metadata []*entities.TraceRequestMetadata, tags []*entities.TraceTag)) *MockTrackingStore_SetTrace_Call

func (*MockTrackingStore_SetTrace_Call) RunAndReturn 

func (_c *MockTrackingStore_SetTrace_Call) RunAndReturn(run func(context.Context, string, int64, []*entities.TraceRequestMetadata, []*entities.TraceTag) (*entities.TraceInfo, error)) *MockTrackingStore_SetTrace_Call

type MockTrackingStore_UpdateRun_Call 

type MockTrackingStore_UpdateRun_Call struct {
	*mock.Call
}

MockTrackingStore_UpdateRun_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UpdateRun'

func (*MockTrackingStore_UpdateRun_Call) Return 

func (_c *MockTrackingStore_UpdateRun_Call) Return(_a0 *contract.Error) *MockTrackingStore_UpdateRun_Call

func (*MockTrackingStore_UpdateRun_Call) Run 

func (_c *MockTrackingStore_UpdateRun_Call) Run(run func(ctx context.Context, runID string, runStatus string, endTime *int64, runName string)) *MockTrackingStore_UpdateRun_Call

func (*MockTrackingStore_UpdateRun_Call) RunAndReturn 

func (_c *MockTrackingStore_UpdateRun_Call) RunAndReturn(run func(context.Context, string, string, *int64, string) *contract.Error) *MockTrackingStore_UpdateRun_Call

type RunTrackingStore 

type RunTrackingStore interface {
	GetRun(ctx context.Context, runID string) (*entities.Run, *contract.Error)
	CreateRun(
		ctx context.Context,
		experimentID string,
		userID string,
		startTime int64,
		tags []*entities.RunTag,
		runName string,
	) (*entities.Run, *contract.Error)
	UpdateRun(
		ctx context.Context,
		runID string,
		runStatus string,
		endTime *int64,
		runName string,
	) *contract.Error
	DeleteRun(ctx context.Context, runID string) *contract.Error
	RestoreRun(ctx context.Context, runID string) *contract.Error
	GetRunTag(ctx context.Context, runID, tagKey string) (*entities.RunTag, *contract.Error)
	DeleteTag(ctx context.Context, runID, key string) *contract.Error
	SetTag(ctx context.Context, runID, key, value string) *contract.Error
}

type TraceTrackingStore 

type TraceTrackingStore interface {
	SetTrace(
		ctx context.Context,
		experimentID string,
		timestampMS int64,
		metadata []*entities.TraceRequestMetadata,
		tags []*entities.TraceTag,
	) (*entities.TraceInfo, error)
	EndTrace(
		ctx context.Context,
		reqeustID string,
		timestampMS int64,
		status string,
		metadata []*entities.TraceRequestMetadata,
		tags []*entities.TraceTag,
	) (*entities.TraceInfo, error)
	GetTraceInfo(ctx context.Context, reqeustID string) (*entities.TraceInfo, *contract.Error)
	SetTraceTag(ctx context.Context, requestID, key, value string) error
	GetTraceTag(ctx context.Context, requestID, key string) (*entities.TraceTag, *contract.Error)
	DeleteTraceTag(ctx context.Context, tag *entities.TraceTag) *contract.Error
	DeleteTraces(
		ctx context.Context,
		experimentID string,
		maxTimestampMillis int64,
		maxTraces int32,
		requestIDs []string,
	) (int32, *contract.Error)
}

type TrackingStore 

type TrackingStore interface {
	contract.Destroyer
	RunTrackingStore
	TraceTrackingStore
	MetricTrackingStore
	ExperimentTrackingStore
	InputTrackingStore
}

Source Files

View all Source files

Directories

Path Synopsis
sql
models

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.