trace

type EventFrame struct {
	Name   string `json:"name"`
	Parent int    `json:"parent,omitempty"`
}

type GitInfo struct {
	Commit string `json:"commit"`
	Date   string `json:"date"`
}

type JSONTrace struct {
	ID              string                `json:"id,omitempty"`
	UPREnabled      bool                  `json:"upr_enabled,omitempty"`
	TraceEvents     TraceEvents           `json:"traceEvents,omitempty"`
	DisplayTimeUnit string                `json:"displayTimeUnit,omitempty"`
	Frames          map[string]EventFrame `json:"stackFrames"`
	TimeUnit        string                `json:"timeUnit,omitempty"`
	OtherDataRaw    TraceOtherData        `json:"otherData,omitempty"`
	OtherData       []*TraceOtherData     `json:"otherDatas,omitempty"`
}

type JSONTraceEvent struct {
	Name       string      `json:"name,omitempty"`
	Category   string      `json:"cat,omitempty"`
	EventType  string      `json:"ph,omitempty"`
	Timestamp  int64       `json:"ts,omitempty"`  // displayTimeUnit
	Duration   int64       `json:"dur,omitempty"` // displayTimeUnit
	ProcessID  int64       `json:"pid"`
	ThreadID   int64       `json:"tid,omitempty"`
	Args       interface{} `json:"args,omitempty"`
	Stack      int         `json:"sf,omitempty"`
	EndStack   int         `json:"esf,omitempty"`
	Init       string      `json:"init_time,omitempty"`
	Start      int64       `json:"start,omitempty"`
	End        int64       `json:"end,omitempty"`
	UPREnabled bool        `json:"upr_enabled,omitempty"`
}

type MemoryInformation []*Trace

type Trace struct {
	ID                    string                `json:"id,omitempty"`
	ExperimentDescription string                `json:"experiment_description,omitempty"`
	UPREnabled            bool                  `json:"upr_enabled"`
	Iteration             int64                 `json:"iteration,omitempty"`
	StartTime             time.Time             `json:"start_time,omitempty"`
	EndTime               time.Time             `json:"end_time,omitempty"`
	TraceEvents           TraceEvents           `json:"traceEvents,omitempty"`
	DisplayTimeUnit       string                `json:"displayTimeUnit,omitempty"`
	Frames                map[string]EventFrame `json:"stackFrames"`
	TimeUnit              string                `json:"timeUnit,omitempty"`
	OtherDataRaw          *TraceOtherData       `json:"otherData,omitempty"`
	OtherData             []*TraceOtherData     `json:"otherDatas,omitempty"`
}

type TraceEvent struct {
	Name       string        `json:"name,omitempty"`
	Category   string        `json:"cat,omitempty"`
	EventType  string        `json:"ph,omitempty"`
	Timestamp  int64         `json:"ts"`  // displayTimeUnit
	Duration   time.Duration `json:"dur"` // displayTimeUnit
	ProcessID  int64         `json:"pid"`
	ThreadID   int64         `json:"tid,omitempty"`
	Args       interface{}   `json:"args,omitempty"`
	Stack      int           `json:"sf,omitempty"`
	EndStack   int           `json:"esf,omitempty"`
	Init       string        `json:"init_time,omitempty"`
	Start      int64         `json:"start,omitempty"`
	End        int64         `json:"end,omitempty"`
	InitTime   time.Time     `json:"init_time_t,omitempty"`
	StartTime  time.Time     `json:"start_time_t,omitempty"`
	EndTime    time.Time     `json:"end_time_t,omitempty"`
	Time       time.Time     `json:"time_t,omitempty"`
	TimeUnit   time.Duration `json:"timeUnit,omitempty"`
	UPREnabled bool          `json:"upr_enabled"`
	TraceID    string        `json:"trace_id,omitempty"`
}

type TraceEvents []TraceEvent

