[extension/solarwindsapmsettingsextension] Added remaining implementation of solarwindsapmsettingsextension

.chloggen/feature_solarwindsapmsettingsetension-impl-2.yaml

@@ -0,0 +1,27 @@
+# Use this changelog template to create an entry for release notes.
+
+# One of 'breaking', 'deprecation', 'new_component', 'enhancement', 'bug_fix'
+change_type: enhancement
+
+# The name of the component, or a single word describing the area of concern, (e.g. filelogreceiver)
+component: solarwindsapmsettingsextension
+
+# A brief description of the change.  Surround your text with quotes ("") if it needs to start with a backtick (`).
+note: Added logic for refresh function
+
+# Mandatory: One or more tracking issues related to the change. You can use the PR number here if no issue exists.
+issues: [27668]
+
+# (Optional) One or more lines of additional information to render under the primary note.
+# These lines will be padded with 2 spaces and then inserted directly into the document.
+# Use pipe (|) for multiline entries.
+subtext:
+
+# If your change doesn't affect end users or the exported elements of any package,
+# you should instead start your pull request title with [chore] or use the "Skip Changelog" label.
+# Optional: The change log or logs in which this entry should be included.
+# e.g. '[user]' or '[user, api]'
+# Include 'user' if the change is relevant to end users.
+# Include 'api' if there is a change to a library API.
+# Default: '[user]'
+change_logs: []

extension/solarwindsapmsettingsextension/README.md

@@ -12,7 +12,7 @@
 <!-- end autogenerated section -->
 
 ## Overview
-The Solarwinds APM Settings extension gets Solarwinds APM specific settings from Solarwinds APM collector and `/tmp/solarwinds-apm-settings.json` & `/tmp/solarwinds-apm-settings-raw` periodically.
+The Solarwinds APM Settings extension gets Solarwinds APM specific settings from Solarwinds APM collector and `/tmp/solarwinds-apm-settings.json` periodically.
 
 ## Configuration
 

extension/solarwindsapmsettingsextension/extension.go

