131 lines
3.6 KiB
Go
131 lines
3.6 KiB
Go
package handler
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"time"
|
|
|
|
"go_rain_dtu/internal/dao"
|
|
"go_rain_dtu/internal/tcp"
|
|
"go_rain_dtu/pkg/logger"
|
|
)
|
|
|
|
type SensorHandler struct {
|
|
dao *dao.SensorDAO
|
|
}
|
|
|
|
func NewSensorHandler(dao *dao.SensorDAO) *SensorHandler {
|
|
return &SensorHandler{dao: dao}
|
|
}
|
|
|
|
func (h *SensorHandler) GetAggregatedData(w http.ResponseWriter, r *http.Request) {
|
|
interval := r.URL.Query().Get("interval")
|
|
if interval == "" {
|
|
interval = "1hour"
|
|
}
|
|
|
|
endTime := time.Now()
|
|
startTime := endTime.Add(-24 * time.Hour) // 默认显示24小时数据
|
|
|
|
if startStr := r.URL.Query().Get("start"); startStr != "" {
|
|
if t, err := time.Parse("2006-01-02T15:04", startStr); err == nil {
|
|
startTime = t
|
|
}
|
|
}
|
|
if endStr := r.URL.Query().Get("end"); endStr != "" {
|
|
if t, err := time.Parse("2006-01-02T15:04", endStr); err == nil {
|
|
endTime = t
|
|
}
|
|
}
|
|
|
|
data, err := h.dao.GetAggregatedData(startTime, endTime, interval)
|
|
if err != nil {
|
|
logger.Logger.Printf("获取聚合数据失败: %v", err)
|
|
http.Error(w, "服务器内部错误", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
// 使用自定义的编码器，设置时间格式
|
|
encoder := json.NewEncoder(w)
|
|
encoder.SetIndent("", " ")
|
|
if err := encoder.Encode(data); err != nil {
|
|
logger.Logger.Printf("JSON编码失败: %v", err)
|
|
http.Error(w, "服务器内部错误", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
// 处理静态文件
|
|
func (h *SensorHandler) ServeStatic(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/" {
|
|
http.ServeFile(w, r, "static/index.html")
|
|
return
|
|
}
|
|
http.FileServer(http.Dir("static")).ServeHTTP(w, r)
|
|
}
|
|
|
|
// GetConnectionStatus 返回当前TCP连接状态
|
|
func (h *SensorHandler) GetConnectionStatus(w http.ResponseWriter, r *http.Request) {
|
|
connected, ip, port := tcp.GetConnectionStatus()
|
|
|
|
status := struct {
|
|
Connected bool `json:"connected"`
|
|
IP string `json:"ip"`
|
|
Port string `json:"port"`
|
|
Count int `json:"count"` // 添加连接数量
|
|
}{
|
|
Connected: connected,
|
|
IP: ip,
|
|
Port: port,
|
|
Count: tcp.GetActiveConnectionCount(), // 获取活跃连接数
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
if err := json.NewEncoder(w).Encode(status); err != nil {
|
|
logger.Logger.Printf("JSON编码失败: %v", err)
|
|
http.Error(w, "服务器内部错误", http.StatusInternalServerError)
|
|
}
|
|
}
|
|
|
|
// GetLatestData 获取最新的聚合数据
|
|
func (h *SensorHandler) GetLatestData(w http.ResponseWriter, r *http.Request) {
|
|
interval := r.URL.Query().Get("interval")
|
|
if interval == "" {
|
|
interval = "5min" // 默认使用更细粒度的时间间隔
|
|
}
|
|
|
|
// 默认查询最近1小时的数据
|
|
endTime := time.Now()
|
|
startTime := endTime.Add(-1 * time.Hour)
|
|
|
|
// 如果提供了时间范围，则使用提供的范围
|
|
if startStr := r.URL.Query().Get("start"); startStr != "" {
|
|
if t, err := time.Parse(time.RFC3339, startStr); err == nil {
|
|
startTime = t
|
|
}
|
|
}
|
|
if endStr := r.URL.Query().Get("end"); endStr != "" {
|
|
if t, err := time.Parse(time.RFC3339, endStr); err == nil {
|
|
endTime = t
|
|
}
|
|
}
|
|
|
|
// 查询数据库获取最新数据
|
|
data, err := h.dao.GetAggregatedData(startTime, endTime, interval)
|
|
if err != nil {
|
|
logger.Logger.Printf("获取最新聚合数据失败: %v", err)
|
|
http.Error(w, "服务器内部错误", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
encoder := json.NewEncoder(w)
|
|
encoder.SetIndent("", " ")
|
|
if err := encoder.Encode(data); err != nil {
|
|
logger.Logger.Printf("JSON编码失败: %v", err)
|
|
http.Error(w, "服务器内部错误", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|