type TraceOtherData struct {
	ID                    string                 `json:"run_id,omitempty"`
	ServerInfo            TraceServerInfo        `json:"server,omitempty"`
	EndToEndProcessTime   time.Duration          `json:"end_to_end_process_time,omitempty"`
	EndToEndTime          time.Duration          `json:"end_to_end_time,omitempty"`
	UPREnabled            bool                   `json:"upr_enabled"`
	UPRBaseDirectory      string                 `json:"UPR_BASE_DIR"`
	EagerMode             bool                   `json:"eager_mode"`
	EagerModeAsync        bool                   `json:"eager_mode_async"`
	EndAt                 string                 `json:"end_at"`
	Git                   GitInfo                `json:"git"`
	Hostname              string                 `json:"hostname"`
	IsClient              bool                   `json:"is_client"`
	ModelName             string                 `json:"model_name"`
	ModelParams           string                 `json:"model_params"`
	ModelPath             string                 `json:"model_path"`
	StartAt               string                 `json:"start_at"`
	SymbolParams          string                 `json:"symbol_params"`
	Username              string                 `json:"username"`
	MinEvent              TraceEvent             `json:"min_event"`
	MaxEvent              TraceEvent             `json:"max_event"`
	Iteration             int64                  `json:"iteration,omitempty"`
	Input                 map[string]interface{} `json:"input,omitempty"`
	ExperimentDescription string                 `json:"experiment_description,omitempty"`
	IsProcessed           bool                   `json:"is_processed,omitempty"`
}

type TraceServerInfo struct {
	ID               string    `json:"server_id,omitempty"`
	StartTime        time.Time `json:"server_start_time,omitempty"`
	EvictionPolicty  string    `json:"eviction_policy,omitempty"`
	EstimationRate   float32   `json:"estimation_rate,omitempty"`
	MemoryPercentage float32   `json:"memory_percentage,omitempty"`
	PersistCPU       bool      `json:"persist_cpu"`
}

type TraceSummary struct {
	ID                    string              `json:"run_id,omitempty"`
	ServerInfo            TraceServerInfo     `json:"server,omitempty"`
	EndToEndProcessTime   time.Duration       `json:"end_to_end_process_time,omitempty"`
	EndToEndTime          time.Duration       `json:"end_to_end_time,omitempty"`
	UPREnabled            bool                `json:"upr_enabled"`
	EagerMode             bool                `json:"eager_mode"`
	EagerModeAsync        bool                `json:"eager_mode_async"`
	Hostname              string              `json:"hostname"`
	ModelName             string              `json:"model_name"`
	ExperimentDescription string              `json:"experiment_description,omitempty"`
	InitTime              time.Time           `json:"init_time,omitempty"`
	Events                *TraceSummaryEvents `json:"events,omitempty"`
}

type TraceSummaryEvent struct {
	Name      string        `json:"name,omitempty"`
	Category  string        `json:"category,omitempty"`
	StartTime time.Time     `json:"start_time_t,omitempty"`
	Duration  time.Duration `json:"duration,omitempty"` // displayTimeUnit
}

type TraceSummaryEvents struct {
	CreatePrediction      *TraceSummaryEvent `json:"create_prediction,omitempty"`
	LoadModel             *TraceSummaryEvent `json:"load_model,omitempty"`
	GRPCOpen              *TraceSummaryEvent `json:"grpc_open,omitempty"`
	NDArrayConvert        *TraceSummaryEvent `json:"ndarray_convert,omitempty"`
	Predict               *TraceSummaryEvent `json:"predict,omitempty"`
	LoadParams            *TraceSummaryEvent `json:"load_params,omitempty"`
	ReadParams            *TraceSummaryEvent `json:"read_params,omitempty"`
	SetPredictInput       *TraceSummaryEvent `json:"set_predict_input,omitempty"`
	ForwardPredict        *TraceSummaryEvent `json:"forward_predict,omitempty"`
	GetPredictOutputShape *TraceSummaryEvent `json:"get_output_shape,omitempty"`
	GetPredictOutput      *TraceSummaryEvent `json:"get_output,omitempty"`
	FreePredictor         *TraceSummaryEvent `json:"free_predictor,omitempty"`
}