122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
package httpreader
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
"FireLeave_tool/logger"
|
|
)
|
|
|
|
// CheckHTTPService 检查HTTP服务是否可用
|
|
func CheckHTTPService(port int, timeout time.Duration) bool {
|
|
client := &http.Client{
|
|
Timeout: timeout,
|
|
}
|
|
|
|
url := fmt.Sprintf("http://127.0.0.1:%d/health", port)
|
|
resp, err := client.Get(url)
|
|
if err != nil {
|
|
logger.Logger.Printf("HTTP服务健康检查失败 (Port=%d): %v", port, err)
|
|
return false
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
return resp.StatusCode == http.StatusOK
|
|
}
|
|
|
|
// CheckPortAvailable 检查端口是否可用
|
|
func CheckPortAvailable(port int) bool {
|
|
address := fmt.Sprintf(":%d", port)
|
|
listener, err := net.Listen("tcp", address)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
listener.Close()
|
|
return true
|
|
}
|
|
|
|
// WaitForHTTPService 等待HTTP服务启动
|
|
func WaitForHTTPService(port int, timeout time.Duration) bool {
|
|
deadline := time.Now().Add(timeout)
|
|
logger.Logger.Printf("等待HTTP服务启动 (Port=%d, Timeout=%v)", port, timeout)
|
|
|
|
attempt := 0
|
|
for time.Now().Before(deadline) {
|
|
attempt++
|
|
|
|
if CheckHTTPService(port, 2*time.Second) {
|
|
logger.Logger.Printf("HTTP服务启动成功 (Port=%d, 尝试次数=%d)", port, attempt)
|
|
return true
|
|
}
|
|
|
|
if attempt == 1 || attempt%5 == 0 {
|
|
logger.Logger.Printf("等待HTTP服务... (Port=%d, 尝试次数=%d)", port, attempt)
|
|
}
|
|
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
|
|
logger.Logger.Printf("HTTP服务启动超时 (Port=%d, 总尝试次数=%d)", port, attempt)
|
|
return false
|
|
}
|
|
|
|
// ValidateInferenceResult 验证推理结果数据
|
|
func ValidateInferenceResult(result *InferenceResult) error {
|
|
if result.Serial == "" {
|
|
return fmt.Errorf("serial字段不能为空")
|
|
}
|
|
if result.At <= 0 {
|
|
return fmt.Errorf("at时间戳无效")
|
|
}
|
|
|
|
// 验证参数数据
|
|
for i, param := range result.Params {
|
|
if param.Number < 0 {
|
|
return fmt.Errorf("参数[%d]的number不能为负数", i)
|
|
}
|
|
if param.ClassIdx < 0 {
|
|
return fmt.Errorf("参数[%d]的class_idx不能为负数", i)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetPersonCountFromResult 从推理结果中获取人员数量
|
|
func GetPersonCountFromResult(result *InferenceResult) int {
|
|
if result == nil || len(result.Params) == 0 {
|
|
return 0
|
|
}
|
|
|
|
// 查找人员类别的参数（假设 class_idx=1 表示人员）
|
|
for _, param := range result.Params {
|
|
if param.ClassIdx == 1 { // 人员类别
|
|
return param.Number
|
|
}
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
// CreateTestInferenceResult 创建测试用的推理结果（用于调试）
|
|
func CreateTestInferenceResult(deviceUUID string, personCount int) InferenceResult {
|
|
return InferenceResult{
|
|
Serial: deviceUUID,
|
|
At: time.Now().Unix(),
|
|
Type: 1,
|
|
Params: []struct {
|
|
ClassIdx int `json:"class_idx"`
|
|
Name string `json:"name"`
|
|
Number int `json:"number"`
|
|
}{
|
|
{
|
|
ClassIdx: 1,
|
|
Name: "person",
|
|
Number: personCount,
|
|
},
|
|
},
|
|
}
|
|
}
|