first release

internal/web/mask_cache.go

@@ -0,0 +1,84 @@
+package web
+
+import (
+	"sync"
+	"time"
+
+	"rond-with-you/internal/fog"
+)
+
+// maskTTL 是遮罩列表的缓存时长。保存/删除遮罩时会主动失效，这里的 TTL 只是兜底
+// （比如多实例部署时另一个进程改了库）。所以可以取长一点。
+const maskTTL = 30 * time.Second
+
+// maskCache 缓存迷雾遮罩列表。
+//
+// 为什么需要：遮罩不在 fogIndex 的内存索引里，每个迷雾瓦片请求都会查一次库，
+// 而地图一屏就是几十张瓦片——同一份数据被反复查几十遍。指纹也顺带缓存，
+// 它被瓦片 URL 与缓存键用到。
+type maskCache struct {
+	mu    sync.Mutex
+	at    time.Time
+	list  []fog.Mask
+	fp    string
+	valid bool
+}
+
+// get 返回缓存的遮罩列表与指纹；ok 为 false 表示需要查库。
+// 「站点一块遮罩都没有」也是有效缓存（list 为 nil），所以不能拿 list 判空。
+func (c *maskCache) get() (list []fog.Mask, fp string, ok bool) {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	if c.valid && time.Since(c.at) < maskTTL {
+		return c.list, c.fp, true
+	}
+	return nil, "", false
+}
+
+// put 写入缓存（由调用方查库后调用）。
+func (c *maskCache) put(list []fog.Mask) ([]fog.Mask, string) {
+	fp := fog.MaskFingerprint(list)
+	c.mu.Lock()
+	c.list, c.fp, c.at, c.valid = list, fp, time.Now(), true
+	c.mu.Unlock()
+	return list, fp
+}
+
+// invalidate 在遮罩被保存 / 删除后立刻失效，保证改完立即对所有访客生效。
+func (c *maskCache) invalidate() {
+	c.mu.Lock()
+	c.valid = false
+	c.mu.Unlock()
+}
+
+// masksFor 取遮罩列表（带缓存）。查库失败时返回 nil（等于不遮），与原先的行为一致。
+func (s *Server) masksFor() []fog.Mask {
+	if list, _, ok := s.masks.get(); ok {
+		return list
+	}
+	list, err := fog.LoadMasks(s.db)
+	if err != nil {
+		return nil
+	}
+	out, _ := s.masks.put(list)
+	return out
+}
+
+// maskSetFor 取遮罩判定集合（带缓存）；没有遮罩时返回 nil。
+//
+// admin 为 true 时不返回遮罩——站长永远看完整迷雾，否则没法核对遮哪里。
+func (s *Server) maskSetFor(admin bool) *fog.MaskSet {
+	if admin {
+		return nil
+	}
+	return fog.NewMaskSet(s.masksFor())
+}
+
+// maskFP 取遮罩指纹（带缓存），供瓦片 URL 与缓存键使用。
+func (s *Server) maskFP() string {
+	if _, fp, ok := s.masks.get(); ok {
+		return fp
+	}
+	list := s.masksFor()
+	return fog.MaskFingerprint(list)
+}