feat: 从黄果 NET hgdjnet-server 9bfbafe 导入 HKDJ 官网后端

来源：hgdjnet-server 9bfbafe10d417deea959bd53117bc88a10e60a83 项目身份 HKDJ / JHA-216，独立官网仓库，Jenkins 守卫键 hkdjnet 测试目录 /server/hkdj_net，App/Web API 30331/30332 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-10-16 11:20:04 +08:00
parent 99a4854def
commit 85c0071fd8
653 changed files with 85000 additions and 0 deletions
@@ -0,0 +1,78 @@
+package ipstat
+
+import (
+	"hkdj-web-server/pub/common/enum/redisconst"
+	"hkdj-web-server/pub/common/ipaddr"
+	"hkdj-web-server/pub/common/log"
+	"hkdj-web-server/pub/common/utils"
+	"hkdj-web-server/pub/pubdef"
+	"hkdj-web-server/pub/redis"
+	"net"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+func IpStatMiddleware(ctx *gin.Context) {
+	ctx.Next()
+
+	normalized := normalizeIP(ipaddr.ResolveClientIP(ctx))
+	if normalized == "" {
+		return
+	}
+
+	date := time.Now().In(utils.LocBJ).Format("20060102")
+	key := redisconst.IpStatKey(pubdef.AppId, date)
+
+	added, err := redis.SAddCount(key, normalized)
+	if err != nil {
+		log.Warn("IpStatMiddleware SAdd failed", log.Any("key", key), log.E(err))
+		return
+	}
+	if added > 0 {
+		if err := redis.Expire(key, redisconst.IpStatExpire); err != nil {
+			log.Warn("IpStatMiddleware Expire failed", log.Any("key", key), log.E(err))
+		}
+	}
+}
+
+func normalizeIP(raw string) string {
+	ip := net.ParseIP(raw)
+	if ip == nil {
+		return ""
+	}
+	if v4 := ip.To4(); v4 != nil {
+		ip = v4
+	}
+	if isPrivateIP(ip) {
+		return ""
+	}
+	return ip.String()
+}
+
+var privateRanges = []net.IPNet{
+	parseCIDR("10.0.0.0/8"),
+	parseCIDR("172.16.0.0/12"),
+	parseCIDR("192.168.0.0/16"),
+	parseCIDR("127.0.0.0/8"),
+	parseCIDR("::1/128"),
+	parseCIDR("fe80::/10"),
+	parseCIDR("fc00::/7"),
+}
+
+func parseCIDR(s string) net.IPNet {
+	_, network, err := net.ParseCIDR(s)
+	if err != nil {
+		panic(err)
+	}
+	return *network
+}
+
+func isPrivateIP(ip net.IP) bool {
+	for _, r := range privateRanges {
+		if r.Contains(ip) {
+			return true
+		}
+	}
+	return false
+}