feat: 核心电费查询 Go 服务
- FasterLZU 登录 + xiaofubao 6 步电费链路 - shiroJID 会话缓存 + 每日记录存储 - 定时任务: 每天 00:00 记录, 失败重试 5 次 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
82
go/crypto.go
Normal file
82
go/crypto.go
Normal file
@@ -0,0 +1,82 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/aes"
|
||||||
|
"crypto/cipher"
|
||||||
|
"crypto/md5"
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// crypto.go - AES-CBC 加密/解密 + MD5 签名（与 FasterLZU APP 行为一致）
|
||||||
|
// 密钥必须通过环境变量 LZU_AES_KEY / LZU_MD5_KEY 提供（16 字节）
|
||||||
|
|
||||||
|
var (
|
||||||
|
aesKey = envOr("LZU_AES_KEY", "") // 16 字节，由部署环境注入
|
||||||
|
md5Key = envOr("LZU_MD5_KEY", "")
|
||||||
|
)
|
||||||
|
|
||||||
|
// aesEncrypt 使用 AES-CBC，IV=key，零填充，输出小写 hex
|
||||||
|
func aesEncrypt(plaintext string) (string, error) {
|
||||||
|
key := []byte(aesKey)
|
||||||
|
if len(key) != 16 {
|
||||||
|
return "", fmt.Errorf("AES key 长度不是 16 字节")
|
||||||
|
}
|
||||||
|
block, err := aes.NewCipher(key)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
data := []byte(plaintext)
|
||||||
|
// 零填充到 blockSize 的整数倍
|
||||||
|
blockSize := aes.BlockSize
|
||||||
|
paddedLen := len(data)
|
||||||
|
if paddedLen%blockSize != 0 {
|
||||||
|
paddedLen = len(data) + (blockSize - len(data)%blockSize)
|
||||||
|
}
|
||||||
|
padded := make([]byte, paddedLen)
|
||||||
|
copy(padded, data)
|
||||||
|
// 剩余部分保持 0x00（零填充）
|
||||||
|
|
||||||
|
iv := key
|
||||||
|
mode := cipher.NewCBCEncrypter(block, iv)
|
||||||
|
ciphertext := make([]byte, paddedLen)
|
||||||
|
mode.CryptBlocks(ciphertext, padded)
|
||||||
|
return hex.EncodeToString(ciphertext), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// aesDecrypt AES-CBC 解密，去除尾部零填充
|
||||||
|
func aesDecrypt(hexStr string) (string, error) {
|
||||||
|
key := []byte(aesKey)
|
||||||
|
if len(key) != 16 {
|
||||||
|
return "", fmt.Errorf("AES key 长度不是 16 字节")
|
||||||
|
}
|
||||||
|
encrypted, err := hex.DecodeString(hexStr)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
block, err := aes.NewCipher(key)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
iv := key
|
||||||
|
mode := cipher.NewCBCDecrypter(block, iv)
|
||||||
|
decrypted := make([]byte, len(encrypted))
|
||||||
|
mode.CryptBlocks(decrypted, encrypted)
|
||||||
|
// 去除尾部零填充
|
||||||
|
validLen := len(decrypted)
|
||||||
|
for i := len(decrypted) - 1; i >= 0 && i >= len(decrypted)-16; i-- {
|
||||||
|
if decrypted[i] != 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
validLen--
|
||||||
|
}
|
||||||
|
return string(decrypted[:validLen]), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// md5Encrypt 签名: 参数值以 | 连接，末尾拼接 key，返回 MD5 hex
|
||||||
|
func md5Encrypt(params []string, key string) string {
|
||||||
|
data := strings.Join(params, "|") + "|" + key
|
||||||
|
sum := md5.Sum([]byte(data))
|
||||||
|
return hex.EncodeToString(sum[:])
|
||||||
|
}
|
||||||
431
go/electric.go
Normal file
431
go/electric.go
Normal file
@@ -0,0 +1,431 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// electric.go - xiaofubao 一码通电费查询链路 + shiroJID 缓存
|
||||||
|
// 对应 Python lzu_electric_full.py 的 6 步流程
|
||||||
|
|
||||||
|
// 平台常量（来自环境变量，默认值对齐 .env）
|
||||||
|
var (
|
||||||
|
xfPlatform = envOr("XF_PLATFORM", "QHIT_EDU")
|
||||||
|
xfAuthAppid = envOr("XF_AUTH_APPID", "2606161276416311297")
|
||||||
|
xfSchoolCode = envOr("XF_SCHOOL_CODE", "10730")
|
||||||
|
xfYmAppid = envOr("XF_YM_APPID", "1810181825222034")
|
||||||
|
xfServiceID = envOr("LZU_SERVICE_ID", "29615")
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
browserUA = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/151.0.0.0 Safari/537.36"
|
||||||
|
// 查询互斥锁：防止定时任务与手动刷新并发登录
|
||||||
|
queryLock = make(chan struct{}, 1)
|
||||||
|
)
|
||||||
|
|
||||||
|
// roomInfo 绑定房间信息
|
||||||
|
type roomInfo struct {
|
||||||
|
AreaID string `json:"areaId"`
|
||||||
|
BuildingCode string `json:"buildingCode"`
|
||||||
|
FloorCode string `json:"floorCode"`
|
||||||
|
RoomCode string `json:"roomCode"`
|
||||||
|
RoomName string `json:"roomName"`
|
||||||
|
AreaName string `json:"areaName"`
|
||||||
|
BuildingName string `json:"buildingName"`
|
||||||
|
FloorName string `json:"floorName"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// cacheData 缓存结构（shiroJID 会话 + 房间）
|
||||||
|
type cacheData struct {
|
||||||
|
AppShiro string `json:"app_shiro"`
|
||||||
|
Room roomInfo `json:"room"`
|
||||||
|
CachedAt int64 `json:"cached_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// electricResult 查询结果
|
||||||
|
type electricResult struct {
|
||||||
|
OK bool `json:"ok"`
|
||||||
|
Room string `json:"room"`
|
||||||
|
Surplus float64 `json:"surplus"`
|
||||||
|
Amount float64 `json:"amount"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Source string `json:"source"`
|
||||||
|
Time int64 `json:"time"`
|
||||||
|
Error string `json:"error,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var cacheFilePath = envOr("CACHE_FILE", "data/electric_cache.json")
|
||||||
|
|
||||||
|
// ---------- 缓存 ----------
|
||||||
|
func loadCache() (*cacheData, error) {
|
||||||
|
b, err := os.ReadFile(cacheFilePath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var c cacheData
|
||||||
|
if err := json.Unmarshal(b, &c); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &c, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func saveCache(c *cacheData) error {
|
||||||
|
if err := os.MkdirAll(dirName(cacheFilePath), 0o755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
b, _ := json.MarshalIndent(c, "", " ")
|
||||||
|
return os.WriteFile(cacheFilePath, b, 0o644)
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearCache() {
|
||||||
|
_ = os.Remove(cacheFilePath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func dirName(p string) string {
|
||||||
|
i := strings.LastIndexByte(p, '/')
|
||||||
|
if i < 0 {
|
||||||
|
i = strings.LastIndexByte(p, '\\')
|
||||||
|
}
|
||||||
|
if i < 0 {
|
||||||
|
return "."
|
||||||
|
}
|
||||||
|
return p[:i]
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- HTTP 辅助 ----------
|
||||||
|
type xfSession struct {
|
||||||
|
client *http.Client
|
||||||
|
cookie string // shiroJID
|
||||||
|
}
|
||||||
|
|
||||||
|
func newXfSession(shiroJID string) *xfSession {
|
||||||
|
return &xfSession{client: &http.Client{}, cookie: shiroJID}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *xfSession) postForm(urlStr, body, referer string) ([]byte, http.Header, error) {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
req, err := http.NewRequestWithContext(ctx, "POST", urlStr, strings.NewReader(body))
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded;charset=UTF-8")
|
||||||
|
req.Header.Set("User-Agent", browserUA)
|
||||||
|
req.Header.Set("X-Requested-With", "XMLHttpRequest")
|
||||||
|
req.Header.Set("Accept", "application/json, text/plain, */*")
|
||||||
|
if referer != "" {
|
||||||
|
req.Header.Set("Referer", referer)
|
||||||
|
}
|
||||||
|
if s.cookie != "" {
|
||||||
|
req.AddCookie(&http.Cookie{Name: "shiroJID", Value: s.cookie})
|
||||||
|
}
|
||||||
|
resp, err := s.client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||||
|
return bodyBytes, resp.Header, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *xfSession) get(urlStr string, allowRedirect bool) (*http.Response, error) {
|
||||||
|
client := s.client
|
||||||
|
if !allowRedirect {
|
||||||
|
// 不跟随重定向
|
||||||
|
client = &http.Client{
|
||||||
|
CheckRedirect: func(req *http.Request, via []*http.Request) error {
|
||||||
|
return http.ErrUseLastResponse
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
req, err := http.NewRequestWithContext(ctx, "GET", urlStr, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
req.Header.Set("User-Agent", browserUA)
|
||||||
|
req.Header.Set("X-Requested-With", "XMLHttpRequest")
|
||||||
|
if s.cookie != "" {
|
||||||
|
req.AddCookie(&http.Cookie{Name: "shiroJID", Value: s.cookie})
|
||||||
|
}
|
||||||
|
return client.Do(req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- 查询链路 ----------
|
||||||
|
|
||||||
|
// step1GetCodeV2 ST票据 -> ymCode（无需 cookie）
|
||||||
|
func step1GetCodeV2(startURL string) (string, error) {
|
||||||
|
cb := url.QueryEscape(startURL)
|
||||||
|
u := fmt.Sprintf("https://auth.xiaofubao.com/authoriz/getCodeV2?bindSkip=1&ymAppId=%s&authType=3&authAppid=%s&callbackUrl=%s",
|
||||||
|
xfYmAppid, xfAuthAppid, cb)
|
||||||
|
s := newXfSession("")
|
||||||
|
resp, err := s.get(u, false)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
if resp.StatusCode != http.StatusFound {
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
return "", fmt.Errorf("getCodeV2 状态码异常: %d %s", resp.StatusCode, string(body))
|
||||||
|
}
|
||||||
|
loc := resp.Header.Get("Location")
|
||||||
|
re := regexp.MustCompile(`ymCode=([0-9a-f]+)`)
|
||||||
|
m := re.FindStringSubmatch(loc)
|
||||||
|
if len(m) < 2 {
|
||||||
|
return "", fmt.Errorf("getCodeV2 未返回 ymCode: %s", loc)
|
||||||
|
}
|
||||||
|
return m[1], nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// extractShiroJID 从 Set-Cookie 头提取最终生效的 shiroJID。
|
||||||
|
// 服务端可能返回多值: 设置 -> deleteMe(删除) -> 重设，须按顺序取最后一个有效值。
|
||||||
|
func extractShiroJID(hdr http.Header) (string, error) {
|
||||||
|
found := ""
|
||||||
|
for _, c := range hdr.Values("Set-Cookie") {
|
||||||
|
parts := strings.Split(c, ";")
|
||||||
|
kv := strings.SplitN(strings.TrimSpace(parts[0]), "=", 2)
|
||||||
|
if len(kv) != 2 || kv[0] != "shiroJID" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if kv[1] == "deleteMe" {
|
||||||
|
found = "" // 删除之前的
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
found = kv[1]
|
||||||
|
}
|
||||||
|
if found == "" {
|
||||||
|
return "", errors.New("未返回有效的 shiroJID cookie")
|
||||||
|
}
|
||||||
|
return found, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// step2GetUserByCodeV2 ymCode -> 用户信息 + webapp shiroJID
|
||||||
|
func step2GetUserByCodeV2(ymCode, startURL string) (string, error) {
|
||||||
|
s := newXfSession("")
|
||||||
|
body := fmt.Sprintf("code=%s&platform=%s", ymCode, xfPlatform)
|
||||||
|
_, hdr, err := s.postForm("https://webapp.xiaofubao.com/user/getUserByCodeV2", body, startURL)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return extractShiroJID(hdr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// flexInt 兼容 JSON 中字符串或数字的 statusCode
|
||||||
|
type flexInt int
|
||||||
|
|
||||||
|
func (f *flexInt) UnmarshalJSON(b []byte) error {
|
||||||
|
s := strings.Trim(string(b), `"`)
|
||||||
|
if s == "" {
|
||||||
|
*f = 0
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var n int
|
||||||
|
if _, err := fmt.Sscanf(s, "%d", &n); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*f = flexInt(n)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// step3GetUserCode 生成一次性 code（必须带 referer）
|
||||||
|
func step3GetUserCode(shiroJID, startURL string) (string, error) {
|
||||||
|
s := newXfSession(shiroJID)
|
||||||
|
body := fmt.Sprintf("ymAppId=%s&ymAuthType=2&platform=%s", xfYmAppid, xfPlatform)
|
||||||
|
b, _, err := s.postForm("https://webapp.xiaofubao.com/routeauth/auth/route/user/getUserCode", body, startURL)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
var res struct {
|
||||||
|
StatusCode flexInt `json:"statusCode"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
Data string `json:"data"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(b, &res); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if res.StatusCode != 0 {
|
||||||
|
return "", fmt.Errorf("getUserCode 失败 %d: %s", res.StatusCode, res.Message)
|
||||||
|
}
|
||||||
|
return res.Data, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// step4GetUser4Authorize code -> application 域 shiroJID
|
||||||
|
func step4GetUser4Authorize(code string) (string, error) {
|
||||||
|
s := newXfSession("")
|
||||||
|
body := fmt.Sprintf("code=%s&authType=2&platform=%s", code, xfPlatform)
|
||||||
|
_, hdr, err := s.postForm("https://application.xiaofubao.com/app/login/getUser4Authorize", body, "https://application.xiaofubao.com/")
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return extractShiroJID(hdr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// step5QueryBind 查询绑定房间
|
||||||
|
func step5QueryBind(shiroJID string) (*roomInfo, error) {
|
||||||
|
s := newXfSession(shiroJID)
|
||||||
|
body := fmt.Sprintf("bindType=3&platform=%s", xfPlatform)
|
||||||
|
b, _, err := s.postForm("https://application.xiaofubao.com/app/electric/queryBind", body, "https://application.xiaofubao.com/")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var res struct {
|
||||||
|
StatusCode flexInt `json:"statusCode"`
|
||||||
|
Rows []struct {
|
||||||
|
AreaID string `json:"areaId"`
|
||||||
|
BuildingCode string `json:"buildingCode"`
|
||||||
|
FloorCode string `json:"floorCode"`
|
||||||
|
RoomCode string `json:"roomCode"`
|
||||||
|
RoomName string `json:"roomName"`
|
||||||
|
AreaName string `json:"areaName"`
|
||||||
|
BuildingName string `json:"buildingName"`
|
||||||
|
FloorName string `json:"floorName"`
|
||||||
|
} `json:"rows"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(b, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.StatusCode != 0 || len(res.Rows) == 0 {
|
||||||
|
return nil, fmt.Errorf("queryBind 失败: %s", string(b))
|
||||||
|
}
|
||||||
|
r := res.Rows[0]
|
||||||
|
return &roomInfo{
|
||||||
|
AreaID: r.AreaID, BuildingCode: r.BuildingCode, FloorCode: r.FloorCode,
|
||||||
|
RoomCode: r.RoomCode, RoomName: r.RoomName, AreaName: r.AreaName,
|
||||||
|
BuildingName: r.BuildingName, FloorName: r.FloorName,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// step6QuerySurplus 查询剩余电量
|
||||||
|
func step6QuerySurplus(shiroJID string, room *roomInfo) (float64, float64, string, error) {
|
||||||
|
s := newXfSession(shiroJID)
|
||||||
|
body := fmt.Sprintf("areaId=%s&buildingCode=%s&floorCode=%s&roomCode=%s&platform=%s",
|
||||||
|
room.AreaID, room.BuildingCode, room.FloorCode, room.RoomCode, xfPlatform)
|
||||||
|
b, _, err := s.postForm("https://application.xiaofubao.com/app/electric/queryISIMSRoomSurplus", body, "https://application.xiaofubao.com/")
|
||||||
|
if err != nil {
|
||||||
|
return 0, 0, "", err
|
||||||
|
}
|
||||||
|
var res struct {
|
||||||
|
StatusCode flexInt `json:"statusCode"`
|
||||||
|
Data struct {
|
||||||
|
DisplayRoomName string `json:"displayRoomName"`
|
||||||
|
SurplusList []struct {
|
||||||
|
Surplus float64 `json:"surplus"`
|
||||||
|
Amount float64 `json:"amount"`
|
||||||
|
RoomStatus string `json:"roomStatus"`
|
||||||
|
} `json:"surplusList"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(b, &res); err != nil {
|
||||||
|
return 0, 0, "", err
|
||||||
|
}
|
||||||
|
if res.StatusCode != 0 || len(res.Data.SurplusList) == 0 {
|
||||||
|
return 0, 0, "", fmt.Errorf("queryISIMSRoomSurplus 失败: %s", string(b))
|
||||||
|
}
|
||||||
|
item := res.Data.SurplusList[0]
|
||||||
|
return item.Surplus, item.Amount, item.RoomStatus, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- 完整登录流程 ----------
|
||||||
|
func fullLoginFlow() (string, *roomInfo, *electricResult, error) {
|
||||||
|
lzu := newLzuClient(envOr("LZU_USERNAME", ""), envOr("LZU_PASSWORD", ""))
|
||||||
|
if lzu.username == "" || lzu.password == "" {
|
||||||
|
return "", nil, nil, errors.New("缺少 LZU_USERNAME / LZU_PASSWORD 配置")
|
||||||
|
}
|
||||||
|
if _, _, err := lzu.Login(); err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
st, err := lzu.GetSt(xfServiceID)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
startURL := fmt.Sprintf("https://webapp.xiaofubao.com/card/apps_all.shtml?authAppid=%s&platform=%s&schoolCode=%s&PersonID=%s&st=%s&ticket=%s",
|
||||||
|
xfAuthAppid, xfPlatform, xfSchoolCode, lzu.username, st, st)
|
||||||
|
|
||||||
|
ymCode, err := step1GetCodeV2(startURL)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
webappShiro, err := step2GetUserByCodeV2(ymCode, startURL)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
code, err := step3GetUserCode(webappShiro, startURL)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
appShiro, err := step4GetUser4Authorize(code)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
room, err := step5QueryBind(appShiro)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
surplus, amount, status, err := step6QuerySurplus(appShiro, room)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, nil, err
|
||||||
|
}
|
||||||
|
result := &electricResult{
|
||||||
|
OK: true, Room: displayRoomName(room), Surplus: round2(surplus),
|
||||||
|
Amount: round2(amount), Status: status, Source: "login", Time: time.Now().Unix(),
|
||||||
|
}
|
||||||
|
return appShiro, room, result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- 查询入口（缓存优先） ----------
|
||||||
|
func queryElectric() *electricResult {
|
||||||
|
queryLock <- struct{}{}
|
||||||
|
defer func() { <-queryLock }()
|
||||||
|
|
||||||
|
// 1. 尝试缓存
|
||||||
|
if c, err := loadCache(); err == nil {
|
||||||
|
if time.Now().Unix()-c.CachedAt < cacheTTL() {
|
||||||
|
surplus, amount, status, err := step6QuerySurplus(c.AppShiro, &c.Room)
|
||||||
|
if err == nil {
|
||||||
|
return &electricResult{
|
||||||
|
OK: true, Room: displayRoomName(&c.Room), Surplus: round2(surplus),
|
||||||
|
Amount: round2(amount), Status: status, Source: "cache", Time: time.Now().Unix(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fmt.Println("[缓存] 会话失效:", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. 完整登录
|
||||||
|
appShiro, room, result, err := fullLoginFlow()
|
||||||
|
if err != nil {
|
||||||
|
clearCache()
|
||||||
|
return &electricResult{OK: false, Error: err.Error()}
|
||||||
|
}
|
||||||
|
_ = saveCache(&cacheData{AppShiro: appShiro, Room: *room, CachedAt: time.Now().Unix()})
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func displayRoomName(r *roomInfo) string {
|
||||||
|
return fmt.Sprintf("%s%s%s%s", r.AreaName, r.BuildingName, r.FloorName, r.RoomName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func round2(f float64) float64 {
|
||||||
|
return float64(int(f*100+0.5)) / 100
|
||||||
|
}
|
||||||
|
|
||||||
|
func cacheTTL() int64 {
|
||||||
|
ttl := int64(6 * 24 * 3600)
|
||||||
|
if v := envOr("XF_CACHE_TTL", ""); v != "" {
|
||||||
|
var n int64
|
||||||
|
if _, err := fmt.Sscanf(v, "%d", &n); err == nil && n > 0 {
|
||||||
|
ttl = n
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ttl
|
||||||
|
}
|
||||||
253
go/history.go
Normal file
253
go/history.go
Normal file
@@ -0,0 +1,253 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"os"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// history.go - 每日电费记录存储（JSON 文件）与耗电统计
|
||||||
|
|
||||||
|
type record struct {
|
||||||
|
Date string `json:"date"` // YYYY-MM-DD
|
||||||
|
Surplus float64 `json:"surplus"`
|
||||||
|
Amount float64 `json:"amount"`
|
||||||
|
Source string `json:"source"` // auto / manual
|
||||||
|
CreatedAt int64 `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type dailyStat struct {
|
||||||
|
Date string `json:"date"`
|
||||||
|
Surplus float64 `json:"surplus"`
|
||||||
|
Amount float64 `json:"amount"`
|
||||||
|
Usage *float64 `json:"usage"` // nil 表示无前一天基线
|
||||||
|
}
|
||||||
|
|
||||||
|
// monthSummary 月度汇总
|
||||||
|
type monthSummary struct {
|
||||||
|
Month string `json:"month"` // YYYY-MM
|
||||||
|
RecordCount int `json:"record_count"` // 记录天数
|
||||||
|
StartSurplus *float64 `json:"start_surplus"` // 月初首次电量
|
||||||
|
EndSurplus *float64 `json:"end_surplus"` // 月末末次电量
|
||||||
|
TotalUsage *float64 `json:"total_usage"` // 月总耗电
|
||||||
|
AvgUsage *float64 `json:"avg_usage"` // 日均耗电
|
||||||
|
RechargeDays int `json:"recharge_days"` // 疑似充值天数（电量回升）
|
||||||
|
MaxUsage *float64 `json:"max_usage"` // 单日最大耗电
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
historyPath = envOr("HISTORY_FILE", "data/electric_history.json")
|
||||||
|
historyLock sync.Mutex
|
||||||
|
historyLoaded bool
|
||||||
|
historyCache []record
|
||||||
|
)
|
||||||
|
|
||||||
|
// loadHistory 读取历史记录
|
||||||
|
func loadHistory() []record {
|
||||||
|
historyLock.Lock()
|
||||||
|
defer historyLock.Unlock()
|
||||||
|
if historyLoaded {
|
||||||
|
return historyCache
|
||||||
|
}
|
||||||
|
b, err := os.ReadFile(historyPath)
|
||||||
|
if err != nil {
|
||||||
|
historyLoaded = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
_ = json.Unmarshal(b, &historyCache)
|
||||||
|
historyLoaded = true
|
||||||
|
return historyCache
|
||||||
|
}
|
||||||
|
|
||||||
|
// saveHistory 持久化历史记录
|
||||||
|
func saveHistory(records []record) error {
|
||||||
|
historyLock.Lock()
|
||||||
|
defer historyLock.Unlock()
|
||||||
|
if err := os.MkdirAll(dirName(historyPath), 0o755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
b, _ := json.MarshalIndent(records, "", " ")
|
||||||
|
if err := os.WriteFile(historyPath, b, 0o644); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
historyCache = records
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// hasRecord 判断指定日期是否已有记录
|
||||||
|
func hasRecord(dateStr string) bool {
|
||||||
|
for _, r := range loadHistory() {
|
||||||
|
if r.Date == dateStr {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// addRecord 写入（或覆盖）一条记录
|
||||||
|
func addRecord(dateStr string, surplus, amount float64, source string) error {
|
||||||
|
records := loadHistory()
|
||||||
|
updated := false
|
||||||
|
for i, r := range records {
|
||||||
|
if r.Date == dateStr {
|
||||||
|
records[i] = record{Date: dateStr, Surplus: surplus, Amount: amount, Source: source, CreatedAt: time.Now().Unix()}
|
||||||
|
updated = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !updated {
|
||||||
|
records = append(records, record{Date: dateStr, Surplus: surplus, Amount: amount, Source: source, CreatedAt: time.Now().Unix()})
|
||||||
|
}
|
||||||
|
return saveHistory(records)
|
||||||
|
}
|
||||||
|
|
||||||
|
// getRecords 返回降序（新→旧）记录列表
|
||||||
|
func getRecords() []record {
|
||||||
|
records := loadHistory()
|
||||||
|
sort.Slice(records, func(i, j int) bool {
|
||||||
|
return records[i].Date > records[j].Date
|
||||||
|
})
|
||||||
|
return records
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildDailyStats 按日期升序生成每日统计（含日耗电 = 前一天剩余 - 当天剩余）
|
||||||
|
// 注意: 耗电计算跨月时使用上月末记录作为基线（若存在）
|
||||||
|
func buildDailyStats(records []record, month string) []dailyStat {
|
||||||
|
all := loadHistory()
|
||||||
|
// 若指定月份，额外包含上一月最后一条记录作为基线（不计入本月输出）
|
||||||
|
var prevMonthLast *record
|
||||||
|
if month != "" {
|
||||||
|
prevPrefix := prevMonth(month) + "-"
|
||||||
|
var last *record
|
||||||
|
for i := range all {
|
||||||
|
if strings.HasPrefix(all[i].Date, prevPrefix) {
|
||||||
|
if last == nil || all[i].Date > last.Date {
|
||||||
|
last = &all[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
prevMonthLast = last
|
||||||
|
}
|
||||||
|
|
||||||
|
// 筛选目标记录
|
||||||
|
var target []record
|
||||||
|
if month != "" {
|
||||||
|
target = filterByMonth(records, month)
|
||||||
|
} else {
|
||||||
|
target = records
|
||||||
|
}
|
||||||
|
sort.Slice(target, func(i, j int) bool {
|
||||||
|
return target[i].Date < target[j].Date
|
||||||
|
})
|
||||||
|
|
||||||
|
var daily []dailyStat
|
||||||
|
var prevSurplus *float64
|
||||||
|
if prevMonthLast != nil {
|
||||||
|
prevSurplus = &prevMonthLast.Surplus
|
||||||
|
}
|
||||||
|
for _, r := range target {
|
||||||
|
var usage *float64
|
||||||
|
if prevSurplus != nil {
|
||||||
|
u := round2(*prevSurplus - r.Surplus)
|
||||||
|
if u < 0 {
|
||||||
|
u = 0 // 充值/异常归零
|
||||||
|
}
|
||||||
|
usage = &u
|
||||||
|
}
|
||||||
|
daily = append(daily, dailyStat{Date: r.Date, Surplus: r.Surplus, Amount: r.Amount, Usage: usage})
|
||||||
|
s := r.Surplus
|
||||||
|
prevSurplus = &s
|
||||||
|
}
|
||||||
|
return daily
|
||||||
|
}
|
||||||
|
|
||||||
|
// prevMonth 返回上个月的 YYYY-MM
|
||||||
|
func prevMonth(month string) string {
|
||||||
|
t, err := time.Parse("2006-01", month)
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return t.AddDate(0, -1, 0).Format("2006-01")
|
||||||
|
}
|
||||||
|
|
||||||
|
// filterByMonth 按月份(YYYY-MM)筛选记录
|
||||||
|
func filterByMonth(records []record, month string) []record {
|
||||||
|
if month == "" {
|
||||||
|
return records
|
||||||
|
}
|
||||||
|
prefix := month + "-"
|
||||||
|
var out []record
|
||||||
|
for _, r := range records {
|
||||||
|
if strings.HasPrefix(r.Date, prefix) {
|
||||||
|
out = append(out, r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildMonthSummary 生成某月的月度汇总
|
||||||
|
func buildMonthSummary(daily []dailyStat, month string) *monthSummary {
|
||||||
|
if len(daily) == 0 {
|
||||||
|
return &monthSummary{Month: month, RecordCount: 0}
|
||||||
|
}
|
||||||
|
// daily 已是升序
|
||||||
|
m := month
|
||||||
|
if m == "" {
|
||||||
|
m = daily[0].Date[:7]
|
||||||
|
}
|
||||||
|
start := daily[0].Surplus
|
||||||
|
end := daily[len(daily)-1].Surplus
|
||||||
|
|
||||||
|
var total float64
|
||||||
|
var max float64
|
||||||
|
rechargeDays := 0
|
||||||
|
countWithUsage := 0
|
||||||
|
for _, d := range daily {
|
||||||
|
if d.Usage != nil {
|
||||||
|
total += *d.Usage
|
||||||
|
if *d.Usage > max {
|
||||||
|
max = *d.Usage
|
||||||
|
}
|
||||||
|
countWithUsage++
|
||||||
|
}
|
||||||
|
// 疑似充值：当天剩余比前一天高
|
||||||
|
}
|
||||||
|
// 充值与最大耗电基于原始升序记录重新精确计算
|
||||||
|
if len(daily) >= 2 {
|
||||||
|
rechargeDays = countRecharges(daily)
|
||||||
|
}
|
||||||
|
|
||||||
|
var totalP, avgP, maxP, startP, endP *float64
|
||||||
|
startP, endP = &start, &end
|
||||||
|
totalP = &total
|
||||||
|
if countWithUsage > 0 {
|
||||||
|
avg := round2(total / float64(countWithUsage))
|
||||||
|
avgP = &avg
|
||||||
|
}
|
||||||
|
maxP = &max
|
||||||
|
|
||||||
|
return &monthSummary{
|
||||||
|
Month: m,
|
||||||
|
RecordCount: len(daily),
|
||||||
|
StartSurplus: startP,
|
||||||
|
EndSurplus: endP,
|
||||||
|
TotalUsage: totalP,
|
||||||
|
AvgUsage: avgP,
|
||||||
|
RechargeDays: rechargeDays,
|
||||||
|
MaxUsage: maxP,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// countRecharges 统计疑似充值天数（当天剩余较前一天回升）
|
||||||
|
func countRecharges(daily []dailyStat) int {
|
||||||
|
n := 0
|
||||||
|
for i := 1; i < len(daily); i++ {
|
||||||
|
if daily[i].Surplus > daily[i-1].Surplus {
|
||||||
|
n++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return n
|
||||||
|
}
|
||||||
130
go/lzuapi.go
Normal file
130
go/lzuapi.go
Normal file
@@ -0,0 +1,130 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// lzuapi.go - FasterLZU (appservice.lzu.edu.cn) 登录客户端
|
||||||
|
// 对应 Python lzuapi.py 中的 login() 与 get_st()
|
||||||
|
|
||||||
|
const (
|
||||||
|
appServiceBase = "https://appservice.lzu.edu.cn"
|
||||||
|
lzuUserAgent = "Mozilla/5.0 (Linux; Android 12; SM-S7110 Build/V417IR; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/101.0.4951.61 Mobile Safari/537.36 lzdx_ua JHZF_LZDXAPP"
|
||||||
|
)
|
||||||
|
|
||||||
|
// lzuClient FasterLZU API 客户端
|
||||||
|
type lzuClient struct {
|
||||||
|
http *http.Client
|
||||||
|
username string
|
||||||
|
password string
|
||||||
|
loginToken string
|
||||||
|
gatewayToken string
|
||||||
|
}
|
||||||
|
|
||||||
|
func newLzuClient(username, password string) *lzuClient {
|
||||||
|
return &lzuClient{
|
||||||
|
http: &http.Client{},
|
||||||
|
username: username,
|
||||||
|
password: password,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// encryptedHeaders 加密请求头
|
||||||
|
func encryptedHeaders() http.Header {
|
||||||
|
h := http.Header{}
|
||||||
|
h.Set("Content-Type", "application/json;charset=UTF-8")
|
||||||
|
h.Set("Transfer-Encrypt", "true")
|
||||||
|
h.Set("User-Agent", lzuUserAgent)
|
||||||
|
return h
|
||||||
|
}
|
||||||
|
|
||||||
|
// Login 登录，返回 (loginToken, gatewayToken)
|
||||||
|
func (c *lzuClient) Login() (string, string, error) {
|
||||||
|
url := appServiceBase + "/api/eusp-unify-terminal/app-user/login"
|
||||||
|
data := map[string]any{
|
||||||
|
"app_os": 2,
|
||||||
|
"name": c.username,
|
||||||
|
"pwd": c.password,
|
||||||
|
}
|
||||||
|
plaintext, _ := json.Marshal(data) // 无空格，与 Python separators=(",",":") 一致
|
||||||
|
ciphertext, err := aesEncrypt(string(plaintext))
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
req, err := http.NewRequestWithContext(ctx, "POST", url, bytes.NewBufferString(ciphertext))
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
req.Header = encryptedHeaders()
|
||||||
|
resp, err := c.http.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
|
||||||
|
decrypted, err := aesDecrypt(strings.TrimSpace(string(body)))
|
||||||
|
if err != nil {
|
||||||
|
return "", "", fmt.Errorf("登录响应解密失败: %w", err)
|
||||||
|
}
|
||||||
|
var res struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data struct {
|
||||||
|
LoginToken string `json:"login_token"`
|
||||||
|
GatewayToken string `json:"gateway_token"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal([]byte(decrypted), &res); err != nil {
|
||||||
|
return "", "", fmt.Errorf("登录响应解析失败: %w", err)
|
||||||
|
}
|
||||||
|
if res.Code != 1 {
|
||||||
|
return "", "", fmt.Errorf("登录失败 code=%d", res.Code)
|
||||||
|
}
|
||||||
|
c.loginToken = res.Data.LoginToken
|
||||||
|
c.gatewayToken = res.Data.GatewayToken
|
||||||
|
return c.loginToken, c.gatewayToken, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetSt 获取服务令牌 ST
|
||||||
|
func (c *lzuClient) GetSt(serviceID string) (string, error) {
|
||||||
|
url := fmt.Sprintf("%s/api/eusp-unify-terminal/app-user/getSt?loginToken=%s&serviceId=%s&service=",
|
||||||
|
appServiceBase, c.loginToken, serviceID)
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
req.Header = encryptedHeaders()
|
||||||
|
resp, err := c.http.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
|
||||||
|
decrypted, err := aesDecrypt(strings.TrimSpace(string(body)))
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("getSt 响应解密失败: %w", err)
|
||||||
|
}
|
||||||
|
var res struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data string `json:"data"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal([]byte(decrypted), &res); err != nil {
|
||||||
|
return "", fmt.Errorf("getSt 响应解析失败: %w", err)
|
||||||
|
}
|
||||||
|
if res.Code != 1 {
|
||||||
|
return "", fmt.Errorf("getSt 失败 code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return res.Data, nil
|
||||||
|
}
|
||||||
223
go/main.go
Normal file
223
go/main.go
Normal file
@@ -0,0 +1,223 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"embed"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// main.go - 宿舍电费监控 Web 服务（零第三方依赖，单二进制）
|
||||||
|
// 路由:
|
||||||
|
// GET / -> 静态前端
|
||||||
|
// GET /api/electric -> 实时电费
|
||||||
|
// GET /api/history -> 历史记录 + 每日耗电统计
|
||||||
|
// 定时任务: 每天 00:00 记录电费，失败重试 5 次（间隔 60s）
|
||||||
|
|
||||||
|
//go:embed static
|
||||||
|
var staticFS embed.FS
|
||||||
|
|
||||||
|
const (
|
||||||
|
retryTimes = 5
|
||||||
|
retryInterval = 60 * time.Second
|
||||||
|
checkInterval = 30 * time.Second
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
host = envOr("HOST", "0.0.0.0")
|
||||||
|
port = envOr("PORT", "8000")
|
||||||
|
)
|
||||||
|
|
||||||
|
// envOr 读取环境变量，为空时返回默认值
|
||||||
|
func envOr(key, def string) string {
|
||||||
|
if v := os.Getenv(key); v != "" {
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
return def
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- 定时任务 ----------
|
||||||
|
func schedulerLoop() {
|
||||||
|
lastTryDate := ""
|
||||||
|
for {
|
||||||
|
now := time.Now()
|
||||||
|
today := now.Format("2006-01-02")
|
||||||
|
// 00:00~00:05 窗口触发
|
||||||
|
if now.Hour() == 0 && now.Minute() < 5 && lastTryDate != today {
|
||||||
|
lastTryDate = today
|
||||||
|
if !hasRecord(today) {
|
||||||
|
fmt.Printf("[定时] %s 00:00 触发每日电费记录\n", today)
|
||||||
|
runDailyRecordWithRetry()
|
||||||
|
time.Sleep(checkInterval * 2)
|
||||||
|
}
|
||||||
|
} else if now.Hour() == 0 && !hasRecord(today) && lastTryDate != today {
|
||||||
|
// 失败补记
|
||||||
|
lastTryDate = today
|
||||||
|
fmt.Println("[补记] 今日无记录，尝试补记")
|
||||||
|
runDailyRecordWithRetry()
|
||||||
|
}
|
||||||
|
time.Sleep(checkInterval)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func runDailyRecordWithRetry() {
|
||||||
|
for i := 1; i <= retryTimes; i++ {
|
||||||
|
if recordAndLog() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Printf("[定时] 第 %d 次尝试失败，%d 秒后重试\n", i, retryInterval/time.Second)
|
||||||
|
time.Sleep(retryInterval)
|
||||||
|
}
|
||||||
|
fmt.Printf("[定时] %d 次重试均失败，等待下轮\n", retryTimes)
|
||||||
|
}
|
||||||
|
|
||||||
|
func recordAndLog() bool {
|
||||||
|
res := queryElectric()
|
||||||
|
if !res.OK {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
today := time.Now().Format("2006-01-02")
|
||||||
|
if err := addRecord(today, res.Surplus, res.Amount, "auto"); err != nil {
|
||||||
|
fmt.Println("[记录] 写入失败:", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
fmt.Printf("[记录] %s 剩余 %.2f 度 (%.2f 元) [auto]\n", today, res.Surplus, res.Amount)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- HTTP 处理 ----------
|
||||||
|
func apiElectric(w http.ResponseWriter, r *http.Request) {
|
||||||
|
res := queryElectric()
|
||||||
|
writeJSON(w, res)
|
||||||
|
}
|
||||||
|
|
||||||
|
func apiHistory(w http.ResponseWriter, r *http.Request) {
|
||||||
|
q := r.URL.Query()
|
||||||
|
month := q.Get("month") // YYYY-MM，空=全部
|
||||||
|
|
||||||
|
records := getRecords()
|
||||||
|
allDaily := buildDailyStats(records, "") // 全量（用于可用月份列表）
|
||||||
|
monthDaily := buildDailyStats(records, month)
|
||||||
|
|
||||||
|
// 可用月份列表（降序）
|
||||||
|
months := availableMonths(records)
|
||||||
|
|
||||||
|
summary := buildMonthSummary(monthDaily, month)
|
||||||
|
writeJSON(w, map[string]any{
|
||||||
|
"ok": true,
|
||||||
|
"month": month,
|
||||||
|
"months": months,
|
||||||
|
"records": monthDaily, // 升序每日统计（含 usage）
|
||||||
|
"daily": monthDaily,
|
||||||
|
"summary": summary,
|
||||||
|
"all_count": len(allDaily),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// availableMonths 返回所有有记录的月份（YYYY-MM，降序）
|
||||||
|
func availableMonths(records []record) []string {
|
||||||
|
seen := map[string]bool{}
|
||||||
|
var months []string
|
||||||
|
for _, r := range records {
|
||||||
|
if len(r.Date) >= 7 {
|
||||||
|
m := r.Date[:7]
|
||||||
|
if !seen[m] {
|
||||||
|
seen[m] = true
|
||||||
|
months = append(months, m)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sort.Sort(sort.Reverse(sort.StringSlice(months)))
|
||||||
|
return months
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeJSON(w http.ResponseWriter, v any) {
|
||||||
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||||||
|
w.Header().Set("Cache-Control", "no-store")
|
||||||
|
_ = json.NewEncoder(w).Encode(v)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- 静态文件 ----------
|
||||||
|
// 直接从 embed FS 读取并写出，避免 http.FileServer 的目录重定向问题
|
||||||
|
func serveStatic(w http.ResponseWriter, r *http.Request) {
|
||||||
|
p := r.URL.Path
|
||||||
|
orig := p
|
||||||
|
if strings.HasPrefix(p, "/static/") {
|
||||||
|
p = strings.TrimPrefix(p, "/static/")
|
||||||
|
}
|
||||||
|
if p == "/" {
|
||||||
|
p = "/index.html"
|
||||||
|
}
|
||||||
|
if p == "" {
|
||||||
|
p = "/index.html"
|
||||||
|
}
|
||||||
|
|
||||||
|
// 规范化为以 / 开头的 embed 路径
|
||||||
|
if !strings.HasPrefix(p, "/") {
|
||||||
|
p = "/" + p
|
||||||
|
}
|
||||||
|
full := "static" + p
|
||||||
|
data, err := staticFS.ReadFile(full)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("[静态] %s -> %s: 读取失败 %v\n", orig, full, err)
|
||||||
|
http.NotFound(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// 设置 Content-Type
|
||||||
|
switch {
|
||||||
|
case strings.HasSuffix(p, ".html"):
|
||||||
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||||||
|
case strings.HasSuffix(p, ".css"):
|
||||||
|
w.Header().Set("Content-Type", "text/css; charset=utf-8")
|
||||||
|
case strings.HasSuffix(p, ".js"):
|
||||||
|
w.Header().Set("Content-Type", "text/javascript; charset=utf-8")
|
||||||
|
case strings.HasSuffix(p, ".png"):
|
||||||
|
w.Header().Set("Content-Type", "image/png")
|
||||||
|
default:
|
||||||
|
w.Header().Set("Content-Type", "application/octet-stream")
|
||||||
|
}
|
||||||
|
w.Header().Set("Cache-Control", "no-cache")
|
||||||
|
_, _ = w.Write(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// 启动定时任务
|
||||||
|
go schedulerLoop()
|
||||||
|
|
||||||
|
// 启动时若今日无记录，立即查询建立基线
|
||||||
|
today := time.Now().Format("2006-01-02")
|
||||||
|
if !hasRecord(today) {
|
||||||
|
fmt.Println("[启动] 今日无记录，立即查询并记录")
|
||||||
|
go runDailyRecordWithRetry()
|
||||||
|
}
|
||||||
|
|
||||||
|
mux := http.NewServeMux()
|
||||||
|
mux.HandleFunc("/api/electric", apiElectric)
|
||||||
|
mux.HandleFunc("/api/history", apiHistory)
|
||||||
|
mux.HandleFunc("/", serveStatic)
|
||||||
|
|
||||||
|
addr := fmt.Sprintf("%s:%s", host, port)
|
||||||
|
fmt.Println("=" + strings.Repeat("=", 49))
|
||||||
|
fmt.Println("宿舍电费监控服务已启动")
|
||||||
|
fmt.Println(" 访问地址: http://" + addr)
|
||||||
|
fmt.Println(" 定时记录: 每天 00:00，失败重试", retryTimes, "次")
|
||||||
|
fmt.Println("=" + strings.Repeat("=", 49))
|
||||||
|
|
||||||
|
// 带超时的 HTTP 服务器：防止慢速/空闲连接堆积 goroutine
|
||||||
|
srv := &http.Server{
|
||||||
|
Addr: addr,
|
||||||
|
Handler: mux,
|
||||||
|
ReadHeaderTimeout: 10 * time.Second,
|
||||||
|
ReadTimeout: 30 * time.Second,
|
||||||
|
WriteTimeout: 60 * time.Second,
|
||||||
|
IdleTimeout: 90 * time.Second,
|
||||||
|
}
|
||||||
|
if err := srv.ListenAndServe(); err != nil {
|
||||||
|
fmt.Println("服务启动失败:", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user