55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package common
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"fiskerinc.com/modules/grpc/kafka_grpc"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type TRexLog struct {
|
|
Level LogLevel `json:"level"`
|
|
Timestamp string `json:"timestamp"`
|
|
LineNumber int `json:"line_number"`
|
|
FileName string `json:"filename"`
|
|
Channel string `json:"channel"`
|
|
Message string `json:"msg"`
|
|
}
|
|
type LogTrexLog struct {
|
|
TRexLog
|
|
ReceivedTimestamp string `json:"received_timestamp"`
|
|
}
|
|
|
|
type TRexLogs []TRexLog
|
|
|
|
func (tl *TRexLogs) ToGrpc(data MessageRawJSON) (*kafka_grpc.TRexLogs_BatchPayload, error) {
|
|
var logs TRexLogs
|
|
err := json.Unmarshal(data.Data, &logs)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
var msg kafka_grpc.TRexLogs
|
|
msg.Logs = make([]*kafka_grpc.TRexLog, len(logs))
|
|
for i, log := range logs {
|
|
msg.Logs[i] = &kafka_grpc.TRexLog{
|
|
Level: int32(log.Level),
|
|
Timestamp: log.Timestamp,
|
|
LineNumber: int32(log.LineNumber),
|
|
Filename: log.FileName,
|
|
Channel: log.Channel,
|
|
Msg: log.Message,
|
|
}
|
|
}
|
|
//msg is the type TRexLogs
|
|
nested := kafka_grpc.TRexLogs_BatchPayload_Data{
|
|
Data: &msg,
|
|
}
|
|
batchPayload := kafka_grpc.TRexLogs_BatchPayload{
|
|
Handler: data.Handler,
|
|
TRexMessage: &nested,
|
|
Version: data.Version,
|
|
}
|
|
|
|
return &batchPayload, nil
|
|
}
|