From 202fd4d29ccd7918fbd1d8042c5a2cd4923bb4b5 Mon Sep 17 00:00:00 2001
From: ben Gutier <xingctd@gmail.com>
Date: Sat, 9 Sep 2023 11:59:46 +0800
Subject: [PATCH] Initial commit

---
 source/main.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/source/main.go b/source/main.go
index ae2f8f7..ca2d8f5 100644
--- a/source/main.go
+++ b/source/main.go
@@ -12,11 +12,13 @@ import (
 	"math/rand"
 	"net/http"
 	"os"
+	"path/filepath"
 	"strconv"
 	"strings"
 	"sync"
 	"time"
 )
+
 type Config struct {
 	Server struct {
 		Domain   string `json:"domain"`
@@ -31,12 +33,14 @@ type Config struct {
 	} `json:"copilot_config"`
 	Verification string `json:"verification"`
 }
+
 var (
 	requestCountMutex sync.Mutex
 	requestCount      int
 	successCount      int
 	configFile        = Config{}
 )
+
 func main() {
 	gin.SetMode(gin.ReleaseMode)
 	gin.DefaultWriter = io.Discard
@@ -152,9 +156,9 @@ func initConfig() Config {
 	// 设置日志输出到文件
 	log.SetOutput(logFile)
 	// 读取配置文件
-	configFile, err := os.Open("./config.json")
+	exePath, err := os.Executable()
 	if err != nil {
-		panic("file \"./config.json\" not found")
+		panic(err)
 	}
 	// 获取可执行文件所在目录
 	exeDir := filepath.Dir(exePath)
@@ -218,7 +222,6 @@ func showRequestCount() {
 	countStr := fmt.Sprintf("\x1b[34m%d\x1b[0m", count)
 	// 绿色文本
 	successCountStr := fmt.Sprintf("\x1b[32m%d\x1b[0m", sCount)
-	// 红色文本
 	failureCountStr := fmt.Sprintf("\x1b[31m%d\x1b[0m", count-sCount)
 	fmt.Printf("\033[1G%s - Total Count: %s | Success Count %s | Fail Count %s   ", timeStr, countStr, successCountStr, failureCountStr)
 }