STAC-22985: List mappings and show their status

Author: aacevedoosorio

cmd/otelcomponentmapping.go

@@ -0,0 +1,20 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+	"github.com/stackvista/stackstate-cli/cmd/otelcomponentmapping"
+	"github.com/stackvista/stackstate-cli/internal/di"
+)
+
+func OtelComponentMappingCommand(deps *di.Deps) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "otel-component-mapping",
+		Short: "Manage the Otel Component Mapping",
+		Long:  "Manage the Otel Component Mapping.",
+	}
+
+	cmd.AddCommand(otelcomponentmapping.OtelComponentMappingListCommand(deps))
+	cmd.AddCommand(otelcomponentmapping.OtelComponentMappingStatusCommand(deps))
+
+	return cmd
+}

cmd/otelcomponentmapping/otelcomponentmapping_list.go

@@ -0,0 +1,43 @@
+package otelcomponentmapping
+
+import (
+	"sort"
+
+	"github.com/spf13/cobra"
+	"github.com/stackvista/stackstate-cli/cmd/otelmapping"
+	"github.com/stackvista/stackstate-cli/generated/stackstate_api"
+	"github.com/stackvista/stackstate-cli/internal/common"
+	"github.com/stackvista/stackstate-cli/internal/di"
+)
+
+func OtelComponentMappingListCommand(deps *di.Deps) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "list",
+		Short: "Lists active Otel Component Mappings",
+		Long:  "Lists active Otel Component Mappings.",
+		RunE:  deps.CmdRunEWithApi(RunListComponentCommand),
+	}
+
+	return cmd
+}
+
+func RunListComponentCommand(cmd *cobra.Command, cli *di.Deps, api *stackstate_api.APIClient, serverInfo *stackstate_api.ServerInfo) common.CLIError {
+	mappingsList, resp, err := api.OtelMappingApi.GetOtelComponentMappings(cli.Context).Execute()
+	if err != nil {
+		return common.NewResponseError(err, resp)
+	}
+
+	sort.SliceStable(mappingsList, func(i, j int) bool {
+		return mappingsList[i].Name < mappingsList[j].Name
+	})
+
+	if cli.IsJson() {
+		cli.Printer.PrintJson(map[string]interface{}{
+			"otel component mappings": mappingsList,
+		})
+	} else {
+		cli.Printer.Table(otelmapping.FormatOtelMappingTable(mappingsList))
+	}
+
+	return nil
+}

cmd/otelcomponentmapping/otelcomponentmapping_list_test.go