@@ -6,6 +6,11 @@ package solarwindsapmsettingsextension // import "github.com/open-telemetry/open
 import (
 	"context"
 	"crypto/tls"
+	"encoding/binary"
+	"encoding/json"
+	"math"
+	"os"
+	"strconv"
 	"time"
 
 	"github.com/solarwindscloud/apm-proto/go/collectorpb"
@@ -14,6 +19,12 @@ import (
 	"go.uber.org/zap"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
+	"google.golang.org/protobuf/encoding/protojson"
+)
+
+const (
+	JSONOutputFile      = "/tmp/solarwinds-apm-settings.json"
+	GrpcContextDeadline = time.Duration(1) * time.Second
 )
 
 type solarwindsapmSettingsExtension struct {
@@ -76,6 +87,131 @@ func (extension *solarwindsapmSettingsExtension) Shutdown(_ context.Context) err
 }
 
 func refresh(extension *solarwindsapmSettingsExtension) {
-	// Concrete implementation will be available in later PR
-	extension.logger.Info("refresh task")
+	extension.logger.Info("Time to refresh from " + extension.config.Endpoint)
+	if hostname, err := os.Hostname(); err != nil {
+		extension.logger.Error("Unable to call os.Hostname() " + err.Error())
+	} else {
+		ctx, cancel := context.WithTimeout(context.Background(), GrpcContextDeadline)
+		defer cancel()
+
+		request := &collectorpb.SettingsRequest{
+			ApiKey: extension.config.Key,
+			Identity: &collectorpb.HostID{
+				Hostname: hostname,
+			},
+			ClientVersion: "2",
+		}
+		if response, err := extension.client.GetSettings(ctx, request); err != nil {
+			extension.logger.Error("Unable to getSettings from " + extension.config.Endpoint + " " + err.Error())
+		} else {
+			switch result := response.GetResult(); result {
+			case collectorpb.ResultCode_OK:
+				if len(response.GetWarning()) > 0 {
+					extension.logger.Warn(response.GetWarning())
+				}
+				var settings []map[string]any
+				for _, item := range response.GetSettings() {
+					marshalOptions := protojson.MarshalOptions{
+						UseEnumNumbers:  true,
+						EmitUnpopulated: true,
+					}
+					if settingBytes, err := marshalOptions.Marshal(item); err != nil {
+						extension.logger.Warn("Error to marshal setting JSON[] byte from response.GetSettings() " + err.Error())
+					} else {
+						setting := make(map[string]any)
+						if err := json.Unmarshal(settingBytes, &setting); err != nil {
+							extension.logger.Warn("Error to unmarshal setting JSON object from setting JSON[]byte " + err.Error())
+						} else {
+							if value, ok := setting["value"].(string); ok {
+								if num, err := strconv.ParseInt(value, 10, 0); err != nil {
+									extension.logger.Warn("Unable to parse value " + value + " as number " + err.Error())
+								} else {
+									setting["value"] = num
+								}
+							}
+							if timestamp, ok := setting["timestamp"].(string); ok {
+								if num, err := strconv.ParseInt(timestamp, 10, 0); err != nil {
+									extension.logger.Warn("Unable to parse timestamp " + timestamp + " as number " + err.Error())
+								} else {
+									setting["timestamp"] = num
+								}
+							}
+							if ttl, ok := setting["ttl"].(string); ok {
+								if num, err := strconv.ParseInt(ttl, 10, 0); err != nil {
+									extension.logger.Warn("Unable to parse ttl " + ttl + " as number " + err.Error())
+								} else {
+									setting["ttl"] = num
+								}
+							}
+							if _, ok := setting["flags"]; ok {
+								setting["flags"] = string(item.Flags)
+							}
+							if arguments, ok := setting["arguments"].(map[string]any); ok {
+								if value, ok := item.Arguments["BucketCapacity"]; ok {
+									arguments["BucketCapacity"] = math.Float64frombits(binary.LittleEndian.Uint64(value))
+								}
+								if value, ok := item.Arguments["BucketRate"]; ok {
+									arguments["BucketRate"] = math.Float64frombits(binary.LittleEndian.Uint64(value))
+								}
+								if value, ok := item.Arguments["TriggerRelaxedBucketCapacity"]; ok {
+									arguments["TriggerRelaxedBucketCapacity"] = math.Float64frombits(binary.LittleEndian.Uint64(value))
+								}
+								if value, ok := item.Arguments["TriggerRelaxedBucketRate"]; ok {
+									arguments["TriggerRelaxedBucketRate"] = math.Float64frombits(binary.LittleEndian.Uint64(value))
+								}
+								if value, ok := item.Arguments["TriggerStrictBucketCapacity"]; ok {
+									arguments["TriggerStrictBucketCapacity"] = math.Float64frombits(binary.LittleEndian.Uint64(value))
+								}
+								if value, ok := item.Arguments["TriggerStrictBucketRate"]; ok {
+									arguments["TriggerStrictBucketRate"] = math.Float64frombits(binary.LittleEndian.Uint64(value))
+								}
+								if value, ok := item.Arguments["MetricsFlushInterval"]; ok {
+									arguments["MetricsFlushInterval"] = int32(binary.LittleEndian.Uint32(value))
+								}
+								if value, ok := item.Arguments["MaxTransactions"]; ok {
+									arguments["MaxTransactions"] = int32(binary.LittleEndian.Uint32(value))
+								}
+								if value, ok := item.Arguments["MaxCustomMetrics"]; ok {
+									arguments["MaxCustomMetrics"] = int32(binary.LittleEndian.Uint32(value))
+								}
+								if value, ok := item.Arguments["EventsFlushInterval"]; ok {
+									arguments["EventsFlushInterval"] = int32(binary.LittleEndian.Uint32(value))
+								}
+								if value, ok := item.Arguments["ProfilingInterval"]; ok {
+									arguments["ProfilingInterval"] = int32(binary.LittleEndian.Uint32(value))
+								}
+								// Remove SignatureKey from collector response
+								delete(arguments, "SignatureKey")
+							}
+							settings = append(settings, setting)
+						}
+					}
+				}
+				if content, err := json.Marshal(settings); err != nil {
+					extension.logger.Warn("Error to marshal setting JSON[] byte from settings " + err.Error())
+				} else {
+					if err := os.WriteFile(JSONOutputFile, content, 0600); err != nil {
+						extension.logger.Error("Unable to write " + JSONOutputFile + " " + err.Error())
+					} else {
+						if len(response.GetWarning()) > 0 {
+							extension.logger.Warn(JSONOutputFile + " is refreshed (soft disabled)")
+						} else {
+							extension.logger.Info(JSONOutputFile + " is refreshed")
+						}
+						extension.logger.Info(string(content))
+					}
+				}
+			case collectorpb.ResultCode_TRY_LATER:
+				extension.logger.Warn("GetSettings returned TRY_LATER " + response.GetWarning())
+			case collectorpb.ResultCode_INVALID_API_KEY:
+				extension.logger.Warn("GetSettings returned INVALID_API_KEY " + response.GetWarning())
+			case collectorpb.ResultCode_LIMIT_EXCEEDED:
+				extension.logger.Warn("GetSettings returned LIMIT_EXCEEDED " + response.GetWarning())
+			case collectorpb.ResultCode_REDIRECT:
+				extension.logger.Warn("GetSettings returned REDIRECT " + response.GetWarning())
+			default:
+				extension.logger.Warn("Unknown ResultCode from GetSettings " + response.GetWarning())
+			}
+		}
+	}
 }

extension/solarwindsapmsettingsextension/go.mod

@@ -13,6 +13,7 @@ require (
 	go.uber.org/goleak v1.3.0
 	go.uber.org/zap v1.27.0
 	google.golang.org/grpc v1.64.0
+	google.golang.org/protobuf v1.34.1
 )
 
 require (
@@ -45,6 +46,5 @@ require (
 	golang.org/x/sys v0.20.0 // indirect
 	golang.org/x/text v0.14.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20240318140521-94a12d6c2237 // indirect
-	google.golang.org/protobuf v1.34.1 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )