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

feat: HTTP FLOOD Speed Limit

Browse Source
This commit is contained in:
Roi Feng
2025-04-09 21:47:47 -04:00
parent b49c3573bb
commit 4893e95bb7
3 changed files with 34 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
main.go
View File

@ -5,10 +5,13 @@ import (
"log"
"os"
"os/signal"
"runtime"
"server_torii/internal/config"
"server_torii/internal/dataType"
"server_torii/internal/server"
"server_torii/internal/utils"
"syscall"
"time"
)
func main() {
@ -49,10 +52,15 @@ func main() {
//allocate shared memory
sharedMem := &dataType.SharedMemory{
HTTPFloodSpeedLimitCounter: dataType.NewCounter(64, 60),
HTTPFloodSameURILimitCounter: dataType.NewCounter(64, 60),
HTTPFloodSpeedLimitCounter: dataType.NewCounter(max(runtime.NumCPU()*8, 16), utils.FindMaxRateTime(ruleSet.HTTPFloodRule.HTTPFloodSpeedLimit)),
HTTPFloodSameURILimitCounter: dataType.NewCounter(max(runtime.NumCPU()*8, 16), utils.FindMaxRateTime(ruleSet.HTTPFloodRule.HTTPFloodSameURILimit)),
}
//GC
gcStopCh := make(chan struct{})
go dataType.StartCounterGC(sharedMem.HTTPFloodSpeedLimitCounter, time.Minute, gcStopCh)
go dataType.StartCounterGC(sharedMem.HTTPFloodSameURILimitCounter, time.Minute, gcStopCh)
// Start server
stop := make(chan os.Signal, 1)
signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
@ -65,6 +73,7 @@ func main() {
select {
case <-stop:
log.Println("Stopping server...")
close(gcStopCh)
case err := <-serverErr:
if err != nil {
log.Fatalf("Failed to start server: %v", err)