io.go

package main

import (
	"encoding/json"
	"fmt"
	"os"
	"sync"
	"time"
)

var (
	configFile string = getEnv("CONFIG_FILE", "config.json")
)

type API struct {
	URLs []string `json:"urls"`
	Body string   `json:"body,omitempty"`
}

type APIs struct {
	APIs []API `json:"apis"`
}

type Config struct {
	Version    string                `json:"version"`
	InputFile  string                `json:"inputFile"`
	Random     bool                  `json:"random,omitempty"`
	Split      bool                  `json:"split,omitempty"`
	NumProcess int                   `json:"numProcess,omitempty"`
	NumMinutes int                   `json:"numMinutes,omitempty"`
	duration   time.Duration         `json:"duration,omitempty"`
	MinDelay   int                   `json:"minDelay,omitempty"`
	MaxDelay   int                   `json:"maxDelay,omitempty"`
	BaseURL    string                `json:"baseURL,omitempty"`
	Params     []map[string][]string `json:"params,omitempty"`
	group      int
	mu         sync.Mutex
}

func readConfig() (*Config, error) {
	// Get endpoint list
	data, err := os.ReadFile(configFile)
	if err != nil {
		return nil, fmt.Errorf("error reading endpoint file:%s", err)
	}

	// Parse JSON data
	var config Config
	err = json.Unmarshal(data, &config)
	if err != nil {
		return nil, fmt.Errorf("error parsing JSON:%s", err)
	}

	return &config, nil
}

func (c *Config) readInput() (*APIs, error) {
	// Get endpoint list
	data, err := os.ReadFile(c.InputFile)
	if err != nil {
		return nil, fmt.Errorf("error reading endpoint file:%s", err)
	}

	// Parse JSON data
	var apis APIs
	err = json.Unmarshal(data, &apis)
	if err != nil {
		return nil, fmt.Errorf("error parsing JSON:%s", err)
	}

	return &apis, nil
}

func printDetails(t *RequestTrace, path, version, group string) {
	// Print timing information
	output := fmt.Sprintf("%d,%d,%d,%d,%d,%d,%d,%d,%d,%s,%s,%s", t.StatusCode, t.TotalResponseTime, t.DNSLookup,
		t.Connection, t.TLSHandshake, t.ServerProcessing, t.TimeToFirstResponseByte,
		t.DownloadTime, t.ResponseSize, path, version, group)

	// Print details
	fmt.Println(output)
}