update file trigger

alibaba · Oct 18, 2024 · b55203c · b55203c
1 parent e2df97b
commit b55203c
Showing 1 changed file with 38 additions and 53 deletions.
diff --git a/test/engine/trigger/file.go b/test/engine/trigger/file.go
@@ -6,7 +6,6 @@ import (
 	"math/rand"
 	"os"
 	"path/filepath"
-	"sync"
 	"time"
 
 	"golang.org/x/time/rate"
@@ -16,64 +15,50 @@ import (
 
 // JSON template
 func GenerateRandomJSONLogToFile(ctx context.Context, speed, totalTime int, path string) (context.Context, error) {
-	numGoRoutines := (speed-1)/5 + 1
-	if numGoRoutines > 8 {
-		numGoRoutines = 8
+	JSONTemplates := []string{
+		`{"url": "POST /PutData?Category=paskdnkwja HTTP/1.1", "ip": "10.200.98.220", "user-agent": "aliyun-sdk-java", "request": {"status": "200", "latency": "18"}, "time": "12/Sep/2024:11:30:02"}\n`,
+		`{"url": "GET /PutData?Category=dwds HTTP/1.1", "ip": "10.7.159.1", "user-agent": "aliyun-sdk-java", "request": {"status": "404", "latency": "22024"}, "time": "06/Jun/2001:09:35:59"}\n`,
+		`{"url": "GET /PutData?Category=ubkjbkjgiuiu HTTP/1.1", "ip": "172.130.98.250", "user-agent": "aliyun-sdk-java", "request": {"301": "200", "latency": "12334"}, "time": "08/Aug/2018:10:30:28"}\n`,
+		`{"url": "POST /PutData?Category=oasjdwbjkbjkg HTTP/1.1", "ip": "1.20.9.220", "user-agent": "aliyun-sdk-java", "request": {"401": "200", "latency": "112"}, "time": "09/Sep/2024:15:25:28"}\n`,
+		`{"url": "POST /PutData?Category=asdjhoiasjdoOpLog HTTP/1.1", "ip": "172.168.0.1", "user-agent": "aliyun-sdk-java", "request": {"status": "200", "latency": "8815"}, "time": "01/Jan/2022:10:30:28"}\n`,
+	}
+	maxLen := 0
+	for j := 0; j < len(JSONTemplates); j++ {
+		if len(JSONTemplates[j]) > maxLen {
+			maxLen = len(JSONTemplates[j])
+		}
 	}
-	wg := sync.WaitGroup{}
-	for i := 0; i < numGoRoutines; i++ {
-		wg.Add(1)
-		go func(ctx context.Context, thread_id, speed, totalTime int, path string) {
-			defer wg.Done()
-			JSONTemplates := []string{
-				`{"url": "POST /PutData?Category=paskdnkwja HTTP/1.1", "ip": "10.200.98.220", "user-agent": "aliyun-sdk-java", "request": {"status": "200", "latency": "18"}, "time": "12/Sep/2024:11:30:02"}\n`,
-				`{"url": "GET /PutData?Category=dwds HTTP/1.1", "ip": "10.7.159.1", "user-agent": "aliyun-sdk-java", "request": {"status": "404", "latency": "22024"}, "time": "06/Jun/2001:09:35:59"}\n`,
-				`{"url": "GET /PutData?Category=ubkjbkjgiuiu HTTP/1.1", "ip": "172.130.98.250", "user-agent": "aliyun-sdk-java", "request": {"301": "200", "latency": "12334"}, "time": "08/Aug/2018:10:30:28"}\n`,
-				`{"url": "POST /PutData?Category=oasjdwbjkbjkg HTTP/1.1", "ip": "1.20.9.220", "user-agent": "aliyun-sdk-java", "request": {"401": "200", "latency": "112"}, "time": "09/Sep/2024:15:25:28"}\n`,
-				`{"url": "POST /PutData?Category=asdjhoiasjdoOpLog HTTP/1.1", "ip": "172.168.0.1", "user-agent": "aliyun-sdk-java", "request": {"status": "200", "latency": "8815"}, "time": "01/Jan/2022:10:30:28"}\n`,
-			}
-			maxLen := 0
-			for j := 0; j < len(JSONTemplates); j++ {
-				if len(JSONTemplates[j]) > maxLen {
-					maxLen = len(JSONTemplates[j])
-				}
-			}
 
-			// clear file
-			path = filepath.Clean(path)
-			path = fmt.Sprintf("%d_%s", thread_id, path)
-			path = filepath.Join(config.CaseHome, path)
-			fmt.Println(path)
-			_ = os.WriteFile(path, []byte{}, 0600)
-			file, _ := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600) // #nosec G304
+	// clear file
+	path = filepath.Clean(path)
+	path = filepath.Join(config.CaseHome, path)
+	fmt.Println(path)
+	_ = os.WriteFile(path, []byte{}, 0600)
+	file, _ := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600) // #nosec G304
 
-			limiter := rate.NewLimiter(rate.Limit(speed*1024*1024), maxLen)
+	limiter := rate.NewLimiter(rate.Limit(speed*1024*1024), maxLen)
 
-			timeout := time.After(time.Minute * time.Duration(totalTime))
+	timeout := time.After(time.Minute * time.Duration(totalTime))
 
-			rand.Seed(time.Now().UnixNano())
+	rand.Seed(time.Now().UnixNano())
 
-			for {
-				select {
-				// context is done
-				case <-ctx.Done():
-					// clear file
-					_ = file.Close()
-					return
-				// all time is done
-				case <-timeout:
-					// clear file
-					_ = file.Close()
-					return
-				default:
-					if limiter.AllowN(time.Now(), maxLen) {
-						randomIndex := rand.Intn(len(JSONTemplates)) // #nosec G404
-						_, _ = file.WriteString(JSONTemplates[randomIndex])
-					}
-				}
+	for {
+		select {
+		// context is done
+		case <-ctx.Done():
+			// clear file
+			_ = file.Close()
+			return ctx, nil
+		// all time is done
+		case <-timeout:
+			// clear file
+			_ = file.Close()
+			return ctx, nil
+		default:
+			if limiter.AllowN(time.Now(), maxLen) {
+				randomIndex := rand.Intn(len(JSONTemplates)) // #nosec G404
+				_, _ = file.WriteString(JSONTemplates[randomIndex])
 			}
-		}(ctx, i, speed/numGoRoutines, totalTime, path)
+		}
 	}
-	wg.Wait()
-	return ctx, nil
 }