From f1c2121b2ca9b4dbd90833081781af1b6fb06c14 Mon Sep 17 00:00:00 2001
From: qydysky <qydysky@foxmail.com>
Date: Sun, 7 Jan 2024 15:16:09 +0800
Subject: [PATCH] 1

---
 web/Web.go | 112 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)

diff --git a/web/Web.go b/web/Web.go
index 59cb80a..d18a438 100644
--- a/web/Web.go
+++ b/web/Web.go
@@ -3,6 +3,7 @@ package part
 import (
 	"context"
 	"crypto/tls"
+	"errors"
 	"net"
 	"net/http"
 	"strconv"
@@ -12,6 +13,7 @@ import (
 	"time"
 
 	"github.com/dustin/go-humanize"
+	"github.com/google/uuid"
 	psync "github.com/qydysky/part/sync"
 	sys "github.com/qydysky/part/sys"
 )
@@ -106,6 +108,11 @@ type WebPath struct {
 	sync.RWMutex
 }
 
+// WebSync
+func (t *WebPath) GetConn(r *http.Request) net.Conn {
+	return r.Context().Value(t).(net.Conn)
+}
+
 func (t *WebPath) Load(path string) (func(w http.ResponseWriter, r *http.Request), bool) {
 	t.RLock()
 	defer t.RUnlock()
@@ -419,6 +426,111 @@ func (t withflush) WriteHeader(i int) {
 	}
 }
 
+type Exprier struct {
+	max int
+	m   psync.Map
+}
+
+var (
+	ErrExprie   = errors.New("ErrExprie")
+	ErrNoFound  = errors.New("ErrNoFound")
+	ErrOverflow = errors.New("ErrOverflow")
+)
+
+func NewExprier(max int) *Exprier {
+	return &Exprier{max: max}
+}
+
+func (t *Exprier) Reg(key string, dur time.Duration) (string, error) {
+	if t.max <= 0 {
+		return "noExprie", nil
+	}
+	if key != "" {
+		if _, ok := t.m.Load(key); ok {
+			t.m.Store(key, time.Now().Add(dur))
+			return key, nil
+		} else {
+			return key, ErrNoFound
+		}
+	} else if t.m.Len() >= t.max {
+		return "", ErrOverflow
+	} else {
+		key = strings.ToUpper(uuid.New().String())
+		t.m.Store(key, time.Now().Add(dur))
+		return key, nil
+	}
+}
+
+func (t *Exprier) Check(key string) error {
+	if t.max <= 0 {
+		return nil
+	}
+	if key == "" {
+		return ErrNoFound
+	}
+	ey, ok := t.m.LoadV(key).(time.Time)
+	if !ok {
+		return ErrNoFound
+	} else if time.Now().After(ey) {
+		t.m.Delete(key)
+		return ErrExprie
+	}
+	return nil
+}
+
+func (t *Exprier) LoopCheck(key string, checkDru time.Duration, conn net.Conn) (done func()) {
+	if t.max <= 0 {
+		return func() {}
+	}
+	if key == "" {
+		conn.Close()
+		return func() {}
+	}
+	ey, ok := t.m.LoadV(key).(time.Time)
+	if !ok {
+		conn.Close()
+		return func() {}
+	} else if time.Now().After(ey) {
+		t.m.Delete(key)
+		conn.Close()
+		return func() {}
+	}
+
+	c := make(chan struct{})
+	go func() {
+		for t.max > 0 {
+			ey, ok := t.m.LoadV(key).(time.Time)
+			if !ok {
+				conn.Close()
+			} else if time.Now().After(ey) {
+				t.m.Delete(key)
+				conn.Close()
+			}
+			select {
+			case <-c:
+				return
+			case <-time.After(checkDru):
+			}
+		}
+	}()
+	return func() {
+		close(c)
+	}
+}
+
+func (t *Exprier) Disable() {
+	t.max = 0
+	t.m.ClearAll()
+}
+
+func (t *Exprier) Drop(key string) {
+	t.m.Delete(key)
+}
+
+func (t *Exprier) Len() int {
+	return t.m.Len()
+}
+
 func WithFlush(w http.ResponseWriter) http.ResponseWriter {
 	return withflush{w}
 }
-- 
2.39.2