From 52d43861f357365d862c1547d6fa96eea6549886 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Thu, 30 Dec 2021 15:48:17 +0800
Subject: [PATCH] feat: set ulimit to 10240 before load testing

---
 docs/CHANGELOG.md |  1 +
 hrp/cmd/boom.go   | 26 ++++++++++++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md
index 9b970960..b6ea98a4 100644
--- a/docs/CHANGELOG.md
+++ b/docs/CHANGELOG.md
@@ -2,6 +2,7 @@
 
 ## v0.4.0 (2021-12-30)
 
+- feat: set ulimit to 10240 before load testing
 - fix: concurrent map writes in load testing
 
 ## v0.3.0 (2021-12-24)
diff --git a/hrp/cmd/boom.go b/hrp/cmd/boom.go
index 41c67758..0bbc25c6 100644
--- a/hrp/cmd/boom.go
+++ b/hrp/cmd/boom.go
@@ -1,8 +1,10 @@
 package cmd
 
 import (
+	"syscall"
 	"time"
 
+	"github.com/rs/zerolog/log"
 	"github.com/spf13/cobra"
 
 	"github.com/httprunner/hrp"
@@ -19,6 +21,7 @@ var boomCmd = &cobra.Command{
   $ hrp boom examples/	# run testcases in specified folder`,
 	Args: cobra.MinimumNArgs(1),
 	PreRun: func(cmd *cobra.Command, args []string) {
+		setUlimit()
 		setLogLevel("WARN") // disable info logs for load testing
 	},
 	Run: func(cmd *cobra.Command, args []string) {
@@ -67,3 +70,26 @@ func init() {
 	boomCmd.Flags().StringVar(&prometheusPushgatewayURL, "prometheus-gateway", "", "Prometheus Pushgateway url.")
 	boomCmd.Flags().BoolVar(&disableConsoleOutput, "disable-console-output", false, "Disable console output.")
 }
+
+// set resource limit
+// ulimit -n 10240
+func setUlimit() {
+	var rLimit syscall.Rlimit
+	err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit)
+	if err != nil {
+		log.Error().Err(err).Msg("get ulimit failed")
+		return
+	}
+	log.Info().Uint64("limit", rLimit.Cur).Msg("get current ulimit")
+	if rLimit.Cur >= 10240 {
+		return
+	}
+
+	rLimit.Cur = 10240
+	log.Info().Uint64("limit", rLimit.Cur).Msg("set current ulimit")
+	err = syscall.Setrlimit(syscall.RLIMIT_NOFILE, &rLimit)
+	if err != nil {
+		log.Error().Err(err).Msg("set ulimit failed")
+		return
+	}
+}