feat: add hook system, QR code login, and layered cache
- Implement extensible hook system for content moderation with builtin and AI-powered moderation hooks - Add QR code login feature with SSE for real-time status updates and scan/confirm/cancel workflow - Introduce layered cache with local LRU + Redis backend for improved read performance - Refactor post and comment services to use hook-based moderation instead of direct AI service calls - Add local cache configuration options (size, buckets, ttl)
This commit is contained in:
122
internal/pkg/hook/metrics.go
Normal file
122
internal/pkg/hook/metrics.go
Normal file
@@ -0,0 +1,122 @@
|
||||
package hook
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Metrics struct {
|
||||
totalCalls atomic.Int64
|
||||
failedCalls atomic.Int64
|
||||
totalTime atomic.Int64
|
||||
hookMetrics map[string]*hookMetric
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
type hookMetric struct {
|
||||
calls atomic.Int64
|
||||
failures atomic.Int64
|
||||
totalTime atomic.Int64
|
||||
lastCalled atomic.Int64
|
||||
}
|
||||
|
||||
func NewMetrics() *Metrics {
|
||||
return &Metrics{
|
||||
hookMetrics: make(map[string]*hookMetric),
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Metrics) RecordHookExecution(name string, hookType HookType, duration time.Duration, success bool) {
|
||||
m.totalCalls.Add(1)
|
||||
m.totalTime.Add(int64(duration))
|
||||
|
||||
if !success {
|
||||
m.failedCalls.Add(1)
|
||||
}
|
||||
|
||||
key := string(hookType) + ":" + name
|
||||
|
||||
m.mu.Lock()
|
||||
metric, ok := m.hookMetrics[key]
|
||||
if !ok {
|
||||
metric = &hookMetric{}
|
||||
m.hookMetrics[key] = metric
|
||||
}
|
||||
m.mu.Unlock()
|
||||
|
||||
metric.calls.Add(1)
|
||||
metric.totalTime.Add(int64(duration))
|
||||
metric.lastCalled.Store(time.Now().Unix())
|
||||
|
||||
if !success {
|
||||
metric.failures.Add(1)
|
||||
}
|
||||
}
|
||||
|
||||
type MetricSnapshot struct {
|
||||
Name string
|
||||
Type HookType
|
||||
Calls int64
|
||||
Failures int64
|
||||
TotalTime time.Duration
|
||||
AvgTime time.Duration
|
||||
LastCalled time.Time
|
||||
}
|
||||
|
||||
func (m *Metrics) GetTotalCalls() int64 {
|
||||
return m.totalCalls.Load()
|
||||
}
|
||||
|
||||
func (m *Metrics) GetFailedCalls() int64 {
|
||||
return m.failedCalls.Load()
|
||||
}
|
||||
|
||||
func (m *Metrics) GetTotalTime() time.Duration {
|
||||
return time.Duration(m.totalTime.Load())
|
||||
}
|
||||
|
||||
func (m *Metrics) GetHookMetrics() []MetricSnapshot {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
|
||||
var snapshots []MetricSnapshot
|
||||
for key, metric := range m.hookMetrics {
|
||||
calls := metric.calls.Load()
|
||||
var avgTime time.Duration
|
||||
if calls > 0 {
|
||||
avgTime = time.Duration(metric.totalTime.Load() / calls)
|
||||
}
|
||||
|
||||
hookType, name := parseKey(key)
|
||||
snapshots = append(snapshots, MetricSnapshot{
|
||||
Name: name,
|
||||
Type: hookType,
|
||||
Calls: calls,
|
||||
Failures: metric.failures.Load(),
|
||||
TotalTime: time.Duration(metric.totalTime.Load()),
|
||||
AvgTime: avgTime,
|
||||
LastCalled: time.Unix(metric.lastCalled.Load(), 0),
|
||||
})
|
||||
}
|
||||
return snapshots
|
||||
}
|
||||
|
||||
func parseKey(key string) (HookType, string) {
|
||||
for i := 0; i < len(key); i++ {
|
||||
if key[i] == ':' {
|
||||
return HookType(key[:i]), key[i+1:]
|
||||
}
|
||||
}
|
||||
return HookType(key), ""
|
||||
}
|
||||
|
||||
func (m *Metrics) Reset() {
|
||||
m.totalCalls.Store(0)
|
||||
m.failedCalls.Store(0)
|
||||
m.totalTime.Store(0)
|
||||
|
||||
m.mu.Lock()
|
||||
m.hookMetrics = make(map[string]*hookMetric)
|
||||
m.mu.Unlock()
|
||||
}
|
||||
Reference in New Issue
Block a user