mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-23 18:01:43 +01:00
53f3397b7a
* Add comprehensive post-request rules support for response phase * Enable response body, status, and header manipulation via set commands * Refactor command handlers to support both request and response phases * Implement response modifier system for post-request template execution * Support response-based rule matching with status and header checks * Add comprehensive benchmarks for matcher performance * Refactor authentication and proxying commands for unified error handling * Support negated conditions with ! * Enhance error handling, error formatting and validation * Routes: add `rule_file` field with rule preset support * Environment variable substitution: now supports variables without `GODOXY_` prefix * new conditions: * `on resp_header <key> [<value>]` * `on status <status>` * new commands: * `require_auth` * `set resp_header <key> <template>` * `set resp_body <template>` * `set status <code>` * `log <level> <path> <template>` * `notify <level> <provider> <title_template> <body_template>`
91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
package accesslog
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/yusing/goutils/task"
|
|
expect "github.com/yusing/goutils/testing"
|
|
)
|
|
|
|
func TestConcurrentFileLoggersShareSameAccessLogIO(t *testing.T) {
|
|
var wg sync.WaitGroup
|
|
|
|
cfg := DefaultRequestLoggerConfig()
|
|
cfg.Path = "test.log"
|
|
|
|
loggerCount := 10
|
|
accessLogIOs := make([]WriterWithName, loggerCount)
|
|
|
|
// make test log file
|
|
file, err := os.Create(cfg.Path)
|
|
expect.NoError(t, err)
|
|
file.Close()
|
|
t.Cleanup(func() {
|
|
expect.NoError(t, os.Remove(cfg.Path))
|
|
})
|
|
|
|
for i := range loggerCount {
|
|
wg.Add(1)
|
|
go func(index int) {
|
|
defer wg.Done()
|
|
file, err := NewFileIO(cfg.Path)
|
|
expect.NoError(t, err)
|
|
accessLogIOs[index] = file
|
|
}(i)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
firstIO := accessLogIOs[0]
|
|
for _, io := range accessLogIOs {
|
|
expect.Equal(t, io, firstIO)
|
|
}
|
|
}
|
|
|
|
func TestConcurrentAccessLoggerLogAndFlush(t *testing.T) {
|
|
file := NewMockFile()
|
|
|
|
cfg := DefaultRequestLoggerConfig()
|
|
parent := task.RootTask("test", false)
|
|
|
|
loggerCount := 5
|
|
logCountPerLogger := 10
|
|
loggers := make([]*AccessLogger, loggerCount)
|
|
|
|
for i := range loggerCount {
|
|
loggers[i] = NewAccessLoggerWithIO(parent, file, cfg)
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
|
|
resp := &http.Response{StatusCode: http.StatusOK}
|
|
|
|
wg.Add(len(loggers))
|
|
for _, logger := range loggers {
|
|
go func(l *AccessLogger) {
|
|
defer wg.Done()
|
|
parallelLog(l, req, resp, logCountPerLogger)
|
|
l.Flush()
|
|
}(logger)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
expected := loggerCount * logCountPerLogger
|
|
actual := file.NumLines()
|
|
expect.Equal(t, actual, expected)
|
|
}
|
|
|
|
func parallelLog(logger *AccessLogger, req *http.Request, resp *http.Response, n int) {
|
|
var wg sync.WaitGroup
|
|
for range n {
|
|
wg.Go(func() {
|
|
logger.Log(req, resp)
|
|
})
|
|
}
|
|
wg.Wait()
|
|
}
|