mirror of
https://github.com/logos-messaging/logos-messaging-go-bindings.git
synced 2026-01-02 05:53:07 +00:00
39 lines
802 B
Go
39 lines
802 B
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/logos-messaging/logos-messaging-go-bindings/utils"
|
|
)
|
|
|
|
var (
|
|
testName string
|
|
iteration int
|
|
phase string
|
|
)
|
|
|
|
func main() {
|
|
flag.StringVar(&testName, "testName", "FullTestSuite", "Name of the test ")
|
|
flag.IntVar(&iteration, "iteration", 0, "Iteration number")
|
|
flag.StringVar(&phase, "phase", "", "'start' or 'end')")
|
|
flag.Parse()
|
|
|
|
var memStats runtime.MemStats
|
|
runtime.ReadMemStats(&memStats)
|
|
heapKB := memStats.HeapAlloc / 1024
|
|
|
|
rssKB, err := utils.GetRSSKB()
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "Failed to get RSS:", err)
|
|
rssKB = 0
|
|
}
|
|
|
|
if err := utils.RecordMemoryMetricsCSV(testName, iteration, phase, heapKB, rssKB); err != nil {
|
|
fmt.Fprintln(os.Stderr, "Error recording metrics:", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|