oss/pages-server
oss/pages-server
1
0
Fork 0
mirror of https://codeberg.org/Codeberg/pages-server.git synced 2025-09-30 21:42:05 +00:00
Code Wiki Activity

feat: make logging interval configurable

Browse source
This commit is contained in:
crapStone 2025-09-08 16:36:11 +02:00
parent 30bc87089c
commit 373044f289
No known key found for this signature in database
GPG key ID: 22D4BF0CF7CC29C8
5 changed files with 25 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
cli/flags.go
View file

@ -117,10 +117,16 @@ var (
},
&cli.UintFlag{
Name: "log-most-active-ips",
Usage: "logs a the n most active IPs every hour",
Usage: "logs a the n most active IPs every hour (or value of --ip-logging-interval)",
EnvVars: []string{"LOG_MOST_ACTIVE_IPS"},
Value: 10,
},
&cli.UintFlag{
Name: "ip-logging-interval",
Usage: "interval in seconds for ip address logging",
EnvVars: []string{"IP_LOGGING_INTERVAL"},
Value: 0,
},
// Default branches to fetch assets from
&cli.StringSliceFlag{

1
config/config.go
View file

@ -16,6 +16,7 @@ type ServerConfig struct {
UseProxyProtocol bool `default:"false"`
LogMostActiveIps bool `default:"false"`
MostActiveIpCount uint `default:"10"`
LoggingInterval uint `default:"0"`
MainDomain string
RawDomain string
PagesBranches []string

3
config/setup.go
View file

@ -76,6 +76,9 @@ func mergeServerConfig(ctx *cli.Context, config *ServerConfig) {
config.LogMostActiveIps = true
config.MostActiveIpCount = ctx.Uint("log-most-active-ips")
}
if ctx.IsSet("ip-logging-interval") {
config.LoggingInterval = ctx.Uint("ip-logging-interval")
}
if ctx.IsSet("pages-domain") {
config.MainDomain = ctx.String("pages-domain")

8
config/setup_test.go
View file

@ -143,6 +143,7 @@ func TestMergeConfigShouldReplaceAllExistingValuesGivenAllArgsExist(t *testing.T
UseProxyProtocol: false,
LogMostActiveIps: false,
MostActiveIpCount: 10,
LoggingInterval: 0,
MainDomain: "original",
RawDomain: "original",
PagesBranches: []string{"original"},
@ -185,6 +186,7 @@ func TestMergeConfigShouldReplaceAllExistingValuesGivenAllArgsExist(t *testing.T
HttpServerEnabled: true,
UseProxyProtocol: true,
LogMostActiveIps: true,
LoggingInterval: 300,
MostActiveIpCount: 42,
MainDomain: "changed",
RawDomain: "changed",
@ -235,6 +237,7 @@ func TestMergeConfigShouldReplaceAllExistingValuesGivenAllArgsExist(t *testing.T
"--enable-http-server",
"--use-proxy-protocol",
"--log-most-active-ips", "42",
"--ip-logging-interval", "300",
// Forge
"--forge-root", "changed",
"--forge-api-token", "changed",
@ -288,6 +291,7 @@ func TestMergeServerConfigShouldReplaceAllExistingValuesGivenAllArgsExist(t *tes
UseProxyProtocol: false,
LogMostActiveIps: false,
MostActiveIpCount: 10,
LoggingInterval: 0,
MainDomain: "original",
RawDomain: "original",
AllowedCorsDomains: []string{"original"},
@ -304,6 +308,7 @@ func TestMergeServerConfigShouldReplaceAllExistingValuesGivenAllArgsExist(t *tes
UseProxyProtocol: true,
LogMostActiveIps: true,
MostActiveIpCount: 21,
LoggingInterval: 300,
MainDomain: "changed",
RawDomain: "changed",
AllowedCorsDomains: fixArrayFromCtx(ctx, "allowed-cors-domains", []string{"changed"}),
@ -325,6 +330,7 @@ func TestMergeServerConfigShouldReplaceAllExistingValuesGivenAllArgsExist(t *tes
"--enable-http-server",
"--use-proxy-protocol",
"--log-most-active-ips", "21",
"--ip-logging-interval", "300",
},
)
}
@ -350,6 +356,7 @@ func TestMergeServerConfigShouldReplaceOnlyOneValueExistingValueGivenOnlyOneArgE
sc.LogMostActiveIps = true
sc.MostActiveIpCount = 42
}},
{args: []string{"--ip-logging-interval", "300"}, callback: func(sc *ServerConfig) { sc.LoggingInterval = 300 }},
}
for _, pair := range testValuePairs {
@ -369,6 +376,7 @@ func TestMergeServerConfigShouldReplaceOnlyOneValueExistingValueGivenOnlyOneArgE
UseProxyProtocol: false,
LogMostActiveIps: false,
MostActiveIpCount: 10,
LoggingInterval: 0,
}
expectedConfig := cfg

7
server/startup.go
View file

@ -140,7 +140,12 @@ func Serve(ctx *cli.Context) error {
mostActiveIpMap := &sync.Map{}
if cfg.Server.LogMostActiveIps {
go func() {
ticker := time.NewTicker(1 * time.Hour)
interval := 1 * time.Hour
if cfg.Server.LoggingInterval > 0 {
interval = time.Duration(cfg.Server.LoggingInterval) * time.Second
}
ticker := time.NewTicker(interval)
for range ticker.C {
type kv struct {
Key string