feat: Create connection string helper function and update telemetry handle

Author: BeegiiK

aitelemetry/connection_string_parser.go

@@ -0,0 +1,45 @@
+package aitelemetry
+
+import (
+	"fmt"
+	"strings"
+)
+
+type connectionVars struct {
+	InstrumentationKey string
+	IngestionUrl       string
+}
+
+func parseConnectionString(connectionString string) (*connectionVars, error) {
+	connectionVars := &connectionVars{}
+
+	if connectionString == "" {
+		return nil, fmt.Errorf("Connection string cannot be empty")
+	}
+
+	pairs := strings.Split(connectionString, ";")
+	for _, pair := range pairs {
+		kv := strings.SplitN(pair, "=", 2)
+		if len(kv) != 2 {
+			return nil, fmt.Errorf("Invalid connection string format: %s", pair)
+		}
+		key, value := strings.TrimSpace(kv[0]), strings.TrimSpace(kv[1])
+
+		if key == "" {
+			return nil, fmt.Errorf("Key in connection string cannot be empty")
+		}
+
+		switch strings.ToLower(key) {
+		case "instrumentationkey":
+			connectionVars.InstrumentationKey = value
+		case "ingestionendpoint":
+			connectionVars.IngestionUrl = value + "v2.1/track"
+		}
+	}
+
+	if connectionVars.InstrumentationKey == "" || connectionVars.IngestionUrl == "" {
+		return nil, fmt.Errorf("Missing required fields in connection string")
+	}
+
+	return connectionVars, nil
+}

aitelemetry/connection_string_parser_test.go

@@ -0,0 +1,66 @@
+package aitelemetry
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+const connectionString = "InstrumentationKey=00000000-0000-0000-0000-000000000000;IngestionEndpoint=https://ingestion.endpoint.com/;LiveEndpoint=https://live.endpoint.com/;ApplicationId=11111111-1111-1111-1111-111111111111"
+
+func TestParseConnectionString(t *testing.T) {
+	tests := []struct {
+		name             string
+		connectionString string
+		want             *connectionVars
+		wantErr          bool
+	}{
+		{
+			name:             "Valid connection string and instrumentation key",
+			connectionString: connectionString,
+			want: &connectionVars{
+				InstrumentationKey: "00000000-0000-0000-0000-000000000000",
+				IngestionUrl:       "https://ingestion.endpoint.com/v2.1/track",
+			},
+			wantErr: false,
+		},
+		{
+			name:             "Invalid connection string format",
+			connectionString: "InvalidConnectionString",
+			want:             nil,
+			wantErr:          true,
+		},
+		{
+			name:             "Valid instrumentation key with missing ingestion endpoint",
+			connectionString: "InstrumentationKey=00000000-0000-0000-0000-000000000000;IngestionEndpoint=;",
+			want:             nil,
+			wantErr:          true,
+		},
+		{
+			name:             "Missing instrumentation key with valid ingestion endpoint",
+			connectionString: "InstrumentationKey=;IngestionEndpoint=https://ingestion.endpoint.com/;LiveEndpoint=https://live.endpoint.com/;",
+			want:             nil,
+			wantErr:          true,
+		},
+		{
+			name:             "Empty connection string",
+			connectionString: "",
+			want:             nil,
+			wantErr:          true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := parseConnectionString(tt.connectionString)
+			if tt.wantErr {
+				require.Error(t, err, "Expected error but got none")
+			} else {
+				require.NoError(t, err, "Expected no error but got one")
+				require.NotNil(t, got, "Expected a non-nil result")
+				require.Equal(t, tt.want.InstrumentationKey, got.InstrumentationKey, "Instrumentation Key does not match")
+				require.Equal(t, tt.want.IngestionUrl, got.IngestionUrl, "Ingestion URL does not match")
+			}
+		})
+	}
+}

aitelemetry/telemetrywrapper.go

@@ -228,7 +228,14 @@ func NewAITelemetryWithConnectionString(
 
 	setAIConfigDefaults(&aiConfig)
 
-	telemetryConfig := appinsights.NewTelemetryConfigurationWithConnectionString(cString)
+	connectionVars, err := parseConnectionString(cString)
+	if err != nil {
+		debugLog("Error parsing connection string: %v", err)
+		return nil, err
+	}
+
+	telemetryConfig := appinsights.NewTelemetryConfiguration(connectionVars.InstrumentationKey)
+	telemetryConfig.EndpointUrl = connectionVars.IngestionUrl
 	telemetryConfig.MaxBatchSize = aiConfig.BatchSize
 	telemetryConfig.MaxBatchInterval = time.Duration(aiConfig.BatchInterval) * time.Second
 

aitelemetry/telemetrywrapper_test.go

@@ -19,7 +19,7 @@ var (
 	hostAgentUrl     = "localhost:3501"
 	getCloudResponse = "AzurePublicCloud"
 	httpURL          = "http://" + hostAgentUrl
-	connectionString = "InstrumentationKey=00000000-0000-0000-0000-000000000000;IngestionEndpoint=https://ingestion.endpoint.com/;LiveEndpoint=https://live.endpoint.com/;ApplicationId=11111111-1111-1111-1111-111111111111"
+	// connectionString = "InstrumentationKey=00000000-0000-0000-0000-000000000000;IngestionEndpoint=https://ingestion.endpoint.com/;LiveEndpoint=https://live.endpoint.com/;ApplicationId=11111111-1111-1111-1111-111111111111"
 )
 
 func TestMain(m *testing.M) {