50 lines
919 B
Go
50 lines
919 B
Go
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package logs
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
const (
|
|
defaultMaxBufferSize = 2 << 15
|
|
defaultTimeout = 2 * time.Second
|
|
)
|
|
|
|
func LogBufferedDataEvents(dataCh <-chan []byte, logger *zerolog.Logger, eventType string) {
|
|
timer := time.NewTicker(defaultTimeout)
|
|
defer timer.Stop()
|
|
|
|
var buffer []byte
|
|
defer func() {
|
|
if len(buffer) > 0 {
|
|
logger.Info().Str(eventType, string(buffer)).Msg("Streaming process event (flush)")
|
|
}
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-timer.C:
|
|
if len(buffer) > 0 {
|
|
logger.Info().Str(eventType, string(buffer)).Msg("Streaming process event")
|
|
buffer = nil
|
|
}
|
|
case data, ok := <-dataCh:
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
buffer = append(buffer, data...)
|
|
|
|
if len(buffer) >= defaultMaxBufferSize {
|
|
logger.Info().Str(eventType, string(buffer)).Msg("Streaming process event")
|
|
buffer = nil
|
|
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|