@@ -0,0 +1,53 @@
+package otelcomponentmapping
+
+import (
+	"testing"
+
+	"github.com/stackvista/stackstate-cli/cmd/otelmapping"
+	"github.com/stackvista/stackstate-cli/internal/di"
+	"github.com/stackvista/stackstate-cli/internal/printer"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestListOtelComponentMappingsJson(t *testing.T) {
+	cli := di.NewMockDeps(t)
+	cmd := OtelComponentMappingListCommand(&cli.Deps)
+	cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingsResponse.Result = otelmapping.TestAllMappingItems
+
+	di.ExecuteCommandWithContextUnsafe(&cli.Deps, cmd, "-o", "json")
+
+	calls := *cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingsCalls
+	assert.Len(t, calls, 1)
+
+	expected := []map[string]interface{}{
+		{
+			"otel component mappings": otelmapping.TestAllMappingItems,
+		},
+	}
+
+	assert.Equal(t, expected, *cli.MockPrinter.PrintJsonCalls)
+}
+
+func TestOtelComponentMappingListTable(t *testing.T) {
+	cli := di.NewMockDeps(t)
+	cmd := OtelComponentMappingListCommand(&cli.Deps)
+	cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingsResponse.Result = otelmapping.TestAllMappingItems
+
+	di.ExecuteCommandWithContextUnsafe(&cli.Deps, cmd)
+
+	calls := *cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingsCalls
+	assert.Len(t, calls, 1)
+
+	expectedTableCall := []printer.TableData{
+		{
+			Header: []string{"Name", "Identifier"},
+			Data: [][]interface{}{
+				{otelmapping.TestSomeOtelMappingItem.Name, "identifier"},
+				{otelmapping.TestSomeOtelMappingItem2.Name, "identifier2"},
+			},
+			MissingTableDataMsg: printer.NotFoundMsg{Types: "otel mappings"},
+		},
+	}
+
+	assert.Equal(t, expectedTableCall, *cli.MockPrinter.TableCalls)
+}

cmd/otelcomponentmapping/otelcomponentmapping_status.go

@@ -0,0 +1,23 @@
+package otelcomponentmapping
+
+import (
+	"github.com/spf13/cobra"
+	"github.com/stackvista/stackstate-cli/cmd/otelmapping"
+	"github.com/stackvista/stackstate-cli/internal/common"
+	"github.com/stackvista/stackstate-cli/internal/di"
+)
+
+// if cli.IsJson() {
+// cli.Printer.PrintJson(
+
+func OtelComponentMappingStatusCommand(deps *di.Deps) *cobra.Command {
+	args := &otelmapping.StatusArgs{}
+	cmd := &cobra.Command{
+		Use:   "status",
+		Short: "Get the status of an Otel Component Mappings",
+		Long:  "Get the status of an Otel Component Mappings.",
+		RunE:  deps.CmdRunEWithApi(otelmapping.RunStatus(args, "component", otelmapping.FetchComponentStatus)),
+	}
+	common.AddRequiredIdentifierFlagVar(cmd, &args.Identifier, "Identifier of the Otel Component Mapping")
+	return cmd
+}

cmd/otelcomponentmapping/otelcomponentmapping_status_test.go

@@ -0,0 +1,67 @@
+package otelcomponentmapping
+
+import (
+	"testing"
+
+	"github.com/stackvista/stackstate-cli/cmd/otelmapping"
+	"github.com/stackvista/stackstate-cli/internal/di"
+	"github.com/stackvista/stackstate-cli/internal/printer"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestOtelComponentMappingStatusJson(t *testing.T) {
+	cli := di.NewMockDeps(t)
+	cmd := OtelComponentMappingStatusCommand(&cli.Deps)
+	cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingStatusResponse.Result = *otelmapping.TestSomeOtelMappingStatus
+
+	di.ExecuteCommandWithContextUnsafe(&cli.Deps, cmd, "--identifier", "identifier", "-o", "json")
+
+	calls := *cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingStatusCalls
+	assert.Len(t, calls, 1)
+
+	expected := []map[string]interface{}{
+		{
+			"otel-component-mapping": otelmapping.TestSomeOtelMappingStatus,
+		},
+	}
+
+	assert.Equal(t, expected, *cli.MockPrinter.PrintJsonCalls)
+}
+
+func TestOtelComponentMappingStatusTable(t *testing.T) {
+	cli := di.NewMockDeps(t)
+	cmd := OtelComponentMappingStatusCommand(&cli.Deps)
+	cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingStatusResponse.Result = *otelmapping.TestSomeOtelMappingStatus
+
+	di.ExecuteCommandWithContextUnsafe(&cli.Deps, cmd, "--identifier", "identifier")
+
+	calls := *cli.MockClient.ApiMocks.OtelMappingApi.GetOtelComponentMappingStatusCalls
+	assert.Len(t, calls, 1)
+
+	expectedTableCall := []printer.TableData{
+		{
+			Header: []string{"Name", "Identifier", "Components", "Relations"},
+			Data: [][]interface{}{
+				{otelmapping.TestSomeOtelMappingStatusItem.Name, "identifier", otelmapping.TestComponentCount, otelmapping.TestRelationCount},
+			},
+			MissingTableDataMsg: printer.NotFoundMsg{Types: "otel mappings"},
+		},
+		{
+			Header: []string{"Metric", "10s ago", "10-20s ago", "20-30s ago"},
+			Data: [][]interface{}{
+				{"latency seconds", otelmapping.TestMetricValue, otelmapping.TestMetricValue, otelmapping.TestMetricValue},
+			},
+			MissingTableDataMsg: printer.NotFoundMsg{Types: "metrics"},
+		},
+		{
+			Header: []string{"Issue Id", "Level", "Message"},
+			Data: [][]interface{}{
+				{"-", otelmapping.TestError1.Level, otelmapping.TestError1.Message},
+				{"-", otelmapping.TestError2.Level, otelmapping.TestError2.Message},
+			},
+			MissingTableDataMsg: printer.NotFoundMsg{Types: "otel component mapping errors"},
+		},
+	}
+
+	assert.Equal(t, expectedTableCall, *cli.MockPrinter.TableCalls)
+}

cmd/otelmapping/otelmapping.go

@@ -0,0 +1,130 @@
+package otelmapping
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/spf13/cobra"
+	"github.com/stackvista/stackstate-cli/generated/stackstate_api"
+	"github.com/stackvista/stackstate-cli/internal/common"
+	"github.com/stackvista/stackstate-cli/internal/di"
+	"github.com/stackvista/stackstate-cli/internal/printer"
+	"golang.org/x/text/cases"
+	"golang.org/x/text/language"
+)
+
+type StatusArgs struct {
+	Identifier string
+}
+
+func FormatOtelMappingStatusTable(otelmappings []stackstate_api.OtelMappingStatusItem) printer.TableData {
+	data := make([][]interface{}, len(otelmappings))
+
+	for i, otelmapping := range otelmappings {
+		identifier := "-"
+		if otelmapping.HasIdentifier() {
+			identifier = otelmapping.GetIdentifier()
+		}
+		data[i] = []interface{}{
+			otelmapping.Name,
+			identifier,
+			otelmapping.ComponentCount,
+			otelmapping.RelationCount,
+		}
+	}
+	return printer.TableData{
+		Header:              []string{"Name", "Identifier", "Components", "Relations"},
+		Data:                data,
+		MissingTableDataMsg: printer.NotFoundMsg{Types: "otel mappings"},
+	}
+}
+
+func FormatOtelMappingTable(otelmappings []stackstate_api.OtelMappingItem) printer.TableData {
+	data := make([][]interface{}, len(otelmappings))
+
+	for i, otelmapping := range otelmappings {
+		identifier := "-"
+		if otelmapping.HasIdentifier() {
+			identifier = otelmapping.GetIdentifier()
+		}
+		data[i] = []interface{}{
+			otelmapping.Name,
+			identifier,
+		}
+	}
+	return printer.TableData{
+		Header:              []string{"Name", "Identifier"},
+		Data:                data,
+		MissingTableDataMsg: printer.NotFoundMsg{Types: "otel mappings"},
+	}
+}
+
+type StatusFetcher func(cli *di.Deps, api *stackstate_api.APIClient, identifier string) (*stackstate_api.OtelMappingStatus, *http.Response, error)
+
+func FetchComponentStatus(cli *di.Deps, api *stackstate_api.APIClient, identifier string) (*stackstate_api.OtelMappingStatus, *http.Response, error) {
+	return api.OtelMappingApi.GetOtelComponentMappingStatus(cli.Context, identifier).Execute()
+}
+
+func FetchRelationStatus(cli *di.Deps, api *stackstate_api.APIClient, identifier string) (*stackstate_api.OtelMappingStatus, *http.Response, error) {
+	return api.OtelMappingApi.GetOtelRelationMappingStatus(cli.Context, identifier).Execute()
+}
+
+func RunStatus(args *StatusArgs, mappingType string, fetch StatusFetcher) di.CmdWithApiFn {
+	return func(cmd *cobra.Command, cli *di.Deps, api *stackstate_api.APIClient, serverInfo *stackstate_api.ServerInfo) common.CLIError {
+		mappingStatus, resp, err := fetch(cli, api, args.Identifier)
+		if err != nil {
+			return common.NewResponseError(err, resp)
+		}
+
+		jsonKey := fmt.Sprintf("otel-%s-mapping", mappingType)
+		title := cases.Title(language.English).String(mappingType)
+		mappingTitle := fmt.Sprintf("Otel %s Mapping:", title)
+		metricsTitle := fmt.Sprintf("Otel %s Mapping Metrics:", title)
+		errorsTitle := fmt.Sprintf("Otel %s Mapping Errors:", title)
+		errorsNotFoundMsg := fmt.Sprintf("otel %s mapping errors", mappingType)
+
+		if cli.IsJson() {
+			cli.Printer.PrintJson(map[string]interface{}{
+				jsonKey: mappingStatus,
+			})
+		} else {
+			cli.Printer.PrintLn("\n")
+			cli.Printer.PrintLn(mappingTitle)
+			cli.Printer.Table(FormatOtelMappingStatusTable([]stackstate_api.OtelMappingStatusItem{
+				mappingStatus.Item,
+			}))
+
+			if mappingStatus.HasMetrics() {
+				cli.Printer.PrintLn("\n")
+				cli.Printer.PrintLn(metricsTitle)
+				size := mappingStatus.Metrics.BucketSizeSeconds
+				cli.Printer.Table(printer.TableData{
+					Header: []string{"Metric", fmt.Sprintf("%ds ago", size), fmt.Sprintf("%d-%ds ago", size, 2*size), fmt.Sprintf("%d-%ds ago", 2*size, 3*size)}, //nolint:mnd
+					Data: [][]interface{}{
+						printer.MetricBucketToRow("latency seconds", mappingStatus.Metrics.LatencySeconds),
+					},
+					MissingTableDataMsg: printer.NotFoundMsg{Types: "metrics"},
+				})
+			}
+
+			data := make([][]interface{}, len(mappingStatus.ErrorDetails))
+			for i, error := range mappingStatus.ErrorDetails {
+				id := "-"
+				if error.HasIssueId() {
+					id = *error.IssueId
+				}
+				data[i] = []interface{}{id, error.Level, error.Message}
+			}
+
+			cli.Printer.PrintLn("\n")
+			cli.Printer.PrintLn(errorsTitle)
+			cli.Printer.Table(printer.TableData{
+				Header:              []string{"Issue Id", "Level", "Message"},
+				Data:                data,
+				MissingTableDataMsg: printer.NotFoundMsg{Types: errorsNotFoundMsg},
+			})
+		}
+
+		return nil
+	}
+}

cmd/otelmapping/test_data.go

@@ -0,0 +1,53 @@
+package otelmapping
+
+import (
+	"github.com/stackvista/stackstate-cli/generated/stackstate_api"
+)
+
+const (
+	TestMetricValue    = float64(23.5)
+	TestComponentCount = int64(200)
+	TestRelationCount  = int64(100)
+)
+
+var (
+	TestError1     = stackstate_api.NewOtelMappingError(stackstate_api.MESSAGELEVEL_WARN, "Oops")
+	TestError2     = stackstate_api.NewOtelMappingError(stackstate_api.MESSAGELEVEL_ERROR, "Oops2")
+	TestSomeErrors = []stackstate_api.OtelMappingError{
+		*TestError1,
+		*TestError2,
+	}
+	TestMetricBucketValue = (func() *stackstate_api.MetricBucketValue {
+		v := stackstate_api.NewMetricBucketValue()
+		v.SetValue(float64(TestMetricValue))
+		return v
+	})()
+	TestSomeMetrics = stackstate_api.OtelMappingMetrics{
+		BucketSizeSeconds: 10,
+		LatencySeconds:    []stackstate_api.MetricBucketValue{*TestMetricBucketValue, *TestMetricBucketValue, *TestMetricBucketValue},
+	}
+	TestSomeOtelMappingStatusItem = (func() *stackstate_api.OtelMappingStatusItem {
+		v := stackstate_api.NewOtelMappingStatusItem("name", TestRelationCount, TestComponentCount)
+		v.SetIdentifier("identifier")
+		return v
+	})()
+	TestSomeOtelMappingStatus = (func() *stackstate_api.OtelMappingStatus {
+		v := stackstate_api.NewOtelMappingStatus(*TestSomeOtelMappingStatusItem, TestSomeErrors)
+		v.SetMetrics(TestSomeMetrics)
+		return v
+	})()
+)
+
+var (
+	TestSomeOtelMappingItem = (func() *stackstate_api.OtelMappingItem {
+		v := stackstate_api.NewOtelMappingItem("name")
+		v.SetIdentifier("identifier")
+		return v
+	})()
+	TestSomeOtelMappingItem2 = (func() *stackstate_api.OtelMappingItem {
+		v := stackstate_api.NewOtelMappingItem("name2")
+		v.SetIdentifier("identifier2")
+		return v
+	})()
+	TestAllMappingItems = []stackstate_api.OtelMappingItem{*TestSomeOtelMappingItem, *TestSomeOtelMappingItem2}
+)