pyroscope

type Annotation struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

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

type Exemplar struct {
	Timestamp json.Number `json:"timestamp"` // ms since epoch, encoded as string
	ProfileID string      `json:"profileId,omitempty"`
	SpanID    string      `json:"spanId,omitempty"`
	Value     json.Number `json:"value"`
	Labels    []LabelPair `json:"labels,omitempty"`
}

type Flamegraph struct {
	Names   []string `json:"names"`
	Levels  []Level  `json:"levels"`
	Total   int64    `json:"total,string"`
	MaxSelf int64    `json:"maxSelf,string"`
}

type FunctionSample struct {
	Name       string
	Self       int64
	Total      int64
	Percentage float64
}

type HeatmapSeries struct {
	Labels []LabelPair   `json:"labels"`
	Slots  []HeatmapSlot `json:"slots"`
}

type HeatmapSlot struct {
	Timestamp json.Number `json:"timestamp"` // right edge of time bucket, ms since epoch
	Exemplars []Exemplar  `json:"exemplars,omitempty"`
}

type LabelNamesRequest struct {
	Matchers []string
	Start    time.Time
	End      time.Time
}

type LabelNamesResponse struct {
	Names []string `json:"names"`
}

type LabelPair struct {
	Name  string `json:"name"`
	Value string `json:"value"`
}

type LabelValuesRequest struct {
	Name     string
	Matchers []string
	Start    time.Time
	End      time.Time
}

type LabelValuesResponse struct {
	Names []string `json:"names"` // Pyroscope uses "names" for both labels and values
}

type Level struct {
	Values []string `json:"values"` // API returns strings that need to be parsed
}

type Location struct {
	Name string `json:"name"`
}

type PprofRequest struct {
	ProfileTypeID string
	LabelSelector string
	Start         time.Time
	End           time.Time
	MaxNodes      int64
}

type PprofWriteResult struct {
	Path string `json:"path"`
}

type ProfileExemplar struct {
	ProfileID string            `json:"profileId"`
	Timestamp time.Time         `json:"timestamp"`
	Value     int64             `json:"value"`
	SpanID    string            `json:"spanId,omitempty"`
	Labels    map[string]string `json:"labels,omitempty"`
}

type ProfileExemplarsResult struct {
	From        time.Time         `json:"from"`
	To          time.Time         `json:"to"`
	ProfileType string            `json:"profileType"`
	Exemplars   []ProfileExemplar `json:"exemplars"`
}

type ProfileType struct {
	ID         string `json:"ID"`
	Name       string `json:"name"`
	SampleType string `json:"sampleType"`
	SampleUnit string `json:"sampleUnit"`
	PeriodType string `json:"periodType"`
	PeriodUnit string `json:"periodUnit"`
}

type ProfileTypesRequest struct {
	Start time.Time
	End   time.Time
}

type ProfileTypesResponse struct {
	ProfileTypes []ProfileType `json:"profileTypes"`
}

type QueryRequest struct {
	LabelSelector      string
	ProfileTypeID      string
	Start              time.Time
	End                time.Time
	MaxNodes           int64
	ProfileIDs         []string
	StackTraceSelector *StackTraceSelector
}

type QueryResponse struct {
	Flamegraph *Flamegraph `json:"flamegraph,omitempty"`
}

type SelectHeatmapRequest struct {
	ProfileTypeID string
	LabelSelector string
	Start         time.Time
	End           time.Time
	Step          float64
	QueryType     string // e.g. "HEATMAP_QUERY_TYPE_SPAN"
	ExemplarType  string // e.g. "EXEMPLAR_TYPE_SPAN"
	Limit         int64  // optional; max number of exemplars per slot
}

type SelectHeatmapResponse struct {
	Series []HeatmapSeries `json:"series"`
}

type SelectSeriesRequest struct {
	ProfileTypeID string
	LabelSelector string
	Start         time.Time
	End           time.Time
	GroupBy       []string
	Step          float64 // resolution step in seconds
	Aggregation   string  // "SUM" or "AVERAGE"
	Limit         int64   // top-N series by total value
	ExemplarType  string  // "EXEMPLAR_TYPE_INDIVIDUAL" or "EXEMPLAR_TYPE_SPAN"; empty = no exemplars
}

type SelectSeriesResponse struct {
	Series []TimeSeries `json:"series"`
}

type SpanExemplar struct {
	SpanID    string            `json:"spanId"`
	Timestamp time.Time         `json:"timestamp"`
	Value     int64             `json:"value"`
	Labels    map[string]string `json:"labels,omitempty"`
}

type SpanExemplarsResult struct {
	From        time.Time      `json:"from"`
	To          time.Time      `json:"to"`
	ProfileType string         `json:"profileType"`
	Exemplars   []SpanExemplar `json:"exemplars"`
}

type StackTraceSelector struct {
	CallSite []Location `json:"callSite,omitempty"`
}

type TimePoint struct {
	Value       json.Number  `json:"value"`
	Timestamp   json.Number  `json:"timestamp"` // milliseconds since epoch, encoded as string
	Annotations []Annotation `json:"annotations,omitempty"`
	Exemplars   []Exemplar   `json:"exemplars,omitempty"`
}

type TimeSeries struct {
	Labels []LabelPair `json:"labels"`
	Points []TimePoint `json:"points"`
}

type TopSeriesEntry struct {
	Rank   int               `json:"rank"`
	Labels map[string]string `json:"labels"`
	Total  float64           `json:"total"`
}

type TopSeriesResponse struct {
	ProfileType string           `json:"profileType"`
	GroupBy     []string         `json:"groupBy"`
	Series      []TopSeriesEntry `json:"series"`
}