🐶 重构部分用户数据库操作代码

pkg/global/global.go

@@ -1,26 +0,0 @@
-package global
-
-import (
-	"next-terminal/pkg/config"
-
-	"github.com/patrickmn/go-cache"
-	"github.com/robfig/cron/v3"
-	"gorm.io/gorm"
-)
-
-var DB *gorm.DB
-
-var Cache *cache.Cache
-
-var Config *config.Config
-
-var Store *TunStore
-
-var Cron *cron.Cron
-
-type Security struct {
-	Rule string
-	IP   string
-}
-
-var Securities []*Security

pkg/global/store.go

@@ -1,71 +0,0 @@
-package global
-
-import (
-	"strconv"
-	"sync"
-
-	"next-terminal/pkg/guacd"
-	"next-terminal/pkg/term"
-
-	"github.com/gorilla/websocket"
-)
-
-type Tun struct {
-	Protocol     string
-	Mode         string
-	WebSocket    *websocket.Conn
-	Tunnel       *guacd.Tunnel
-	NextTerminal *term.NextTerminal
-}
-
-func (r *Tun) Close(code int, reason string) {
-	if r.Tunnel != nil {
-		_ = r.Tunnel.Close()
-	}
-	if r.NextTerminal != nil {
-		_ = r.NextTerminal.Close()
-	}
-
-	ws := r.WebSocket
-	if ws != nil {
-		if r.Mode == "guacd" {
-			err := guacd.NewInstruction("error", "", strconv.Itoa(code))
-			_ = ws.WriteMessage(websocket.TextMessage, []byte(err.String()))
-			disconnect := guacd.NewInstruction("disconnect")
-			_ = ws.WriteMessage(websocket.TextMessage, []byte(disconnect.String()))
-		} else {
-			msg := `{"type":"closed","content":"` + reason + `"}`
-			_ = ws.WriteMessage(websocket.TextMessage, []byte(msg))
-		}
-	}
-}
-
-type Observable struct {
-	Subject   *Tun
-	Observers []Tun
-}
-
-type TunStore struct {
-	m sync.Map
-}
-
-func (s *TunStore) Set(k string, v *Observable) {
-	s.m.Store(k, v)
-}
-
-func (s *TunStore) Del(k string) {
-	s.m.Delete(k)
-}
-
-func (s *TunStore) Get(k string) (item *Observable, ok bool) {
-	value, ok := s.m.Load(k)
-	if ok {
-		return value.(*Observable), true
-	}
-	return item, false
-}
-
-func NewStore() *TunStore {
-	store := TunStore{sync.Map{}}
-	return &store
-}