Reporter package

x/reporter/reporter.go

@@ -0,0 +1,94 @@
+// Copyright 2023 Jigsaw Operations LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package reporter
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"log"
+	"math/rand"
+	"net/http"
+)
+
+var debugLog log.Logger = *log.New(io.Discard, "", 0)
+var httpClient = &http.Client{}
+
+func sendReport(record map[string]interface{}, collectorURL string) error {
+	jsonData, err := json.Marshal(record)
+	if err != nil {
+		log.Printf("Error encoding JSON: %s\n", err)
+		return err
+	}
+
+	req, err := http.NewRequest("POST", collectorURL, bytes.NewReader(jsonData))
+	if err != nil {
+		debugLog.Printf("Error creating the HTTP request: %s\n", err)
+		return err
+	}
+
+	req.Header.Set("Content-Type", "application/json; charset=utf-8")
+	resp, err := httpClient.Do(req)
+	if err != nil {
+		log.Printf("Error sending the HTTP request: %s\n", err)
+		return err
+	}
+	defer resp.Body.Close()
+
+	respBody, err := io.ReadAll(resp.Body)
+	if err != nil {
+		debugLog.Printf("Error reading the HTTP response body: %s\n", err)
+		return err
+	}
+	debugLog.Printf("Response: %s\n", respBody)
+	return nil
+}
+
+func sendReportRandomly(record map[string]interface{}, collectorURL string, success bool, successSampleRate float64, failureSampleRate float64) error {
+
+	// Perform custom range validation for sampling rate
+	if successSampleRate < 0.0 || successSampleRate > 1.0 {
+		return errors.New("Error: successSampleRate must be between 0 and 1.")
+	}
+
+	if failureSampleRate < 0.0 || failureSampleRate > 1.0 {
+		return errors.New("Error: failureSampleRate must be between 0 and 1.")
+	}
+
+	var samplingRate float64
+	if success {
+		samplingRate = successSampleRate
+	} else {
+		samplingRate = failureSampleRate
+	}
+	// Generate a random number between 0 and 1
+	random := rand.Float64()
+	if random < samplingRate {
+		// Run your function here
+		err := sendReport(record, collectorURL)
+		if err != nil {
+			log.Printf("HTTP request failed: %v", err)
+			return err
+		} else {
+			fmt.Println("Report sent")
+			return nil
+		}
+	} else {
+		fmt.Println("Report was not sent this time")
+		return nil
+	}
+}

x/reporter/reporter_test.go

@@ -0,0 +1,55 @@
+// Copyright 2023 Jigsaw Operations LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package reporter
+
+import (
+	"encoding/json"
+	"fmt"
+	"testing"
+)
+
+// When success is true and random number is less than successSampleRate, report is sent successfully
+func TestSendReportSuccessfully(t *testing.T) {
+	// Example JSON data
+	jsonData := `{
+		"proxy": "192.168.1.1:65000",
+		"resolver": "8.8.8.8:53",
+		"proto": "tcp",
+		"prefix": "HTTP1/1",
+		"time": "2021-01-01T00:00:00Z",
+		"durationMs": 100,
+		"error": {
+			"operation": "read",
+			"posixError": "ETIMEDOUT",
+			"msg": "i/o timeout"
+		}
+	}`
+	var record map[string]interface{}
+	err := json.Unmarshal([]byte(jsonData), &record)
+	if err != nil {
+		fmt.Println(err)
+		t.Errorf("Expected no error, but got: %v", err)
+	}
+	collectorURL := "example.com"
+	success := true
+	successSampleRate := 1.0
+	failureSampleRate := 0.0
+
+	err = sendReportRandomly(record, collectorURL, success, successSampleRate, failureSampleRate)
+
+	if err != nil {
+		t.Errorf("Expected no error, but got: %v", err)
+	}
+}