mirror of https://gitee.com/godoos/godoos.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
3.1 KiB
123 lines
3.1 KiB
// MIT License
|
|
//
|
|
// Copyright (c) 2024 godoos.com
|
|
// Email: xpbb@qq.com
|
|
// GitHub: github.com/phpk/godoos
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in all
|
|
// copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
// SOFTWARE.
|
|
package sys
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"godo/localchat"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// 定义消息结构体
|
|
type Message struct {
|
|
Type string `json:"type"`
|
|
Data interface{} `json:"data"`
|
|
}
|
|
|
|
// 定义一个结构体来保存客户端连接
|
|
type client struct {
|
|
conn http.ResponseWriter
|
|
}
|
|
|
|
var clients = make(map[*client]bool)
|
|
var mutex sync.Mutex
|
|
|
|
// 处理客户端连接
|
|
func HandleSystemEvents(w http.ResponseWriter, r *http.Request) {
|
|
// 设置响应头
|
|
w.Header().Set("Content-Type", "text/event-stream")
|
|
w.Header().Set("Cache-Control", "no-cache")
|
|
w.Header().Set("Connection", "keep-alive")
|
|
|
|
// 创建客户端连接对象
|
|
c := &client{conn: w}
|
|
|
|
// 添加客户端到列表
|
|
mutex.Lock()
|
|
clients[c] = true
|
|
mutex.Unlock()
|
|
// 发送初始消息
|
|
updateInfo, err := GetUpdateInfo()
|
|
if err == nil {
|
|
SendToClient(c, Message{
|
|
Type: "update",
|
|
Data: updateInfo,
|
|
})
|
|
}
|
|
|
|
// 监听客户端关闭
|
|
defer func() {
|
|
mutex.Lock()
|
|
delete(clients, c)
|
|
mutex.Unlock()
|
|
}()
|
|
|
|
// 使用定时器轮询客户端请求
|
|
ticker := time.NewTicker(1 * time.Second)
|
|
defer ticker.Stop()
|
|
for range ticker.C {
|
|
// 检查客户端是否已断开连接
|
|
if r.Context().Err() != nil {
|
|
return
|
|
}
|
|
userMessages := localchat.GetMessages()
|
|
msg := Message{
|
|
Type: "localchat",
|
|
Data: userMessages,
|
|
}
|
|
Broadcast(msg)
|
|
localchat.ClearAllUserMessages()
|
|
}
|
|
}
|
|
|
|
// 向客户端发送消息
|
|
func SendToClient(c *client, msg Message) {
|
|
jsonMsg, err := json.Marshal(msg)
|
|
if err != nil {
|
|
fmt.Println("Error marshaling JSON:", err)
|
|
return
|
|
}
|
|
fmt.Fprintf(c.conn, "data: %s\n\n", jsonMsg)
|
|
c.conn.(http.Flusher).Flush()
|
|
}
|
|
|
|
// 广播消息给所有客户端
|
|
func Broadcast(msg Message) {
|
|
mutex.Lock()
|
|
defer mutex.Unlock()
|
|
for c := range clients {
|
|
SendToClient(c, msg)
|
|
}
|
|
}
|
|
|
|
// 每隔一段时间广播一条消息
|
|
func SendMessagePeriodically(msg Message) {
|
|
ticker := time.NewTicker(1 * time.Second)
|
|
for range ticker.C {
|
|
Broadcast(msg)
|
|
}
|
|
}
|
|
|