feat: reuse dump records

This commit is contained in:
SunBK201 2025-12-11 00:13:32 +08:00
parent 9de776f21b
commit 241cc6a743
3 changed files with 31 additions and 22 deletions

View File

@ -17,8 +17,10 @@ type ConnectionRecordList struct {
recordRemoveChan chan *ConnectionRecord
records map[string]*ConnectionRecord
mu sync.RWMutex
dumpFile string
dumpWriter *bufio.Writer
dumpRecords []*ConnectionRecord
dumpFile string
dumpWriter *bufio.Writer
}
type ConnectionRecord struct {
@ -34,6 +36,7 @@ func NewConnectionRecordList(dumpFile string) *ConnectionRecordList {
recordRemoveChan: make(chan *ConnectionRecord, 100),
records: make(map[string]*ConnectionRecord, 500),
mu: sync.RWMutex{},
dumpRecords: make([]*ConnectionRecord, 0, 500),
dumpFile: dumpFile,
dumpWriter: bufio.NewWriter(nil),
}
@ -98,16 +101,16 @@ func (l *ConnectionRecordList) Dump() {
}
}()
l.dumpRecords = l.dumpRecords[:0]
l.mu.RLock()
records := make([]*ConnectionRecord, 0, len(l.records))
for _, r := range l.records {
records = append(records, r)
l.dumpRecords = append(l.dumpRecords, r)
}
l.mu.RUnlock()
// Sort by start time (newest first)
sort.SliceStable(records, func(i, j int) bool {
return records[i].StartTime.After(records[j].StartTime)
sort.SliceStable(l.dumpRecords, func(i, j int) bool {
return l.dumpRecords[i].StartTime.After(l.dumpRecords[j].StartTime)
})
l.dumpWriter.Reset(f)
@ -118,7 +121,7 @@ func (l *ConnectionRecordList) Dump() {
}()
now := time.Now()
for _, record := range records {
for _, record := range l.dumpRecords {
duration := now.Sub(record.StartTime)
_, err := fmt.Fprintf(l.dumpWriter, "%s %s %s %d\n",
record.Protocol, record.SrcAddr, record.DestAddr, int(duration.Seconds()))

View File

@ -15,8 +15,10 @@ type PassThroughRecordList struct {
recordAddChan chan *PassThroughRecord
records map[string]*PassThroughRecord
mu sync.RWMutex
dumpFile string
dumpWriter *bufio.Writer
dumpRecords []*PassThroughRecord
dumpFile string
dumpWriter *bufio.Writer
}
type PassThroughRecord struct {
@ -31,6 +33,7 @@ func NewPassThroughRecordList(dumpFile string) *PassThroughRecordList {
recordAddChan: make(chan *PassThroughRecord, 100),
records: make(map[string]*PassThroughRecord, 100),
mu: sync.RWMutex{},
dumpRecords: make([]*PassThroughRecord, 0, 100),
dumpFile: dumpFile,
dumpWriter: bufio.NewWriter(nil),
}
@ -86,15 +89,15 @@ func (l *PassThroughRecordList) Dump() {
}
}()
l.dumpRecords = l.dumpRecords[:0]
l.mu.RLock()
records := make([]*PassThroughRecord, 0, len(l.records))
for _, r := range l.records {
records = append(records, r)
l.dumpRecords = append(l.dumpRecords, r)
}
l.mu.RUnlock()
sort.SliceStable(records, func(i, j int) bool {
return records[i].Count > records[j].Count
sort.SliceStable(l.dumpRecords, func(i, j int) bool {
return l.dumpRecords[i].Count > l.dumpRecords[j].Count
})
l.dumpWriter.Reset(f)
@ -104,7 +107,7 @@ func (l *PassThroughRecordList) Dump() {
}
}()
for _, record := range records {
for _, record := range l.dumpRecords {
_, err := fmt.Fprintf(l.dumpWriter, "%s %s %d %s\n",
record.SrcAddr, record.DestAddr, record.Count, record.UA)
if err != nil {

View File

@ -14,8 +14,10 @@ type RewriteRecordList struct {
recordAddChan chan *RewriteRecord
records map[string]*RewriteRecord
mu sync.RWMutex
dumpFile string
dumpWriter *bufio.Writer
dumpRecords []*RewriteRecord
dumpFile string
dumpWriter *bufio.Writer
}
type RewriteRecord struct {
@ -28,8 +30,9 @@ type RewriteRecord struct {
func NewRewriteRecordList(dumpFile string) *RewriteRecordList {
return &RewriteRecordList{
recordAddChan: make(chan *RewriteRecord, 100),
records: make(map[string]*RewriteRecord, 100),
records: make(map[string]*RewriteRecord, 300),
mu: sync.RWMutex{},
dumpRecords: make([]*RewriteRecord, 0, 300),
dumpFile: dumpFile,
dumpWriter: bufio.NewWriter(nil),
}
@ -81,15 +84,15 @@ func (l *RewriteRecordList) Dump() {
}
}()
l.dumpRecords = l.dumpRecords[:0]
l.mu.RLock()
records := make([]*RewriteRecord, 0, len(l.records))
for _, record := range l.records {
records = append(records, record)
l.dumpRecords = append(l.dumpRecords, record)
}
l.mu.RUnlock()
sort.SliceStable(records, func(i, j int) bool {
return records[i].Count > records[j].Count
sort.SliceStable(l.dumpRecords, func(i, j int) bool {
return l.dumpRecords[i].Count > l.dumpRecords[j].Count
})
l.dumpWriter.Reset(f)
@ -99,7 +102,7 @@ func (l *RewriteRecordList) Dump() {
}
}()
for _, record := range records {
for _, record := range l.dumpRecords {
_, err := fmt.Fprintf(l.dumpWriter, "%s %d %sSEQSEQ%s\n",
record.Host, record.Count, record.OriginalUA, record.MockedUA)
if err != nil {