重新规划网关与节点的交互协议,实现统一命令位的识别和处理
This commit is contained in:
73
server/core/conn.go
Normal file
73
server/core/conn.go
Normal file
@@ -0,0 +1,73 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"net"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Conn struct {
|
||||
Conn net.Conn
|
||||
Reader *bufio.Reader
|
||||
Tag [16]byte
|
||||
Protocol string
|
||||
Dest *FwdAddr
|
||||
Auth *AuthContext
|
||||
}
|
||||
|
||||
func (c Conn) Read(b []byte) (n int, err error) {
|
||||
return c.Reader.Read(b)
|
||||
}
|
||||
|
||||
func (c Conn) Write(b []byte) (n int, err error) {
|
||||
return c.Conn.Write(b)
|
||||
}
|
||||
|
||||
func (c Conn) Close() error {
|
||||
return c.Conn.Close()
|
||||
}
|
||||
|
||||
func (c Conn) LocalAddr() net.Addr {
|
||||
return c.Conn.LocalAddr()
|
||||
}
|
||||
|
||||
func (c Conn) RemoteAddr() net.Addr {
|
||||
return c.Conn.RemoteAddr()
|
||||
}
|
||||
|
||||
func (c Conn) SetDeadline(t time.Time) error {
|
||||
return c.Conn.SetDeadline(t)
|
||||
}
|
||||
|
||||
func (c Conn) SetReadDeadline(t time.Time) error {
|
||||
return c.Conn.SetReadDeadline(t)
|
||||
}
|
||||
|
||||
func (c Conn) SetWriteDeadline(t time.Time) error {
|
||||
return c.Conn.SetWriteDeadline(t)
|
||||
}
|
||||
|
||||
type FwdAddr struct {
|
||||
IP net.IP
|
||||
Port int
|
||||
Domain string
|
||||
}
|
||||
|
||||
func (a FwdAddr) Network() string {
|
||||
return "tcp"
|
||||
}
|
||||
|
||||
func (a FwdAddr) String() string {
|
||||
return fmt.Sprintf("%s:%d", a.IP, a.Port)
|
||||
}
|
||||
|
||||
type AuthContext struct {
|
||||
Timeout float64
|
||||
Payload Payload
|
||||
Meta map[string]any
|
||||
}
|
||||
|
||||
type Payload struct {
|
||||
ID int32
|
||||
}
|
||||
65
server/core/map.go
Normal file
65
server/core/map.go
Normal file
@@ -0,0 +1,65 @@
|
||||
package core
|
||||
|
||||
import "sync"
|
||||
|
||||
type SyncMap[K any, V any] struct {
|
||||
_map sync.Map
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) Store(key K, value V) {
|
||||
m._map.Store(key, value)
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) Load(key K) (value V, ok bool) {
|
||||
v, ok := m._map.Load(key)
|
||||
if ok {
|
||||
value = v.(V)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) Swap(key K, value V) (previous V, loaded bool) {
|
||||
v, loaded := m._map.Swap(key, value)
|
||||
if loaded {
|
||||
previous = v.(V)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) Delete(key K) {
|
||||
m._map.Delete(key)
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) Clear() {
|
||||
m._map.Clear()
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) Range(f func(key K, value V) bool) {
|
||||
m._map.Range(func(k, v any) bool {
|
||||
return f(k.(K), v.(V))
|
||||
})
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) LoadOrStore(key K, value V) (actual V, loaded bool) {
|
||||
v, loaded := m._map.LoadOrStore(key, value)
|
||||
if loaded {
|
||||
actual = v.(V)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) LoadAndDelete(key K) (value V, ok bool) {
|
||||
v, ok := m._map.LoadAndDelete(key)
|
||||
if ok {
|
||||
value = v.(V)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) CompareAndSwap(key K, old, new V) (swapped bool) {
|
||||
return m._map.CompareAndSwap(key, old, new)
|
||||
}
|
||||
|
||||
func (m *SyncMap[K, V]) CompareAndDelete(key K, old V) (deleted bool) {
|
||||
return m._map.CompareAndDelete(key, old)
|
||||
}
|
||||
Reference in New Issue
Block a user