pkg/display/graph/graph.go

// Licensed to Apache Software Foundation (ASF) under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Apache Software Foundation (ASF) licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package graph

import (
	"fmt"
	"reflect"

	"github.com/apache/skywalking-cli/pkg/display/graph/flamegraph"

	api "skywalking.apache.org/repo/goapi/query"

	"github.com/urfave/cli/v2"

	d "github.com/apache/skywalking-cli/pkg/display/displayable"
	db "github.com/apache/skywalking-cli/pkg/display/graph/dashboard"
	"github.com/apache/skywalking-cli/pkg/display/graph/gauge"
	"github.com/apache/skywalking-cli/pkg/display/graph/heatmap"
	"github.com/apache/skywalking-cli/pkg/display/graph/linear"
	"github.com/apache/skywalking-cli/pkg/display/graph/tree"
	"github.com/apache/skywalking-cli/pkg/graphql/dashboard"
)

type (
	Thermodynamic          = api.HeatMap
	LinearMetrics          = map[string]*d.MetricValue
	MultiLinearMetrics     = map[string]LinearMetrics
	Trace                  = api.Trace
	TraceBrief             = api.TraceBrief
	GlobalMetrics          = [][]*api.SelectedRecord
	GlobalData             = dashboard.GlobalData
	EBPFProfilingAnalysis  = api.EBPFProfilingAnalyzation
	TraceProfilingAnalysis = api.ProfileAnalyzation
)

var (
	ThermodynamicType              = reflect.TypeOf(Thermodynamic{})
	LinearMetricsType              = reflect.TypeOf(LinearMetrics{})
	MultiLinearMetricsType         = reflect.TypeOf(MultiLinearMetrics{})
	TraceType                      = reflect.TypeOf(Trace{})
	TraceBriefType                 = reflect.TypeOf(TraceBrief{})
	GlobalMetricsType              = reflect.TypeOf(GlobalMetrics{})
	GlobalDataType                 = reflect.TypeOf(&GlobalData{})
	EBPFProfilingAnalysisDataType  = reflect.TypeOf(&EBPFProfilingAnalysis{})
	TraceProfilingAnalysisDataType = reflect.TypeOf(TraceProfilingAnalysis{})
)

func Display(ctx *cli.Context, displayable *d.Displayable) error {
	data := displayable.Data

	switch reflect.TypeOf(data) {
	case ThermodynamicType:
		return heatmap.Display(displayable)

	case LinearMetricsType:
		return linear.Display(ctx, map[string]LinearMetrics{"": data.(LinearMetrics)})

	case MultiLinearMetricsType:
		inputs := data.(MultiLinearMetrics)
		return linear.Display(ctx, inputs)

	case TraceType:
		return tree.Display(tree.Adapt(data.(Trace)))

	case TraceBriefType:
		return tree.DisplayList(ctx, displayable)

	case GlobalMetricsType:
		return gauge.Display(ctx, data.(GlobalMetrics))

	case GlobalDataType:
		return db.Display(ctx, data.(*GlobalData))

	case EBPFProfilingAnalysisDataType:
		return flamegraph.DisplayByEBPF(ctx, data.(*EBPFProfilingAnalysis))

	case TraceProfilingAnalysisDataType:
		return flamegraph.DisplayByTrace(ctx, data.(api.ProfileAnalyzation))

	default:
		return fmt.Errorf("type of %T is not supported to be displayed as ascii graph", reflect.TypeOf(data))
	}
}