mirror of
https://github.com/microsoft/ethr.git
synced 2024-09-20 06:46:14 +08:00
ea28260abc
* Flatten the file structure * Fix build errors Co-authored-by: Pankaj Garg <pankaj@intellectualheaven.com>
147 lines
3.2 KiB
Go
147 lines
3.2 KiB
Go
//-----------------------------------------------------------------------------
|
|
// Copyright (C) Microsoft. All rights reserved.
|
|
// Licensed under the MIT license.
|
|
// See LICENSE.txt file in the project root for full license information.
|
|
//-----------------------------------------------------------------------------
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// LogLevel specifies the logging level to use in both screen and
|
|
// file-based logging
|
|
type LogLevel int
|
|
|
|
const (
|
|
LogLevelInfo LogLevel = iota
|
|
LogLevelDebug
|
|
)
|
|
|
|
type logMessage struct {
|
|
Time string
|
|
Type string
|
|
Message string
|
|
}
|
|
|
|
type logLatencyData struct {
|
|
Time string
|
|
Type string
|
|
RemoteAddr string
|
|
Protocol string
|
|
Avg string
|
|
Min string
|
|
P50 string
|
|
P90 string
|
|
P95 string
|
|
P99 string
|
|
P999 string
|
|
P9999 string
|
|
Max string
|
|
}
|
|
|
|
type logTestResults struct {
|
|
Time string
|
|
Type string
|
|
RemoteAddr string
|
|
Protocol string
|
|
BitsPerSecond string
|
|
ConnectionsPerSecond string
|
|
PacketsPerSecond string
|
|
AverageLatency string
|
|
}
|
|
|
|
var loggingActive = false
|
|
var logChan = make(chan string, 64)
|
|
|
|
func logInit(fileName string) {
|
|
if fileName == "" {
|
|
return
|
|
}
|
|
logFile, err := os.OpenFile(fileName, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0666)
|
|
if err != nil {
|
|
fmt.Printf("Unable to open the log file %s, Error: %v", fileName, err)
|
|
return
|
|
}
|
|
log.SetFlags(0)
|
|
log.SetOutput(logFile)
|
|
loggingActive = true
|
|
go runLogger(logFile)
|
|
}
|
|
|
|
func logFini() {
|
|
loggingActive = false
|
|
}
|
|
|
|
func runLogger(logFile *os.File) {
|
|
for loggingActive {
|
|
s := <-logChan
|
|
log.Println(s)
|
|
}
|
|
logFile.Close()
|
|
}
|
|
|
|
func logMsg(prefix, msg string) {
|
|
if loggingActive {
|
|
logData := logMessage{}
|
|
logData.Time = time.Now().UTC().Format(time.RFC3339)
|
|
logData.Type = prefix
|
|
logData.Message = msg
|
|
logJSON, _ := json.Marshal(logData)
|
|
logChan <- string(logJSON)
|
|
}
|
|
}
|
|
|
|
func logInfo(msg string) {
|
|
logMsg("INFO", msg)
|
|
}
|
|
|
|
func logError(msg string) {
|
|
logMsg("ERROR", msg)
|
|
}
|
|
|
|
func logDebug(msg string) {
|
|
logMsg("DEBUG", msg)
|
|
}
|
|
|
|
func logResults(s []string) {
|
|
if loggingActive {
|
|
logData := logTestResults{}
|
|
logData.Time = time.Now().UTC().Format(time.RFC3339)
|
|
logData.Type = "TestResult"
|
|
logData.RemoteAddr = s[0]
|
|
logData.Protocol = s[1]
|
|
logData.BitsPerSecond = s[2]
|
|
logData.ConnectionsPerSecond = s[3]
|
|
logData.PacketsPerSecond = s[4]
|
|
logData.AverageLatency = s[5]
|
|
logJSON, _ := json.Marshal(logData)
|
|
logChan <- string(logJSON)
|
|
}
|
|
}
|
|
|
|
func logLatency(remoteAddr, proto string, avg, min, p50, p90, p95, p99, p999, p9999, max time.Duration) {
|
|
if loggingActive {
|
|
logData := logLatencyData{}
|
|
logData.Time = time.Now().UTC().Format(time.RFC3339)
|
|
logData.Type = "LatencyResult"
|
|
logData.RemoteAddr = remoteAddr
|
|
logData.Protocol = proto
|
|
logData.Avg = durationToString(avg)
|
|
logData.Min = durationToString(min)
|
|
logData.P50 = durationToString(p50)
|
|
logData.P90 = durationToString(p90)
|
|
logData.P95 = durationToString(p95)
|
|
logData.P99 = durationToString(p99)
|
|
logData.P999 = durationToString(p999)
|
|
logData.P9999 = durationToString(p9999)
|
|
logData.Max = durationToString(max)
|
|
logJSON, _ := json.Marshal(logData)
|
|
logChan <- string(logJSON)
|
|
}
|
|
}
|