[v2] Implement GetOperations and GetServices for memory backend (jaegertracing#7053)

## Which problem is this PR solving?
- Fixes a part of: jaegertracing#7052 

## Description of the changes
- Implement GetOperations and GetServices for memory backend

## How was this change tested?
- Unit tests

## Checklist
- [x] I have read
https://github.com/jaegertracing/jaeger/blob/master/CONTRIBUTING_GUIDELINES.md
- [x] I have signed all commits
- [x] I have added unit tests for the new functionality
- [x] I have run lint and test steps successfully
  - for `jaeger`: `make lint test`
  - for `jaeger-ui`: `npm run lint` and `npm run test`

Signed-off-by: Manik2708 <[email protected]>

Author: Manik2708

internal/storage/v2/memory/fixtures/db_traces_01.json

@@ -23,7 +23,7 @@
               "spanId": "0000000000000002",
               "parentSpanId": "0000000000000003",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-2",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [
@@ -120,7 +120,7 @@
               "spanId": "0000000000000004",
               "parentSpanId": "0000000000000011",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-4",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [

internal/storage/v2/memory/fixtures/db_traces_02.json

@@ -23,7 +23,7 @@
               "spanId": "0000000000000003",
               "parentSpanId": "0000000000000010",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-3",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [
@@ -120,7 +120,7 @@
               "spanId": "0000000000000005",
               "parentSpanId": "0000000000000010",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-5",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [

internal/storage/v2/memory/fixtures/otel_traces_01.json

@@ -23,7 +23,7 @@
               "spanId": "0000000000000002",
               "parentSpanId": "0000000000000003",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-2",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [
@@ -112,7 +112,7 @@
               "spanId": "0000000000000003",
               "parentSpanId": "0000000000000010",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-3",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [
@@ -209,7 +209,7 @@
               "spanId": "0000000000000004",
               "parentSpanId": "0000000000000011",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-4",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [
@@ -298,7 +298,7 @@
               "spanId": "0000000000000005",
               "parentSpanId": "0000000000000010",
               "flags": 1,
-              "name": "test-general-conversion",
+              "name": "test-general-conversion-5",
               "startTimeUnixNano": "1485467191639875000",
               "endTimeUnixNano": "1485467191639880000",
               "attributes": [

internal/storage/v2/memory/memory.go

@@ -77,6 +77,34 @@ func (st *Store) WriteTraces(ctx context.Context, td ptrace.Traces) error {
 	return nil
 }
 
+// GetOperations returns operations based on the service name and span kind
+func (st *Store) GetOperations(ctx context.Context, query tracestore.OperationQueryParams) ([]tracestore.Operation, error) {
+	m := st.getTenant(tenancy.GetTenant(ctx))
+	m.RLock()
+	defer m.RUnlock()
+	var retMe []tracestore.Operation
+	if operations, ok := m.operations[query.ServiceName]; ok {
+		for operation := range operations {
+			if query.SpanKind == "" || query.SpanKind == operation.SpanKind {
+				retMe = append(retMe, operation)
+			}
+		}
+	}
+	return retMe, nil
+}
+
+// GetServices returns a list of all known services
+func (st *Store) GetServices(ctx context.Context) ([]string, error) {
+	m := st.getTenant(tenancy.GetTenant(ctx))
+	m.RLock()
+	defer m.RUnlock()
+	var retMe []string
+	for k := range m.services {
+		retMe = append(retMe, k)
+	}
+	return retMe, nil
+}
+
 // reshuffleResourceSpans reshuffles the resource spans so as to group the spans from same traces together. To understand this reshuffling
 // take an example of 2 resource spans, then these two resource spans have 2 scope spans each.
 // Every scope span consists of 2 spans with trace ids: 1 and 2. Now the final structure should look like:

internal/storage/v2/memory/memory_test.go

@@ -10,14 +10,17 @@ import (
 	"encoding/json"
 	"fmt"
 	"os"
+	"sort"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"go.opentelemetry.io/collector/pdata/pcommon"
 	"go.opentelemetry.io/collector/pdata/ptrace"
+	conventions "go.opentelemetry.io/collector/semconv/v1.16.0"
 
 	v1 "github.com/jaegertracing/jaeger/internal/storage/v1/memory"
+	"github.com/jaegertracing/jaeger/internal/storage/v2/api/tracestore"
 	"github.com/jaegertracing/jaeger/internal/tenancy"
 )
 
@@ -37,6 +40,58 @@ func TestNewStore_DefaultConfig(t *testing.T) {
 	require.True(t, ok)
 	expected2 := loadOutputTraces(t, 2)
 	testTraces(t, expected2, traces2)
+	operations, err := store.GetOperations(context.Background(), tracestore.OperationQueryParams{ServiceName: "service-x"})
+	require.NoError(t, err)
+	expectedOperations := []tracestore.Operation{
+		{
+			Name:     "test-general-conversion-2",
+			SpanKind: "Unspecified",
+		},
+		{
+			Name:     "test-general-conversion-3",
+			SpanKind: "Unspecified",
+		},
+		{
+			Name:     "test-general-conversion-4",
+			SpanKind: "Unspecified",
+		},
+		{
+			Name:     "test-general-conversion-5",
+			SpanKind: "Unspecified",
+		},
+	}
+	sort.Slice(operations, func(i, j int) bool {
+		return operations[i].Name < operations[j].Name
+	})
+	assert.Equal(t, expectedOperations, operations)
+	expectedServices := []string{"service-x"}
+	services, err := store.GetServices(context.Background())
+	require.NoError(t, err)
+	assert.Equal(t, expectedServices, services)
+}
+
+func TestGetOperationsWithKind(t *testing.T) {
+	store := NewStore(v1.Configuration{})
+	td := ptrace.NewTraces()
+	resourceSpans := td.ResourceSpans().AppendEmpty()
+	attrs := resourceSpans.Resource().Attributes()
+	attrs.PutStr(conventions.AttributeServiceName, "service-x")
+	span1 := resourceSpans.ScopeSpans().AppendEmpty().Spans().AppendEmpty()
+	span1.SetTraceID(fromString(t, "00000000000000010000000000000000"))
+	span1.SetKind(ptrace.SpanKindConsumer)
+	span1.SetName("operation-with-kind")
+	span2 := resourceSpans.ScopeSpans().AppendEmpty().Spans().AppendEmpty()
+	span2.SetTraceID(fromString(t, "00000000000000010000000000000000"))
+	err := store.WriteTraces(context.Background(), td)
+	require.NoError(t, err)
+	operations, err := store.GetOperations(context.Background(), tracestore.OperationQueryParams{
+		ServiceName: "service-x",
+		SpanKind:    ptrace.SpanKindConsumer.String(),
+	})
+	assert.Len(t, operations, 1)
+	assert.Equal(t, "operation-with-kind", operations[0].Name)
+	assert.Equal(t, operations[0].SpanKind, ptrace.SpanKindConsumer.String())
+	require.NoError(t, err)
 }
 
 func TestWriteTraces_WriteTwoBatches(t *testing.T) {