Implement CheckCle Microservice Operation

A Go-based microservice for service operations including ICMP ping, DNS resolution, and TCP connectivity
2025-06-19 17:49:30 +07:00
parent 89572c44b7
commit d3ea5dc77c
36 changed files with 2895 additions and 0 deletions
@@ -0,0 +1,74 @@
+
+package savers
+
+import (
+	"fmt"
+	"strings"
+	"time"
+
+	"service-operation/pocketbase"
+	"service-operation/types"
+)
+
+func (ms *MetricsSaver) SaveDNSDataToPocketBase(result *types.OperationResult, serviceID string) {
+	// Create a short, professional status message
+	var details string
+	
+	if result.Success {
+		// Success message with record count and query info
+		recordCount := len(result.DNSRecords)
+		details = fmt.Sprintf("✅ DNS %s Query OK - %d records found", 
+			strings.ToUpper(result.DNSType), recordCount)
+		
+		// Add response time
+		details += fmt.Sprintf(" | Response time: %.2fms", 
+			float64(result.ResponseTime.Nanoseconds())/1000000)
+		
+		// Add first few records for context
+		if recordCount > 0 {
+			if recordCount <= 2 {
+				details += fmt.Sprintf(" | Records: %s", strings.Join(result.DNSRecords, ", "))
+			} else {
+				details += fmt.Sprintf(" | Records: %s... (+%d more)", 
+					strings.Join(result.DNSRecords[:2], ", "), recordCount-2)
+			}
+		}
+	} else {
+		// Error message with query type
+		details = fmt.Sprintf("❌ DNS %s Query Failed - %s", 
+			strings.ToUpper(result.DNSType), 
+			GetShortErrorMessage(result.Error))
+		
+		// Add response time if available
+		if result.ResponseTime > 0 {
+			details += fmt.Sprintf(" | Response time: %.2fms", 
+				float64(result.ResponseTime.Nanoseconds())/1000000)
+		}
+	}
+
+	dnsData := pocketbase.DNSDataRecord{
+		ServiceID:    serviceID,
+		Timestamp:    time.Now(),
+		ResponseTime: result.ResponseTime.Milliseconds(),
+		Status:       GetStatusString(result.Success),
+		QueryType:    result.DNSType,
+		ResolveIP:    strings.Join(result.DNSRecords, ","),
+		MsgSize:      fmt.Sprintf("%d", len(result.DNSRecords)),
+		Question:     result.Host,
+		Answer:       strings.Join(result.DNSRecords, ","),
+		Authority:    "", // Not available in current implementation
+		ErrorMessage: result.Error,
+		Details:      details, // Short, clean message
+		RegionName:   "default", // You can make this configurable
+		AgentID:      "1",       // You can make this configurable
+	}
+
+	if err := ms.pbClient.SaveDNSData(dnsData); err != nil {
+		println("Failed to save DNS data to PocketBase:", err.Error())
+	}
+}
+
+// Method for monitoring service usage
+func (ms *MetricsSaver) SaveDNSDataForService(service pocketbase.Service, result *types.OperationResult) {
+	ms.SaveDNSDataToPocketBase(result, service.ID)
+}