You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.5 KiB
65 lines
1.5 KiB
package log_buffer
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
type MemBuffer struct {
|
|
buf []byte
|
|
size int
|
|
startTime time.Time
|
|
stopTime time.Time
|
|
batchIndex int64
|
|
}
|
|
|
|
type SealedBuffers struct {
|
|
buffers []*MemBuffer
|
|
}
|
|
|
|
func newSealedBuffers(size int) *SealedBuffers {
|
|
sbs := &SealedBuffers{}
|
|
|
|
sbs.buffers = make([]*MemBuffer, size)
|
|
for i := 0; i < size; i++ {
|
|
sbs.buffers[i] = &MemBuffer{
|
|
buf: make([]byte, BufferSize),
|
|
}
|
|
}
|
|
|
|
return sbs
|
|
}
|
|
|
|
func (sbs *SealedBuffers) SealBuffer(startTime, stopTime time.Time, buf []byte, pos int, batchIndex int64) (newBuf []byte) {
|
|
oldMemBuffer := sbs.buffers[0]
|
|
size := len(sbs.buffers)
|
|
for i := 0; i < size-1; i++ {
|
|
sbs.buffers[i].buf = sbs.buffers[i+1].buf
|
|
sbs.buffers[i].size = sbs.buffers[i+1].size
|
|
sbs.buffers[i].startTime = sbs.buffers[i+1].startTime
|
|
sbs.buffers[i].stopTime = sbs.buffers[i+1].stopTime
|
|
sbs.buffers[i].batchIndex = sbs.buffers[i+1].batchIndex
|
|
}
|
|
sbs.buffers[size-1].buf = buf
|
|
sbs.buffers[size-1].size = pos
|
|
sbs.buffers[size-1].startTime = startTime
|
|
sbs.buffers[size-1].stopTime = stopTime
|
|
sbs.buffers[size-1].batchIndex = batchIndex
|
|
return oldMemBuffer.buf
|
|
}
|
|
|
|
func (mb *MemBuffer) locateByTs(lastReadTime time.Time) (pos int) {
|
|
lastReadTs := lastReadTime.UnixNano()
|
|
for pos < len(mb.buf) {
|
|
size, t := readTs(mb.buf, pos)
|
|
if t > lastReadTs {
|
|
return
|
|
}
|
|
pos += size + 4
|
|
}
|
|
return len(mb.buf)
|
|
}
|
|
|
|
func (mb *MemBuffer) String() string {
|
|
return fmt.Sprintf("[%v,%v] bytes:%d", mb.startTime, mb.stopTime, mb.size)
|
|
}
|