buyfakett/server_torii
1
0
Fork 0
mirror of https://github.com/Rayzggz/server_torii.git synced 2025-06-19 21:51:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: HTTP FLOOD Speed Limit

Browse Source 
add shared memory
This commit is contained in:
Roi Feng
2025-04-09 20:52:40 -04:00
parent 4dad82eb40
commit b49c3573bb
14 changed files with 63 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
main.go
View File

@ -6,6 +6,7 @@ import (
"os"
"os/signal"
"server_torii/internal/config"
"server_torii/internal/dataType"
"server_torii/internal/server"
"syscall"
)
@ -47,6 +48,10 @@ func main() {
log.SetOutput(logFile)
//allocate shared memory
sharedMem := &dataType.SharedMemory{
HTTPFloodSpeedLimitCounter: dataType.NewCounter(64, 60),
HTTPFloodSameURILimitCounter: dataType.NewCounter(64, 60),
}
// Start server
stop := make(chan os.Signal, 1)
@ -54,7 +59,7 @@ func main() {
serverErr := make(chan error, 1)
go func() {
serverErr <- server.StartServer(cfg, ruleSet)
serverErr <- server.StartServer(cfg, ruleSet, sharedMem)
}()
select {