feat(pool): 添加SQLite连接池实现并集成到TCP和WebSocket服务
- 新增pool包，包含ConnType连接类型和ConnectionInfo连接信息结构体 - 实现SQLitePool连接池，支持添加、获取、删除、更新连接操作 - 为TCP服务器集成SQLite连接池，存储连接信息到数据库 - 为WebSocket管理器集成SQLite连接池，存储连接信息到数据库 - 在TCP和WebSocket的连接生命周期中同步更新SQLite连接状态 - 添加GetAllConnIDs方法获取所有在线连接ID列表 - 在示例代码中添加错误处理和测试功能main
parent
d4f78124d8
commit
2520655fbd
|
|
@ -17,6 +17,20 @@
|
|||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="BadgerDB">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/db/badger.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="BadgerPool">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/pool/badger.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="BaseConfig">
|
||||
<value>
|
||||
<set>
|
||||
|
|
@ -34,6 +48,8 @@
|
|||
<entry key="Config">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/db/badger.go" />
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/server/ws/websocket.go" />
|
||||
<option value="file://$PROJECT_DIR$/ws/websocket.go" />
|
||||
</set>
|
||||
</value>
|
||||
|
|
@ -41,17 +57,49 @@
|
|||
<entry key="Connection">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/server/ws/websocket.go" />
|
||||
<option value="file://$PROJECT_DIR$/ws/websocket.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="ConnectionInfo">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/pool/badger.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="ConnectionMeta">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/db/connection_store.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="ConnectionPool">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcp.go" />
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcp_badger.go" />
|
||||
<option value="file://$PROJECT_DIR$/tcp/tcp.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="ConnectionPoolBadger">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcpConfig.go" />
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcp_badger.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="ConnectionStore">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/db/connection_store.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="Curd">
|
||||
<value>
|
||||
<set>
|
||||
|
|
@ -83,28 +131,30 @@
|
|||
<entry key="Manager">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/server/ws/websocket.go" />
|
||||
<option value="file://$PROJECT_DIR$/ws/websocket.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="ManagerBadger">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/server/ws/websocket_badger.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="Msg">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/server/ws/websocket.go" />
|
||||
<option value="file://$PROJECT_DIR$/ws/websocket.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="NewsOne">
|
||||
<entry key="OfflineMessage">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gin_test/api/new.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="NewsSave">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gin_test/api/new.go" />
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/db/connection_store.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
|
|
@ -125,6 +175,8 @@
|
|||
<entry key="TCPServer">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcp.go" />
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcp_badger.go" />
|
||||
<option value="file://$PROJECT_DIR$/tcp/tcp.go" />
|
||||
</set>
|
||||
</value>
|
||||
|
|
@ -132,6 +184,7 @@
|
|||
<entry key="TcpConnection">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcpConfig.go" />
|
||||
<option value="file://$PROJECT_DIR$/tcp/tcpConfig.go" />
|
||||
</set>
|
||||
</value>
|
||||
|
|
@ -139,6 +192,7 @@
|
|||
<entry key="TcpMessage">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcpConfig.go" />
|
||||
<option value="file://$PROJECT_DIR$/tcp/tcpConfig.go" />
|
||||
</set>
|
||||
</value>
|
||||
|
|
@ -146,24 +200,11 @@
|
|||
<entry key="TcpPoolConfig">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../gf-common/tcp/tcpConfig.go" />
|
||||
<option value="file://$PROJECT_DIR$/tcp/tcpConfig.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="api">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/response/code.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="apiFile">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/response/code.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="resFile">
|
||||
<value>
|
||||
<set>
|
||||
|
|
@ -178,10 +219,140 @@
|
|||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="sTcp">
|
||||
<value>
|
||||
<set>
|
||||
<option value="file://$PROJECT_DIR$/../yingji/socket-server/internal/service/tcp.go" />
|
||||
</set>
|
||||
</value>
|
||||
</entry>
|
||||
</map>
|
||||
</option>
|
||||
<option name="scannedPathMapping">
|
||||
<map>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/db/badger.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772154321394" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="BadgerDB" />
|
||||
<option value="Config" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/db/connection_store.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772154031118" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="ConnectionMeta" />
|
||||
<option value="OfflineMessage" />
|
||||
<option value="ConnectionStore" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/example_badger.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772154243481" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/pool/badger.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772156220301" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="ConnectionInfo" />
|
||||
<option value="BadgerPool" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/server/ws/example.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772156267433" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/server/ws/websocket.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772156237755" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="Config" />
|
||||
<option value="Connection" />
|
||||
<option value="Manager" />
|
||||
<option value="Msg" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/server/ws/websocket_badger.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772154194371" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="ManagerBadger" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/tcp/example.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772156261275" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/tcp/tcp.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772156252696" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="TCPServer" />
|
||||
<option value="ConnectionPool" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/tcp/tcpConfig.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772154391708" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="TcpPoolConfig" />
|
||||
<option value="TcpConnection" />
|
||||
<option value="TcpMessage" />
|
||||
<option value="ConnectionPoolBadger" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gf-common/tcp/tcp_badger.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772154435736" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/config/fun.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
|
|
@ -384,7 +555,7 @@
|
|||
<entry key="file://$PROJECT_DIR$/ws/websocket.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1770018907536" />
|
||||
<option name="lastModified" value="1772185691346" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="Config" />
|
||||
|
|
@ -396,83 +567,47 @@
|
|||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gin_test/api/new.go">
|
||||
<entry key="file://$PROJECT_DIR$/../yingji/socket-server/internal/service/tcp.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772007643850" />
|
||||
<option name="lastModified" value="1772157204272" />
|
||||
<option name="schema">
|
||||
<list>
|
||||
<option value="NewsOne" />
|
||||
<option value="NewsSave" />
|
||||
<option value="sTcp" />
|
||||
</list>
|
||||
</option>
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gin_test/controller/home/index.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772007727399" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gin_test/main.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772006716414" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gin_test/middleware/middleware.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772006700129" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gin_test/router/homeRouter.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772003574712" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gin_test/router/router.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772003574719" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
<entry key="file://$PROJECT_DIR$/../gin_test/utils/customer.go">
|
||||
<value>
|
||||
<ScannedPath>
|
||||
<option name="lastModified" value="1772007185481" />
|
||||
</ScannedPath>
|
||||
</value>
|
||||
</entry>
|
||||
</map>
|
||||
</option>
|
||||
<option name="tableStructMapping">
|
||||
<map>
|
||||
<entry key="api" value="Api" />
|
||||
<entry key="api_file" value="ApiFile" />
|
||||
<entry key="badger_db" value="BadgerDB" />
|
||||
<entry key="badger_pool" value="BadgerPool" />
|
||||
<entry key="base_config" value="BaseConfig" />
|
||||
<entry key="client" value="Client" />
|
||||
<entry key="config" value="Config" />
|
||||
<entry key="connection" value="Connection" />
|
||||
<entry key="connection_info" value="ConnectionInfo" />
|
||||
<entry key="connection_meta" value="ConnectionMeta" />
|
||||
<entry key="connection_pool" value="ConnectionPool" />
|
||||
<entry key="connection_pool_badger" value="ConnectionPoolBadger" />
|
||||
<entry key="connection_store" value="ConnectionStore" />
|
||||
<entry key="curd" value="Curd" />
|
||||
<entry key="data_base_config" value="DataBaseConfig" />
|
||||
<entry key="jwt_claims" value="JWTClaims" />
|
||||
<entry key="jwt_config" value="JwtConfig" />
|
||||
<entry key="manager" value="Manager" />
|
||||
<entry key="manager_badger" value="ManagerBadger" />
|
||||
<entry key="msg" value="Msg" />
|
||||
<entry key="news_one" value="NewsOne" />
|
||||
<entry key="news_save" value="NewsSave" />
|
||||
<entry key="offline_message" value="OfflineMessage" />
|
||||
<entry key="paginate" value="Paginate" />
|
||||
<entry key="res_file" value="resFile" />
|
||||
<entry key="response" value="response" />
|
||||
<entry key="s_tcp" value="sTcp" />
|
||||
<entry key="server_config" value="ServerConfig" />
|
||||
<entry key="tcp_connection" value="TcpConnection" />
|
||||
<entry key="tcp_message" value="TcpMessage" />
|
||||
|
|
@ -480,6 +615,6 @@
|
|||
<entry key="tcp_server" value="TCPServer" />
|
||||
</map>
|
||||
</option>
|
||||
<option name="lastTimeChecked" value="1772003052939" />
|
||||
<option name="lastTimeChecked" value="1772152595532" />
|
||||
</component>
|
||||
</project>
|
||||
|
|
@ -0,0 +1,25 @@
|
|||
package pool
|
||||
|
||||
import (
|
||||
"time"
|
||||
)
|
||||
|
||||
// ConnType 连接类型
|
||||
type ConnType string
|
||||
|
||||
const (
|
||||
ConnTypeWebSocket ConnType = "websocket"
|
||||
ConnTypeTCP ConnType = "tcp"
|
||||
)
|
||||
|
||||
// ConnectionInfo 连接信息
|
||||
type ConnectionInfo struct {
|
||||
ID string `json:"id" gorm:"primaryKey"`
|
||||
Type ConnType `json:"type" gorm:"index"`
|
||||
Address string `json:"address"`
|
||||
IsActive bool `json:"isActive" gorm:"index"`
|
||||
LastUsed time.Time `json:"lastUsed"`
|
||||
CreatedAt time.Time `json:"createdAt"`
|
||||
// 额外的连接数据，根据不同类型存储不同的信息
|
||||
Data map[string]interface{} `json:"data" gorm:"-"`
|
||||
}
|
||||
|
|
@ -0,0 +1,247 @@
|
|||
package pool
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"git.magicany.cc/black1552/gf-common/db"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// SQLitePool SQLite连接池
|
||||
type SQLitePool struct {
|
||||
db *gorm.DB
|
||||
mutex sync.RWMutex
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
// 内存缓存，提高并发性能
|
||||
cache map[string]*ConnectionInfo
|
||||
}
|
||||
|
||||
// NewSQLitePool 创建SQLite连接池
|
||||
func NewSQLitePool() (*SQLitePool, error) {
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
|
||||
// 检查数据库连接是否正常
|
||||
if db.Db == nil {
|
||||
return nil, fmt.Errorf("database connection is not initialized")
|
||||
}
|
||||
|
||||
// 自动迁移ConnectionInfo模型
|
||||
err := db.Db.AutoMigrate(&ConnectionInfo{})
|
||||
if err != nil {
|
||||
cancel()
|
||||
return nil, fmt.Errorf("failed to migrate connection info model: %w", err)
|
||||
}
|
||||
|
||||
return &SQLitePool{
|
||||
db: db.Db,
|
||||
ctx: ctx,
|
||||
cancel: cancel,
|
||||
cache: make(map[string]*ConnectionInfo),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// Close 关闭连接池
|
||||
func (p *SQLitePool) Close() error {
|
||||
p.cancel()
|
||||
// SQLite连接由db包管理，不需要在这里关闭
|
||||
return nil
|
||||
}
|
||||
|
||||
// Add 添加连接
|
||||
func (p *SQLitePool) Add(conn *ConnectionInfo) error {
|
||||
p.mutex.Lock()
|
||||
defer p.mutex.Unlock()
|
||||
|
||||
// 存储到SQLite
|
||||
result := p.db.Create(conn)
|
||||
if result.Error != nil {
|
||||
return fmt.Errorf("failed to store connection: %w", result.Error)
|
||||
}
|
||||
|
||||
// 更新内存缓存
|
||||
p.cache[conn.ID] = conn
|
||||
return nil
|
||||
}
|
||||
|
||||
// Get 获取连接
|
||||
func (p *SQLitePool) Get(connID string) (*ConnectionInfo, error) {
|
||||
p.mutex.RLock()
|
||||
// 先从内存缓存获取
|
||||
if conn, ok := p.cache[connID]; ok {
|
||||
p.mutex.RUnlock()
|
||||
return conn, nil
|
||||
}
|
||||
p.mutex.RUnlock()
|
||||
|
||||
// 从SQLite获取
|
||||
var connInfo ConnectionInfo
|
||||
result := p.db.First(&connInfo, "id = ?", connID)
|
||||
if result.Error != nil {
|
||||
if result.Error == gorm.ErrRecordNotFound {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, fmt.Errorf("failed to get connection: %w", result.Error)
|
||||
}
|
||||
|
||||
// 更新内存缓存
|
||||
p.mutex.Lock()
|
||||
p.cache[connID] = &connInfo
|
||||
p.mutex.Unlock()
|
||||
|
||||
return &connInfo, nil
|
||||
}
|
||||
|
||||
// Remove 移除连接
|
||||
func (p *SQLitePool) Remove(connID string) error {
|
||||
p.mutex.Lock()
|
||||
defer p.mutex.Unlock()
|
||||
|
||||
// 从SQLite删除
|
||||
result := p.db.Delete(&ConnectionInfo{}, "id = ?", connID)
|
||||
if result.Error != nil {
|
||||
return fmt.Errorf("failed to remove connection: %w", result.Error)
|
||||
}
|
||||
|
||||
// 从内存缓存删除
|
||||
delete(p.cache, connID)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Update 更新连接信息
|
||||
func (p *SQLitePool) Update(conn *ConnectionInfo) error {
|
||||
p.mutex.Lock()
|
||||
defer p.mutex.Unlock()
|
||||
|
||||
// 存储到SQLite
|
||||
result := p.db.Save(conn)
|
||||
if result.Error != nil {
|
||||
return fmt.Errorf("failed to update connection: %w", result.Error)
|
||||
}
|
||||
|
||||
// 更新内存缓存
|
||||
p.cache[conn.ID] = conn
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetAll 获取所有连接
|
||||
func (p *SQLitePool) GetAll() ([]*ConnectionInfo, error) {
|
||||
p.mutex.RLock()
|
||||
// 如果内存缓存不为空，直接返回缓存
|
||||
if len(p.cache) > 0 {
|
||||
conns := make([]*ConnectionInfo, 0, len(p.cache))
|
||||
for _, conn := range p.cache {
|
||||
conns = append(conns, conn)
|
||||
}
|
||||
p.mutex.RUnlock()
|
||||
return conns, nil
|
||||
}
|
||||
p.mutex.RUnlock()
|
||||
|
||||
// 从SQLite获取所有连接
|
||||
var conns []*ConnectionInfo
|
||||
result := p.db.Find(&conns)
|
||||
if result.Error != nil {
|
||||
return nil, fmt.Errorf("failed to get all connections: %w", result.Error)
|
||||
}
|
||||
|
||||
// 更新内存缓存
|
||||
p.mutex.Lock()
|
||||
for _, conn := range conns {
|
||||
p.cache[conn.ID] = conn
|
||||
}
|
||||
p.mutex.Unlock()
|
||||
|
||||
return conns, nil
|
||||
}
|
||||
|
||||
// GetByType 根据类型获取连接
|
||||
func (p *SQLitePool) GetByType(connType ConnType) ([]*ConnectionInfo, error) {
|
||||
allConns, err := p.GetAll()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var filtered []*ConnectionInfo
|
||||
for _, conn := range allConns {
|
||||
if conn.Type == connType {
|
||||
filtered = append(filtered, conn)
|
||||
}
|
||||
}
|
||||
|
||||
return filtered, nil
|
||||
}
|
||||
|
||||
// Count 获取连接数量
|
||||
func (p *SQLitePool) Count() (int, error) {
|
||||
p.mutex.RLock()
|
||||
// 如果内存缓存不为空，直接返回缓存大小
|
||||
if len(p.cache) > 0 {
|
||||
count := len(p.cache)
|
||||
p.mutex.RUnlock()
|
||||
return count, nil
|
||||
}
|
||||
p.mutex.RUnlock()
|
||||
|
||||
// 从SQLite统计数量
|
||||
var count int64
|
||||
result := p.db.Model(&ConnectionInfo{}).Count(&count)
|
||||
if result.Error != nil {
|
||||
return 0, fmt.Errorf("failed to count connections: %w", result.Error)
|
||||
}
|
||||
|
||||
return int(count), nil
|
||||
}
|
||||
|
||||
// GetAllConnIDs 获取所有在线连接的ID列表
|
||||
func (p *SQLitePool) GetAllConnIDs() ([]string, error) {
|
||||
p.mutex.RLock()
|
||||
// 如果内存缓存不为空，从缓存中提取在线连接的ID
|
||||
if len(p.cache) > 0 {
|
||||
ids := make([]string, 0, len(p.cache))
|
||||
for id, conn := range p.cache {
|
||||
if conn.IsActive {
|
||||
ids = append(ids, id)
|
||||
}
|
||||
}
|
||||
p.mutex.RUnlock()
|
||||
return ids, nil
|
||||
}
|
||||
p.mutex.RUnlock()
|
||||
|
||||
// 从SQLite获取所有在线连接的ID
|
||||
var conns []*ConnectionInfo
|
||||
result := p.db.Where("is_active = ?", true).Find(&conns)
|
||||
if result.Error != nil {
|
||||
return nil, fmt.Errorf("failed to get all connection IDs: %w", result.Error)
|
||||
}
|
||||
|
||||
ids := make([]string, 0, len(conns))
|
||||
for _, conn := range conns {
|
||||
ids = append(ids, conn.ID)
|
||||
}
|
||||
|
||||
return ids, nil
|
||||
}
|
||||
|
||||
// CleanupInactive 清理不活跃的连接
|
||||
func (p *SQLitePool) CleanupInactive(duration time.Duration) error {
|
||||
allConns, err := p.GetAll()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
for _, conn := range allConns {
|
||||
if !conn.IsActive || now.Sub(conn.LastUsed) > duration {
|
||||
if err := p.Remove(conn.ID); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
@ -18,7 +18,11 @@ func Example() {
|
|||
}
|
||||
|
||||
// 创建TCP服务器
|
||||
server := NewTCPServer("0.0.0.0:8888", config)
|
||||
server, err := NewTCPServer("0.0.0.0:8888", config)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed to create server: %v\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
// 设置消息处理函数
|
||||
server.SetMessageHandler(func(conn *TcpConnection, msg *TcpMessage) error {
|
||||
|
|
@ -45,3 +49,48 @@ func Example() {
|
|||
|
||||
fmt.Println("TCP server stopped.")
|
||||
}
|
||||
|
||||
// TestTCP 测试TCP连接
|
||||
func TestTCP() {
|
||||
fmt.Println("=== 测试TCP连接 ===")
|
||||
fmt.Println("1. 创建TCP服务器配置")
|
||||
config := &TcpPoolConfig{
|
||||
BufferSize: 2048,
|
||||
MaxConnections: 100000,
|
||||
ConnectTimeout: time.Second * 5,
|
||||
ReadTimeout: time.Second * 30,
|
||||
WriteTimeout: time.Second * 10,
|
||||
MaxIdleTime: time.Minute * 5,
|
||||
}
|
||||
fmt.Println("2. 创建TCP服务器")
|
||||
server, err := NewTCPServer("0.0.0.0:8888", config)
|
||||
if err != nil {
|
||||
fmt.Printf("创建服务器失败：%v\n", err)
|
||||
return
|
||||
}
|
||||
fmt.Println("3. 服务器创建成功")
|
||||
fmt.Println("4. 获取在线连接数")
|
||||
count := server.Connection.Count()
|
||||
fmt.Printf("当前在线连接数：%d\n", count)
|
||||
fmt.Println("5. 获取所有在线连接ID")
|
||||
connIDs, err := server.GetAllConnIDs()
|
||||
if err != nil {
|
||||
fmt.Printf("获取在线连接ID失败：%v\n", err)
|
||||
} else {
|
||||
fmt.Printf("在线连接ID：%v\n", connIDs)
|
||||
}
|
||||
fmt.Println("6. 启动服务器")
|
||||
if err := server.Start(); err != nil {
|
||||
fmt.Printf("启动服务器失败：%v\n", err)
|
||||
return
|
||||
}
|
||||
fmt.Println("7. 服务器启动成功，运行2秒后停止")
|
||||
time.Sleep(time.Second * 2)
|
||||
fmt.Println("8. 停止服务器")
|
||||
if err := server.Stop(); err != nil {
|
||||
fmt.Printf("停止服务器失败：%v\n", err)
|
||||
} else {
|
||||
fmt.Println("服务器停止成功")
|
||||
}
|
||||
fmt.Println("=== TCP测试完成 ===")
|
||||
}
|
||||
|
|
|
|||
82
tcp/tcp.go
82
tcp/tcp.go
|
|
@ -6,6 +6,7 @@ import (
|
|||
"sync"
|
||||
"time"
|
||||
|
||||
"git.magicany.cc/black1552/gf-common/pool"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/gtcp"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
|
|
@ -32,18 +33,26 @@ type TCPServer struct {
|
|||
// ConnectionPool 连接池结构
|
||||
type ConnectionPool struct {
|
||||
connections map[string]*TcpConnection
|
||||
sqlitePool *pool.SQLitePool
|
||||
mutex sync.RWMutex
|
||||
config *TcpPoolConfig
|
||||
logger *glog.Logger
|
||||
}
|
||||
|
||||
// NewTCPServer 创建一个新的TCP服务器
|
||||
func NewTCPServer(address string, config *TcpPoolConfig) *TCPServer {
|
||||
func NewTCPServer(address string, config *TcpPoolConfig) (*TCPServer, error) {
|
||||
logger := g.Log(address)
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
|
||||
// 初始化SQLite连接池
|
||||
sqlitePool, err := pool.NewSQLitePool()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to create sqlite pool: %w", err)
|
||||
}
|
||||
|
||||
pool := &ConnectionPool{
|
||||
connections: make(map[string]*TcpConnection),
|
||||
sqlitePool: sqlitePool,
|
||||
config: config,
|
||||
logger: logger,
|
||||
}
|
||||
|
|
@ -58,7 +67,7 @@ func NewTCPServer(address string, config *TcpPoolConfig) *TCPServer {
|
|||
}
|
||||
|
||||
server.Listener = gtcp.NewServer(address, server.handleConnection)
|
||||
return server
|
||||
return server, nil
|
||||
}
|
||||
|
||||
// SetMessageHandler 设置消息处理函数
|
||||
|
|
@ -86,6 +95,11 @@ func (s *TCPServer) Stop() error {
|
|||
s.Listener.Close()
|
||||
s.wg.Wait()
|
||||
s.Connection.Clear()
|
||||
// 关闭SQLite连接池
|
||||
if err := s.Connection.sqlitePool.Close(); err != nil {
|
||||
s.Logger.Error(s.ctx, fmt.Sprintf("Failed to close SQLite pool: %v", err))
|
||||
// 不影响服务器停止，仅记录错误
|
||||
}
|
||||
s.Logger.Info(s.ctx, "TCP server stopped")
|
||||
return nil
|
||||
}
|
||||
|
|
@ -109,6 +123,23 @@ func (s *TCPServer) handleConnection(conn *gtcp.Conn) {
|
|||
s.Connection.Add(tcpConn)
|
||||
s.Logger.Info(s.ctx, fmt.Sprintf("New connection established: %s", connID))
|
||||
|
||||
// 存储到SQLite
|
||||
connInfo := &pool.ConnectionInfo{
|
||||
ID: connID,
|
||||
Type: pool.ConnTypeTCP,
|
||||
Address: conn.RemoteAddr().String(),
|
||||
IsActive: true,
|
||||
LastUsed: time.Now(),
|
||||
CreatedAt: time.Now(),
|
||||
Data: map[string]interface{}{
|
||||
"localAddress": conn.LocalAddr().String(),
|
||||
},
|
||||
}
|
||||
if err := s.Connection.sqlitePool.Add(connInfo); err != nil {
|
||||
s.Logger.Error(s.ctx, fmt.Sprintf("Failed to store connection to SQLite: %v", err))
|
||||
// 不影响连接建立，仅记录错误
|
||||
}
|
||||
|
||||
// 启动消息接收协程
|
||||
go s.receiveMessages(tcpConn)
|
||||
}
|
||||
|
|
@ -121,6 +152,11 @@ func (s *TCPServer) receiveMessages(conn *TcpConnection) {
|
|||
}
|
||||
s.Connection.Remove(conn.Id)
|
||||
conn.Server.Close()
|
||||
// 从SQLite移除
|
||||
if err := s.Connection.sqlitePool.Remove(conn.Id); err != nil {
|
||||
s.Logger.Error(s.ctx, fmt.Sprintf("Failed to remove connection from SQLite: %v", err))
|
||||
// 不影响连接关闭，仅记录错误
|
||||
}
|
||||
s.Logger.Info(s.ctx, fmt.Sprintf("Connection closed: %s", conn.Id))
|
||||
}()
|
||||
|
||||
|
|
@ -142,10 +178,21 @@ func (s *TCPServer) receiveMessages(conn *TcpConnection) {
|
|||
|
||||
if n > 0 {
|
||||
// 更新最后使用时间
|
||||
now := time.Now()
|
||||
conn.Mutex.Lock()
|
||||
conn.LastUsed = time.Now()
|
||||
conn.LastUsed = now
|
||||
conn.Mutex.Unlock()
|
||||
|
||||
// 更新SQLite中的连接信息
|
||||
connInfo, err := s.Connection.sqlitePool.Get(conn.Id)
|
||||
if err == nil && connInfo != nil {
|
||||
connInfo.LastUsed = now
|
||||
if err := s.Connection.sqlitePool.Update(connInfo); err != nil {
|
||||
s.Logger.Error(s.ctx, fmt.Sprintf("Failed to update connection in SQLite: %v", err))
|
||||
// 不影响消息处理，仅记录错误
|
||||
}
|
||||
}
|
||||
|
||||
// 处理消息
|
||||
data := make([]byte, n)
|
||||
copy(data, buffer[:n])
|
||||
|
|
@ -209,7 +256,19 @@ func (s *TCPServer) sendMessage(conn *TcpConnection, data []byte) error {
|
|||
}
|
||||
|
||||
// 更新最后使用时间
|
||||
conn.LastUsed = time.Now()
|
||||
now := time.Now()
|
||||
conn.LastUsed = now
|
||||
|
||||
// 更新SQLite中的连接信息
|
||||
connInfo, err := s.Connection.sqlitePool.Get(conn.Id)
|
||||
if err == nil && connInfo != nil {
|
||||
connInfo.LastUsed = now
|
||||
if err := s.Connection.sqlitePool.Update(connInfo); err != nil {
|
||||
s.Logger.Error(s.ctx, fmt.Sprintf("Failed to update connection in SQLite: %v", err))
|
||||
// 不影响消息发送，仅记录错误
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
|
|
@ -224,11 +283,21 @@ func (s *TCPServer) Kick(connID string) error {
|
|||
conn.Server.Close()
|
||||
// 从连接池移除
|
||||
s.Connection.Remove(connID)
|
||||
// 从SQLite移除
|
||||
if err := s.Connection.sqlitePool.Remove(connID); err != nil {
|
||||
s.Logger.Error(s.ctx, fmt.Sprintf("Failed to remove connection from SQLite: %v", err))
|
||||
// 不影响连接关闭，仅记录错误
|
||||
}
|
||||
|
||||
s.Logger.Info(s.ctx, fmt.Sprintf("Kicked connection: %s", connID))
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetAllConnIDs 获取所有在线连接的ID列表
|
||||
func (s *TCPServer) GetAllConnIDs() ([]string, error) {
|
||||
return s.Connection.GetAllConnIDs()
|
||||
}
|
||||
|
||||
// Add 添加连接到连接池
|
||||
func (p *ConnectionPool) Add(conn *TcpConnection) {
|
||||
p.mutex.Lock()
|
||||
|
|
@ -278,3 +347,8 @@ func (p *ConnectionPool) Count() int {
|
|||
defer p.mutex.RUnlock()
|
||||
return len(p.connections)
|
||||
}
|
||||
|
||||
// GetAllConnIDs 获取所有在线连接的ID列表
|
||||
func (p *ConnectionPool) GetAllConnIDs() ([]string, error) {
|
||||
return p.sqlitePool.GetAllConnIDs()
|
||||
}
|
||||
|
|
|
|||
|
|
@ -19,7 +19,10 @@ func NewWs() *Manager {
|
|||
}
|
||||
|
||||
// 2. 创建管理器
|
||||
m := NewManager(customConfig)
|
||||
m, err := NewManager(customConfig)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to create manager: %v", err)
|
||||
}
|
||||
|
||||
// 3. 覆盖业务回调（核心：自定义消息处理逻辑）
|
||||
// 连接建立回调
|
||||
|
|
@ -71,3 +74,35 @@ func main() {
|
|||
log.Println("WebSocket服务启动：http://localhost:8080/ws")
|
||||
log.Fatal(http.ListenAndServe(":8080", nil))
|
||||
}
|
||||
|
||||
// TestWebSocket 测试WebSocket连接
|
||||
func TestWebSocket() {
|
||||
log.Println("=== 测试WebSocket连接 ===")
|
||||
log.Println("1. 创建WebSocket管理器")
|
||||
m, err := NewManager(DefaultConfig())
|
||||
if err != nil {
|
||||
log.Fatalf("创建管理器失败：%v", err)
|
||||
}
|
||||
log.Println("2. 管理器创建成功")
|
||||
log.Println("3. 获取在线连接数")
|
||||
count, err := m.sqlitePool.Count()
|
||||
if err != nil {
|
||||
log.Printf("获取在线连接数失败：%v", err)
|
||||
} else {
|
||||
log.Printf("当前在线连接数：%d", count)
|
||||
}
|
||||
log.Println("4. 获取所有在线连接ID")
|
||||
connIDs, err := m.GetAllConnIDs()
|
||||
if err != nil {
|
||||
log.Printf("获取在线连接ID失败：%v", err)
|
||||
} else {
|
||||
log.Printf("在线连接ID：%v", connIDs)
|
||||
}
|
||||
log.Println("5. 关闭管理器")
|
||||
if err := m.Close(); err != nil {
|
||||
log.Printf("关闭管理器失败：%v", err)
|
||||
} else {
|
||||
log.Println("管理器关闭成功")
|
||||
}
|
||||
log.Println("=== WebSocket测试完成 ===")
|
||||
}
|
||||
|
|
|
|||
|
|
@ -9,6 +9,7 @@ import (
|
|||
"sync"
|
||||
"time"
|
||||
|
||||
"git.magicany.cc/black1552/gin-base/pool"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/os/gtime"
|
||||
|
|
@ -20,20 +21,20 @@ import (
|
|||
|
||||
// 常量定义：默认配置
|
||||
const (
|
||||
// DefaultReadBufferSize 默认读写缓冲区大小（字节）
|
||||
// 默认读写缓冲区大小（字节）
|
||||
DefaultReadBufferSize = 1024
|
||||
DefaultWriteBufferSize = 1024
|
||||
// DefaultHeartbeatInterval 默认心跳间隔（秒）：每30秒发送一次心跳
|
||||
// 默认心跳间隔（秒）：每30秒发送一次心跳
|
||||
DefaultHeartbeatInterval = 30 * time.Second
|
||||
// DefaultHeartbeatTimeout 默认心跳超时（秒）：60秒未收到客户端心跳响应则关闭连接
|
||||
// 默认心跳超时（秒）：60秒未收到客户端心跳响应则关闭连接
|
||||
DefaultHeartbeatTimeout = 60 * time.Second
|
||||
// DefaultReadTimeout 默认读写超时（秒）
|
||||
// 默认读写超时（秒）
|
||||
DefaultReadTimeout = 60 * time.Second
|
||||
DefaultWriteTimeout = 10 * time.Second
|
||||
// MessageTypeText 消息类型
|
||||
// 消息类型
|
||||
MessageTypeText = websocket.TextMessage
|
||||
MessageTypeBinary = websocket.BinaryMessage
|
||||
// HeartbeatMaxRetry 心跳最大重试次数
|
||||
// 心跳最大重试次数
|
||||
HeartbeatMaxRetry = 3
|
||||
)
|
||||
|
||||
|
|
@ -92,7 +93,8 @@ type Connection struct {
|
|||
type Manager struct {
|
||||
config *Config // 配置
|
||||
upgrader *websocket.Upgrader // HTTP升级器
|
||||
connections map[string]*Connection // 所有在线连接（connID -> Connection）
|
||||
connections map[string]*Connection // 内存中的连接（connID -> Connection）
|
||||
sqlitePool *pool.SQLitePool // SQLite连接池
|
||||
mutex sync.RWMutex // 读写锁（保护connections）
|
||||
// 业务回调：收到消息时触发（用户自定义处理逻辑）
|
||||
OnMessage func(connID string, msgType int, data any)
|
||||
|
|
@ -148,16 +150,16 @@ func (c *Config) Merge(other *Config) *Config {
|
|||
}
|
||||
|
||||
// NewManager 创建连接管理器
|
||||
func NewManager(config *Config) *Manager {
|
||||
func NewManager(config *Config) (*Manager, error) {
|
||||
defaultConfig := DefaultConfig()
|
||||
finalConfig := defaultConfig.Merge(config)
|
||||
// 初始化升级器
|
||||
upgrader := &websocket.Upgrader{
|
||||
ReadBufferSize: config.ReadBufferSize,
|
||||
WriteBufferSize: config.WriteBufferSize,
|
||||
ReadBufferSize: finalConfig.ReadBufferSize,
|
||||
WriteBufferSize: finalConfig.WriteBufferSize,
|
||||
CheckOrigin: func(r *http.Request) bool {
|
||||
// 跨域检查
|
||||
if config.AllowAllOrigins {
|
||||
if finalConfig.AllowAllOrigins {
|
||||
return true
|
||||
}
|
||||
origin := r.Header.Get("Origin")
|
||||
|
|
@ -170,10 +172,17 @@ func NewManager(config *Config) *Manager {
|
|||
},
|
||||
}
|
||||
|
||||
// 初始化SQLite连接池
|
||||
sqlitePool, err := pool.NewSQLitePool()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to create sqlite pool: %w", err)
|
||||
}
|
||||
|
||||
return &Manager{
|
||||
config: finalConfig,
|
||||
upgrader: upgrader,
|
||||
connections: make(map[string]*Connection),
|
||||
sqlitePool: sqlitePool,
|
||||
mutex: sync.RWMutex{},
|
||||
// 默认回调（用户可覆盖）
|
||||
OnMessage: func(connID string, msgType int, data any) {
|
||||
|
|
@ -185,7 +194,7 @@ func NewManager(config *Config) *Manager {
|
|||
OnDisconnect: func(connID string, err error) {
|
||||
log.Printf("[默认回调] 连接[%s]已关闭：%v", connID, err)
|
||||
},
|
||||
}
|
||||
}, nil
|
||||
}
|
||||
|
||||
// Upgrade HTTP升级为WebSocket连接
|
||||
|
|
@ -237,6 +246,24 @@ func (m *Manager) Upgrade(w http.ResponseWriter, r *http.Request, connID string)
|
|||
m.connections[connID] = wsConn
|
||||
m.mutex.Unlock()
|
||||
|
||||
// 存储到SQLite
|
||||
connInfo := &pool.ConnectionInfo{
|
||||
ID: connID,
|
||||
Type: pool.ConnTypeWebSocket,
|
||||
Address: r.RemoteAddr,
|
||||
IsActive: true,
|
||||
LastUsed: time.Now(),
|
||||
CreatedAt: time.Now(),
|
||||
Data: map[string]interface{}{
|
||||
"origin": r.Header.Get("Origin"),
|
||||
"userAgent": r.Header.Get("User-Agent"),
|
||||
},
|
||||
}
|
||||
if err := m.sqlitePool.Add(connInfo); err != nil {
|
||||
log.Printf("[错误] 存储连接到SQLite失败：%v", err)
|
||||
// 不影响连接建立，仅记录错误
|
||||
}
|
||||
|
||||
// 触发连接建立回调
|
||||
m.OnConnect(connID)
|
||||
|
||||
|
|
@ -282,6 +309,18 @@ func (c *Connection) ReadPump() {
|
|||
return
|
||||
}
|
||||
|
||||
// 更新最后使用时间
|
||||
now := time.Now()
|
||||
// 从SQLite获取连接信息并更新
|
||||
connInfo, err := c.manager.sqlitePool.Get(c.connID)
|
||||
if err == nil && connInfo != nil {
|
||||
connInfo.LastUsed = now
|
||||
if err := c.manager.sqlitePool.Update(connInfo); err != nil {
|
||||
log.Printf("[错误] 更新SQLite连接信息失败：%v", err)
|
||||
// 不影响消息处理，仅记录错误
|
||||
}
|
||||
}
|
||||
|
||||
// 尝试解析JSON格式的心跳消息（精准判断，替代包含判断）
|
||||
isHeartbeat := false
|
||||
// 先尝试解析为JSON对象
|
||||
|
|
@ -369,6 +408,19 @@ func (c *Connection) Send(data []byte) error {
|
|||
if err != nil {
|
||||
return fmt.Errorf("发送消息失败：%w", err)
|
||||
}
|
||||
|
||||
// 更新最后使用时间
|
||||
now := time.Now()
|
||||
// 从SQLite获取连接信息并更新
|
||||
connInfo, err := c.manager.sqlitePool.Get(c.connID)
|
||||
if err == nil && connInfo != nil {
|
||||
connInfo.LastUsed = now
|
||||
if err := c.manager.sqlitePool.Update(connInfo); err != nil {
|
||||
log.Printf("[错误] 更新SQLite连接信息失败：%v", err)
|
||||
// 不影响消息发送，仅记录错误
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
|
@ -394,6 +446,12 @@ func (c *Connection) Close(err error) {
|
|||
delete(c.manager.connections, c.connID)
|
||||
c.manager.mutex.Unlock()
|
||||
|
||||
// 从SQLite移除
|
||||
if err := c.manager.sqlitePool.Remove(c.connID); err != nil {
|
||||
log.Printf("[错误] 从SQLite移除连接失败：%v", err)
|
||||
// 不影响连接关闭，仅记录错误
|
||||
}
|
||||
|
||||
// 触发断开回调
|
||||
c.manager.OnDisconnect(c.connID, err)
|
||||
|
||||
|
|
@ -462,12 +520,18 @@ func (m *Manager) GetAllConn() map[string]*Connection {
|
|||
return connCopy
|
||||
}
|
||||
|
||||
// GetConn 获取指定连接
|
||||
func (m *Manager) GetConn(connID string) *Connection {
|
||||
m.mutex.RLock()
|
||||
defer m.mutex.RUnlock()
|
||||
return m.connections[connID]
|
||||
}
|
||||
|
||||
// GetAllConnIDs 获取所有在线连接的ID列表
|
||||
func (m *Manager) GetAllConnIDs() ([]string, error) {
|
||||
return m.sqlitePool.GetAllConnIDs()
|
||||
}
|
||||
|
||||
// CloseAll 关闭所有连接
|
||||
func (m *Manager) CloseAll() {
|
||||
m.mutex.RLock()
|
||||
|
|
@ -486,3 +550,14 @@ func (m *Manager) CloseAll() {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Close 关闭管理器，清理资源
|
||||
func (m *Manager) Close() error {
|
||||
// 关闭所有连接
|
||||
m.CloseAll()
|
||||
// 关闭SQLite连接池
|
||||
if m.sqlitePool != nil {
|
||||
return m.sqlitePool.Close()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue