重命名包 client 为 edge;重命名包 server 为 gateway
This commit is contained in:
95
edge/report/report.go
Normal file
95
edge/report/report.go
Normal file
@@ -0,0 +1,95 @@
|
||||
package report
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"proxy-server/edge/core"
|
||||
"proxy-server/edge/env"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func Online(prov, city, isp string) (id int32, host string, err error) {
|
||||
|
||||
var ispInt = 0
|
||||
switch isp {
|
||||
case "电信":
|
||||
ispInt = 1
|
||||
case "联通":
|
||||
ispInt = 2
|
||||
case "移动":
|
||||
ispInt = 3
|
||||
}
|
||||
|
||||
body, err := json.Marshal(map[string]any{
|
||||
"prov": prov,
|
||||
"city": city,
|
||||
"isp": ispInt,
|
||||
"name": env.Name,
|
||||
"version": core.Version,
|
||||
})
|
||||
if err != nil {
|
||||
return 0, "", err
|
||||
}
|
||||
|
||||
req, err := http.NewRequest("POST", env.EndpointOnline, strings.NewReader(string(body)))
|
||||
if err != nil {
|
||||
return 0, "", fmt.Errorf("创建请求失败: %w", err)
|
||||
}
|
||||
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
return 0, "", fmt.Errorf("执行请求失败: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return 0, "", errors.New("状态码: " + resp.Status)
|
||||
}
|
||||
|
||||
bytes, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return 0, "", fmt.Errorf("读取响应失败: %w", err)
|
||||
}
|
||||
var respBody struct {
|
||||
Id int32 `json:"id"`
|
||||
Host string `json:"host"`
|
||||
}
|
||||
err = json.Unmarshal(bytes, &respBody)
|
||||
if err != nil {
|
||||
return 0, "", fmt.Errorf("解析响应失败: %w", err)
|
||||
}
|
||||
if respBody.Host == "" {
|
||||
return 0, "", errors.New("响应体为空")
|
||||
}
|
||||
|
||||
return respBody.Id, respBody.Host, nil
|
||||
}
|
||||
|
||||
func Offline() error {
|
||||
var bytes, err = json.Marshal(map[string]any{
|
||||
"name": env.Name,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
var body = strings.NewReader(string(bytes))
|
||||
|
||||
req, err := http.NewRequest("POST", env.EndpointOffline, body)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建请求失败: %w", err)
|
||||
}
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("执行请求失败: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return errors.New("状态码: " + resp.Status)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user