feat(ingest): batch event writes and harden transport
This commit is contained in:
@@ -7,6 +7,7 @@ import (
|
|||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
"os/signal"
|
"os/signal"
|
||||||
|
"sync"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -69,15 +70,60 @@ func main() {
|
|||||||
|
|
||||||
log.Printf("subscribed to %s (%s)", natsTopic, natsURL)
|
log.Printf("subscribed to %s (%s)", natsTopic, natsURL)
|
||||||
|
|
||||||
|
const (
|
||||||
|
maxBatchSize = 100
|
||||||
|
flushInterval = 200 * time.Millisecond
|
||||||
|
)
|
||||||
|
|
||||||
|
eventCh := make(chan postgres.InsertEvent, 1000)
|
||||||
|
|
||||||
|
// Flush goroutine
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
wg.Add(1)
|
||||||
|
go func() {
|
||||||
|
defer wg.Done()
|
||||||
|
batch := make([]postgres.InsertEvent, 0, maxBatchSize)
|
||||||
|
ticker := time.NewTicker(flushInterval)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
flush := func() {
|
||||||
|
if len(batch) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := db.InsertEventBatch(ctx, batch); err != nil {
|
||||||
|
log.Printf("batch insert error (%d events): %v", len(batch), err)
|
||||||
|
}
|
||||||
|
batch = batch[:0]
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case evt, ok := <-eventCh:
|
||||||
|
if !ok {
|
||||||
|
flush()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
batch = append(batch, evt)
|
||||||
|
if len(batch) >= maxBatchSize {
|
||||||
|
flush()
|
||||||
|
}
|
||||||
|
case <-ticker.C:
|
||||||
|
flush()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
err = sub.Subscribe(ctx, func(msg []byte) error {
|
err = sub.Subscribe(ctx, func(msg []byte) error {
|
||||||
var env envelope
|
var env envelope
|
||||||
if err := json.Unmarshal(msg, &env); err != nil {
|
if err := json.Unmarshal(msg, &env); err != nil {
|
||||||
return err
|
log.Printf("unmarshal error: %v", err)
|
||||||
|
return nil // don't fail the subscriber
|
||||||
}
|
}
|
||||||
|
|
||||||
ts, err := parseTS(env.Event.TS)
|
ts, err := parseTS(env.Event.TS)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
log.Printf("timestamp parse error: %v", err)
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
e := postgres.InsertEvent{
|
e := postgres.InsertEvent{
|
||||||
@@ -97,9 +143,17 @@ func main() {
|
|||||||
e.ParentSpanID = sqlNull(env.Correlation.ParentSpanID)
|
e.ParentSpanID = sqlNull(env.Correlation.ParentSpanID)
|
||||||
}
|
}
|
||||||
|
|
||||||
return db.InsertEvent(ctx, e)
|
select {
|
||||||
|
case eventCh <- e:
|
||||||
|
default:
|
||||||
|
log.Printf("event channel full, dropping event %s", env.Event.ID)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
})
|
})
|
||||||
|
|
||||||
|
close(eventCh)
|
||||||
|
wg.Wait()
|
||||||
|
|
||||||
if err != nil && err != context.Canceled {
|
if err != nil && err != context.Canceled {
|
||||||
log.Printf("processor stopped: %v", err)
|
log.Printf("processor stopped: %v", err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -38,6 +38,7 @@ func main() {
|
|||||||
})
|
})
|
||||||
|
|
||||||
r.Post("/v1/events", func(w http.ResponseWriter, r *http.Request) {
|
r.Post("/v1/events", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
r.Body = http.MaxBytesReader(w, r.Body, 10<<20) // 10MB limit
|
||||||
var events []json.RawMessage
|
var events []json.RawMessage
|
||||||
if err := json.NewDecoder(r.Body).Decode(&events); err != nil {
|
if err := json.NewDecoder(r.Body).Decode(&events); err != nil {
|
||||||
httpx.WriteJSON(w, http.StatusBadRequest, map[string]any{"error": "invalid_json"})
|
httpx.WriteJSON(w, http.StatusBadRequest, map[string]any{"error": "invalid_json"})
|
||||||
|
|||||||
@@ -26,9 +26,5 @@ func (p *Publisher) Close() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *Publisher) Publish(ctx context.Context, data []byte) error {
|
func (p *Publisher) Publish(ctx context.Context, data []byte) error {
|
||||||
ctx, cancel := context.WithTimeout(ctx, p.timeout)
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
_ = ctx
|
|
||||||
return p.conn.Publish(p.topic, data)
|
return p.conn.Publish(p.topic, data)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -221,7 +221,7 @@ func (e *Emitter) Emit(ctx context.Context, event Event) error {
|
|||||||
|
|
||||||
e.buffer = append(e.buffer, event)
|
e.buffer = append(e.buffer, event)
|
||||||
if len(e.buffer) >= e.bufferSize {
|
if len(e.buffer) >= e.bufferSize {
|
||||||
return e.Flush(ctx)
|
return e.flushLocked(ctx)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -235,6 +235,11 @@ func (e *Emitter) Flush(ctx context.Context) error {
|
|||||||
return fmt.Errorf("emitter is closed")
|
return fmt.Errorf("emitter is closed")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return e.flushLocked(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
// flushLocked sends buffered events. Caller must hold e.mu.
|
||||||
|
func (e *Emitter) flushLocked(ctx context.Context) error {
|
||||||
if len(e.buffer) == 0 {
|
if len(e.buffer) == 0 {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -281,7 +286,7 @@ func (e *Emitter) Close(ctx context.Context) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if len(e.buffer) > 0 {
|
if len(e.buffer) > 0 {
|
||||||
_ = e.Flush(ctx)
|
_ = e.flushLocked(ctx)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
Reference in New Issue
Block a user