From 20caa7d8ba32f2cdaad99cd7f4bb98839ca2e14b Mon Sep 17 00:00:00 2001
From: hujiebin <iamhujiebin@users.noreply.github.com>
Date: Wed, 1 Feb 2023 17:37:08 +0800
Subject: [PATCH] =?UTF-8?q?feat:=E6=B0=B4=E6=9E=9C=E6=9C=BA=E6=A8=A1?=
 =?UTF-8?q?=E6=8B=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 script/test_fruit_machine.go | 75 ++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 script/test_fruit_machine.go

diff --git a/script/test_fruit_machine.go b/script/test_fruit_machine.go
new file mode 100644
index 0000000..c61913f
--- /dev/null
+++ b/script/test_fruit_machine.go
@@ -0,0 +1,75 @@
+package main
+
+import (
+	"fmt"
+	"github.com/hilo-common/model"
+	"github.com/hilo-common/mysql"
+	"github.com/hilo-common/utils/jwt"
+	"io/ioutil"
+	"math/rand"
+	"net/http"
+	"strings"
+	"time"
+)
+
+func init() {
+	rand.Seed(time.Now().UnixNano())
+}
+
+var fruitUserCodes = []string{"1000653", "1000516", "1000675", "1000890", "1000755", "1000593", "1000629", "1000634", "1000765",
+	"1000591", "1000633", "1000224", "1000611", "1000689", "1000467", "1000384", "1000367", "1000623"}
+var fruitAmounts = []int{10, 100, 1000, 3000}
+
+func main() {
+	var users []model.User
+	if err := mysql.TestDB.Model(model.User{}).Where("code in ?", fruitUserCodes).Find(&users).Error; err != nil {
+		panic(err)
+	}
+	for {
+		times := 20 + rand.Intn(30)
+		for i := 0; i < times; i++ {
+
+			arr := []int{1, 2, 3, 4, 5, 6, 7, 8}
+			rand.Shuffle(len(arr), func(i, j int) {
+				arr[i], arr[j] = arr[j], arr[i]
+			})
+			for _, id := range arr {
+				if rand.Intn(100) < 30 {
+					continue
+				}
+				url := "https://test.apiv1.faceline.live/v1/fruitMachine/"
+				method := "POST"
+				amount := fruitAmounts[rand.Intn(len(fruitAmounts))]
+				payload := strings.NewReader(fmt.Sprintf("fruitId=%d&amount=%d", id, amount))
+
+				client := &http.Client{}
+				req, err := http.NewRequest(method, url, payload)
+
+				if err != nil {
+					fmt.Println(err)
+					return
+				}
+				u := users[rand.Intn(len(users))]
+				token, _ := jwt.GenerateToken(u.Id, u.ExternalId, "hiloApi")
+				req.Header.Add("nonce", "hilo")
+				req.Header.Add("token", token)
+				req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
+
+				res, err := client.Do(req)
+				if err != nil {
+					fmt.Println(err)
+					return
+				}
+				defer res.Body.Close()
+
+				body, err := ioutil.ReadAll(res.Body)
+				if err != nil {
+					fmt.Println(err)
+					return
+				}
+				fmt.Println(string(body))
+			}
+		}
+		//time.Sleep(time.Second * 55)
+	}
+}
-- 
2.22.0