feat(watchdog): add watchdog service with TCP client implementation

- Implement TCP client for watchdog service communication
- Add watchdog configuration in config.json
- Update main.go to initialize and connect to watchdog
- Add watchdog related functions (connect, command handling)
- Update README.md with new watchdog feature
- Improve route setup logging in router.go
This commit is contained in:
2026-04-02 18:42:44 +08:00
parent c19bbfbca4
commit aa22b04a1f
7 changed files with 272 additions and 2 deletions
+33
View File
@@ -0,0 +1,33 @@
package watchdog
import (
"fmt"
)
func AddInstance(serviceName string) bool {
if !IsConnected() {
return false
}
message := fmt.Sprintf("[addInstance] <serviceName>%s</serviceName>", serviceName)
response, err := sendMsg(message, 3)
if err != nil {
return false
}
return response == "success"
}
func RemoveInstance(serviceName string) bool {
if !IsConnected() {
return false
}
message := fmt.Sprintf("[removeInstance] <serviceName>%s</serviceName>", serviceName)
response, err := sendMsg(message, 3)
if err != nil {
return false
}
return response == "success"
}
+47
View File
@@ -0,0 +1,47 @@
package watchdog
import (
"time"
)
func Connect(ipaddr string, port int) bool {
if IsConnected() {
return true
}
if err := Init(); err != nil {
return false
}
if err := tcpConnect(ipaddr, port); err != nil {
return false
}
response, err := sendMsg("watchdogAgentConnectionTest", 3)
if err != nil {
Destroy()
return false
}
if response != "success" {
Destroy()
return false
}
return true
}
func Disconnect() bool {
if !IsConnected() {
return true
}
err := Destroy()
if err != nil {
return false
}
time.Sleep(100 * time.Millisecond)
return true
}
+164
View File
@@ -0,0 +1,164 @@
package watchdog
import (
"bufio"
"fmt"
"net"
"super-frpc/postLog"
"sync"
"time"
)
var (
tcpConn net.Conn
tcpConnMutex sync.Mutex
isConnected bool
recvChan chan string
stopRecvChan chan struct{}
)
func notifyMessage(message string) {
postLog.Info(fmt.Sprintf("%s", message))
}
func Init() error {
tcpConnMutex.Lock()
defer tcpConnMutex.Unlock()
if tcpConn != nil {
return fmt.Errorf("TCP client already initialized")
}
recvChan = make(chan string, 100)
stopRecvChan = make(chan struct{})
isConnected = false
return nil
}
func tcpConnect(ipaddr string, port int) error {
tcpConnMutex.Lock()
defer tcpConnMutex.Unlock()
if tcpConn != nil {
return fmt.Errorf("already connected")
}
address := fmt.Sprintf("%s:%d", ipaddr, port)
conn, err := net.DialTimeout("tcp", address, 3*time.Second)
if err != nil {
return fmt.Errorf("failed to connect to %s: %v", address, err)
}
tcpConn = conn
isConnected = true
go recvMsg()
return nil
}
func sendMsg(message string, target int) (string, error) {
tcpConnMutex.Lock()
if tcpConn == nil {
tcpConnMutex.Unlock()
return "", fmt.Errorf("not connected")
}
_, err := tcpConn.Write([]byte(message + "\n"))
if err != nil {
tcpConnMutex.Unlock()
return "", fmt.Errorf("failed to send message: %v", err)
}
tcpConnMutex.Unlock()
select {
case response := <-recvChan:
return response, nil
case <-time.After(time.Duration(target) * time.Second):
return "", fmt.Errorf("timeout waiting for response")
}
}
func recvMsg() {
reader := bufio.NewReader(tcpConn)
for {
select {
case <-stopRecvChan:
return
default:
tcpConn.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
line, err := reader.ReadString('\n')
if err != nil {
if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
continue
}
tcpConnMutex.Lock()
if tcpConn != nil {
tcpConn.Close()
tcpConn = nil
isConnected = false
}
tcpConnMutex.Unlock()
return
}
line = line[:len(line)-1]
if len(line) > 0 {
select {
case recvChan <- line:
default:
select {
case <-recvChan:
recvChan <- line
default:
}
}
if len(line) > 0 && line != "success" && !isResponseMessage(line) {
notifyMessage(line)
}
}
}
}
}
func isResponseMessage(msg string) bool {
return msg == "success" || msg == "failed"
}
func Destroy() error {
tcpConnMutex.Lock()
defer tcpConnMutex.Unlock()
if stopRecvChan != nil {
close(stopRecvChan)
stopRecvChan = nil
}
if tcpConn != nil {
err := tcpConn.Close()
tcpConn = nil
isConnected = false
if err != nil {
return fmt.Errorf("failed to close connection: %v", err)
}
}
if recvChan != nil {
close(recvChan)
recvChan = nil
}
return nil
}
func IsConnected() bool {
tcpConnMutex.Lock()
defer tcpConnMutex.Unlock()
return isConnected && tcpConn != nil
}