fix: http server and grpc with cert

hrp/internal/boomer/data/data.go

@@ -15,30 +15,48 @@
  *
  */
 
-// Package data provides convenience routines to access files in the data
-// directory.
 package data
 
 import (
+	"embed"
+	"os"
 	"path/filepath"
-	"runtime"
+
+	"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
 )
 
-// basepath is the root directory of this package.
+// hrpPath is .hrp directory under the user directory.
-var basepath string
+var hrpPath string
+
+//go:embed x509/*
+var x509Dir embed.FS
 
 func init() {
-	_, currentFile, _, _ := runtime.Caller(0)
+	home, err := os.UserHomeDir()
-	basepath = filepath.Dir(currentFile)
+	if err != nil {
+		return
+	}
+	hrpPath = filepath.Join(home, ".hrp")
+	_ = builtin.EnsureFolderExists(filepath.Join(hrpPath, "x509"))
+
 }
 
-// Path returns the absolute path the given relative file or directory path,
+// Path returns the absolute path the given relative file or directory path
-// relative to the google.golang.org/grpc/examples/data directory in the
+func Path(rel string) (destPath string) {
-// user's GOPATH.  If rel is already absolute, it is returned unmodified.
+	destPath = rel
-func Path(rel string) string {
+	if !filepath.IsAbs(rel) {
-	if filepath.IsAbs(rel) {
+		destPath = filepath.Join(hrpPath, rel)
-		return rel
+	}
+	if !builtin.IsFilePathExists(destPath) {
+		content, err := x509Dir.ReadFile(rel)
+		if err != nil {
+			return
 		}
 
-	return filepath.Join(basepath, rel)
+		err = os.WriteFile(destPath, content, 0o644)
+		if err != nil {
+			return
+		}
+	}
+	return
 }

hrp/server.go

@@ -2,6 +2,7 @@ package hrp
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"io/ioutil"
 	"log"
@@ -52,17 +53,13 @@ func parseBody(r *http.Request) (data map[string]interface{}, err error) {
 	return data, nil
 }
 
-func writeResponse(w http.ResponseWriter, status int, contentType string, body []byte) {
+func writeJSON(w http.ResponseWriter, body []byte, status int) {
-	w.Header().Set("Content-Type", contentType)
+	w.Header().Set("Content-Type", jsonContentType)
 	w.Header().Set("Content-Length", fmt.Sprintf("%d", len(body)))
 	w.WriteHeader(status)
 	w.Write(body)
 }
 
-func writeJSON(w http.ResponseWriter, body []byte, status int) {
-	writeResponse(w, status, jsonContentType, body)
-}
-
 type ServerCode int
 
 // server response code
@@ -169,37 +166,8 @@ func (api *apiHandler) Index(w http.ResponseWriter, r *http.Request) {
 
 func (api *apiHandler) Start(w http.ResponseWriter, r *http.Request) {
 	var resp *CommonResponseBody
-	data, err := parseBody(r)
+	var err error
-
+	defer func() {
-	req := StartRequestBody{
-		Profile: *api.boomer.GetProfile(),
-	}
-	err = mapstructure.Decode(data, &req)
-	if len(req.Other) > 0 {
-		keys := make([]string, 0, len(req.Other))
-		for k := range req.Other {
-			keys = append(keys, k)
-		}
-		resp = &CommonResponseBody{
-			ServerStatus: EnumAPIResponseParamError(fmt.Sprintf("failed to recognize params: %v", keys)),
-		}
-		body, _ := json.Marshal(resp)
-		writeJSON(w, body, http.StatusOK)
-		return
-	}
-	if req.TestCasePath == "" {
-		resp = &CommonResponseBody{
-			ServerStatus: EnumAPIResponseParamError(fmt.Sprint("missing testcases path")),
-		}
-		body, _ := json.Marshal(resp)
-		writeJSON(w, body, http.StatusOK)
-		return
-	}
-	paths := strings.Split(req.TestCasePath, ",")
-	api.boomer.SetTestCasesPath(paths)
-	if err == nil {
-		err = api.boomer.Start(&req.Profile)
-	}
 		if err != nil {
 			resp = &CommonResponseBody{
 				ServerStatus: EnumAPIResponseServerError(err.Error()),
@@ -211,6 +179,87 @@ func (api *apiHandler) Start(w http.ResponseWriter, r *http.Request) {
 		}
 		body, _ := json.Marshal(resp)
 		writeJSON(w, body, http.StatusOK)
+	}()
+
+	// parse body
+	data, err := parseBody(r)
+	if err != nil {
+		return
+	}
+	req := StartRequestBody{
+		Profile: *api.boomer.GetProfile(),
+	}
+	err = mapstructure.Decode(data, &req)
+	if err != nil {
+		return
+	}
+
+	// recognize invalid parameters
+	if len(req.Other) > 0 {
+		keys := make([]string, 0, len(req.Other))
+		for k := range req.Other {
+			keys = append(keys, k)
+		}
+		err = errors.New(fmt.Sprintf("failed to recognize params: %v", keys))
+		return
+	}
+
+	// parse testcase path
+	if req.TestCasePath == "" {
+		err = errors.New("missing testcases path")
+		return
+	}
+	paths := strings.Split(req.TestCasePath, ",")
+
+	// set testcase path
+	api.boomer.SetTestCasesPath(paths)
+
+	// start boomer with profile
+	err = api.boomer.Start(&req.Profile)
+}
+
+func (api *apiHandler) ReBalance(w http.ResponseWriter, r *http.Request) {
+	var resp *CommonResponseBody
+	var err error
+	defer func() {
+		if err != nil {
+			resp = &CommonResponseBody{
+				ServerStatus: EnumAPIResponseServerError(err.Error()),
+			}
+		} else {
+			resp = &CommonResponseBody{
+				ServerStatus: EnumAPIResponseSuccess,
+			}
+		}
+		body, _ := json.Marshal(resp)
+		writeJSON(w, body, http.StatusOK)
+	}()
+
+	// parse body
+	data, err := parseBody(r)
+	if err != nil {
+		return
+	}
+	req := RebalanceRequestBody{
+		Profile: *api.boomer.GetProfile(),
+	}
+	err = mapstructure.Decode(data, &req)
+	if err != nil {
+		return
+	}
+
+	// recognize invalid parameters
+	if len(req.Other) > 0 {
+		keys := make([]string, 0, len(req.Other))
+		for k := range req.Other {
+			keys = append(keys, k)
+		}
+		err = errors.New(fmt.Sprintf("failed to recognize params: %v", keys))
+		return
+	}
+
+	// rebalance boomer with profile
+	err = api.boomer.ReBalance(&req.Profile)
 }
 
 func (api *apiHandler) Stop(w http.ResponseWriter, r *http.Request) {
@@ -223,7 +272,8 @@ func (api *apiHandler) Stop(w http.ResponseWriter, r *http.Request) {
 	}
 
 	var resp *CommonResponseBody
-	err := api.boomer.Stop()
+	var err error
+	defer func() {
 		if err != nil {
 			resp = &CommonResponseBody{
 				ServerStatus: EnumAPIResponseStopError(err.Error()),
@@ -235,6 +285,10 @@ func (api *apiHandler) Stop(w http.ResponseWriter, r *http.Request) {
 		}
 		body, _ := json.Marshal(resp)
 		writeJSON(w, body, http.StatusOK)
+	}()
+
+	// stop boomer
+	err = api.boomer.Stop()
 }
 
 func (api *apiHandler) Quit(w http.ResponseWriter, r *http.Request) {
@@ -245,51 +299,18 @@ func (api *apiHandler) Quit(w http.ResponseWriter, r *http.Request) {
 			data[k] = v
 		}
 	}
-
+	defer func() {
 		resp := &CommonResponseBody{
 			ServerStatus: EnumAPIResponseSuccess,
 		}
 		body, _ := json.Marshal(resp)
 		writeJSON(w, body, http.StatusOK)
+	}()
+
+	// quit boomer
 	api.boomer.Quit()
 }
 
-func (api *apiHandler) ReBalance(w http.ResponseWriter, r *http.Request) {
-	var resp *CommonResponseBody
-	data, err := parseBody(r)
-
-	req := RebalanceRequestBody{
-		Profile: *api.boomer.GetProfile(),
-	}
-	err = mapstructure.Decode(data, &req)
-	if len(req.Other) > 0 {
-		keys := make([]string, 0, len(req.Other))
-		for k := range req.Other {
-			keys = append(keys, k)
-		}
-		resp = &CommonResponseBody{
-			ServerStatus: EnumAPIResponseParamError(fmt.Sprintf("failed to recognize params: %v", keys)),
-		}
-		body, _ := json.Marshal(resp)
-		writeJSON(w, body, http.StatusOK)
-		return
-	}
-	if err == nil {
-		err = api.boomer.ReBalance(&req.Profile)
-	}
-	if err != nil {
-		resp = &CommonResponseBody{
-			ServerStatus: EnumAPIResponseParamError(err.Error()),
-		}
-	} else {
-		resp = &CommonResponseBody{
-			ServerStatus: EnumAPIResponseSuccess,
-		}
-	}
-	body, _ := json.Marshal(resp)
-	writeJSON(w, body, http.StatusOK)
-}
-
 func (api *apiHandler) GetWorkersInfo(w http.ResponseWriter, r *http.Request) {
 	resp := &APIGetWorkersResponseBody{
 		ServerStatus: EnumAPIResponseSuccess,
@@ -305,9 +326,9 @@ func (api *apiHandler) Handler() http.Handler {
 
 	mux.HandleFunc("/", methods(api.Index, "GET"))
 	mux.HandleFunc("/start", methods(api.Start, "POST"))
+	mux.HandleFunc("/rebalance", methods(api.ReBalance, "POST"))
 	mux.HandleFunc("/stop", methods(api.Stop, "GET"))
 	mux.HandleFunc("/quit", methods(api.Quit, "GET"))
-	mux.HandleFunc("/rebalance", methods(api.ReBalance, "POST"))
 	mux.HandleFunc("/workers", methods(api.GetWorkersInfo, "GET"))
 
 	return mux