libOrch

type APITeamOptions struct {
	Distribution `json:"distribution"`
}

type BaseGlobalState interface {
	Ctx() context.Context
	Logger() *logrus.Logger
	OrchestratorClient() *redis.Client
	JobId() string
	OrchestratorId() string
	MetricsStore() *BaseMetricsStore
	GetStatus() string
	SetStatus(string)

	CreditsManager() *lib.CreditsManager
	Standalone() bool
	FuncAuthClient() RunAuthClient
	LoadZones() []string
	MessageQueue() *MessageQueue
}

type BaseMetricsStore interface {
	InitMetricsStore(childJobs map[string]ChildJobDistribution)
	AddMessage(message WorkerMessage, workerLocation string, subFraction float64) error
	Cleanup()
}

type ChildJob struct {
	Job
	ChildJobId        string                 `json:"childJobId"`
	Options           libWorker.Options      `json:"options"`
	UnderlyingRequest map[string]interface{} `json:"underlyingRequest"`
	FinalRequest      map[string]interface{} `json:"finalRequest"`
	SubFraction       float64                `json:"subFraction"`
	WorkerConnection  *websocket.Conn
	ConnWriteMutex    *sync.Mutex
	ConnReadMutex     *sync.Mutex
}

type ChildJobDistribution struct {
	ChildJobs []*ChildJob `json:"jobs"`
}

type CollectionContext struct {
	Variables []libWorker.KeyValueItem `json:"variables"`
	Name      string                   `json:"name"`
}

type Distribution struct {
	LoadZone string `json:"loadZone"`
	Percent  int    `json:"percent"`
}

type EnvironmentContext struct {
	Variables []libWorker.KeyValueItem `json:"variables"`
	Name      string                   `json:"name"`
}

type Job struct {
	Id                     string                 `json:"id"`
	Source                 string                 `json:"source"`
	SourceName             string                 `json:"sourceName"`
	EnvironmentContext     *EnvironmentContext    `json:"environmentContext"`
	CollectionContext      *CollectionContext     `json:"collectionContext"`
	UnderlyingRequest      map[string]interface{} `json:"underlyingRequest"`
	FinalRequest           map[string]interface{} `json:"finalRequest"`
	AssignedOrchestrator   string                 `json:"assignedOrchestrator"`
	Scope                  Scope                  `json:"scope"`
	Options                *libWorker.Options     `json:"options"`
	VerifiedDomains        []string               `json:"verifiedDomains"`
	FuncModeInfo           *lib.FuncModeInfo      `json:"funcModeInfo"`
	PermittedLoadZones     []string               `json:"permittedLoadZones"`
	MaxTestDurationMinutes int64                  `json:"maxTestDurationMinutes"`
	MaxSimulatedUsers      int64                  `json:"maxSimulatedUsers"`
}

type LiveService struct {
	Location string
	Uri      string

	State runpb.Condition_State
}

type LocalhostFile struct {
	FileName string `json:"fileName"`
	Contents string `json:"contents"`
	Kind     string `json:"kind"`
}

type MarkMessage struct {
	Mark    string      `json:"mark"`
	Message interface{} `json:"message"`
}

type MessageQueue struct {
	Mutex sync.Mutex

	// The count of currently actively being sent messages
	QueueCount    int
	NewQueueCount chan int
}

type NamedConnection struct {
	Name       string
	Connection *websocket.Conn
}

type OrchestratorMessage struct {
	JobId          string    `json:"jobId"`
	Time           time.Time `json:"time"`
	OrchestratorId string    `json:"orchestratorId"`
	Message        string    `json:"message"`
	MessageType    string    `json:"messageType"`
}

type OrchestratorOrWorkerMessage struct {
	JobId          string    `json:"jobId"`
	Time           time.Time `json:"time"`
	ChildJobId     string    `json:"childJobId"`
	OrchestratorId string    `json:"orchestratorId"`
	WorkerId       string    `json:"workerId"`
	Message        string    `json:"message"`
	MessageType    string    `json:"messageType"`
}

type RunAuthClient interface {
	Services() []LiveService
	ExecuteService(gs BaseGlobalState, location string) (*websocket.Conn, error)
	CheckServiceAvailability(location string) error
}

type Scope struct {
	Variant         string `json:"variant"`
	VariantTargetId string `json:"variantTargetId"`
}

type WorkerConnections struct {
	Connections       map[string]*NamedConnection
	DefaultConnection *NamedConnection
}

type WorkerMessage struct {
	JobId       string    `json:"jobId"`
	ChildJobId  string    `json:"childJobId"`
	Time        time.Time `json:"time"`
	WorkerId    string    `json:"workerId"`
	Message     string    `json:"message"`
	MessageType string    `json:"messageType"`
}

type WorkerState struct {
	Status     string `json:"status"`
	WorkerId   string `json:"workerId"`
	ChildJobId string `json:"childJobId"`
}