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.
112 lines
2.8 KiB
112 lines
2.8 KiB
package localchat
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
go StartServiceDiscovery()
|
|
go DiscoverServers()
|
|
}
|
|
|
|
func SseHandler(w http.ResponseWriter, r *http.Request) {
|
|
flusher, ok := w.(http.Flusher)
|
|
if !ok {
|
|
http.Error(w, "Streaming unsupported!", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "text/event-stream")
|
|
w.Header().Set("Cache-Control", "no-cache")
|
|
w.Header().Set("Connection", "keep-alive")
|
|
w.WriteHeader(http.StatusOK)
|
|
ticker := time.NewTicker(broadcartTime) // 每3秒检查一次在线用户
|
|
defer ticker.Stop()
|
|
// 处理新消息
|
|
ctx := r.Context()
|
|
// 使用Context来监听请求的取消
|
|
go func() {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
log.Printf("Recovered in SSE goroutine: %v", r)
|
|
}
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done(): // 当请求被取消时,退出循环
|
|
return
|
|
case msg := <-messageChan:
|
|
// 构造JSON数据
|
|
jsonData, err := json.Marshal(msg)
|
|
if err != nil {
|
|
log.Printf("Failed to marshal message to JSON: %v", err)
|
|
continue
|
|
}
|
|
|
|
// 通过SSE发送JSON数据
|
|
fmt.Fprintf(w, "data: %s\n\n", string(jsonData))
|
|
flusher.Flush()
|
|
}
|
|
}
|
|
}()
|
|
myIP, myHostname, err := getMyIPAndHostname()
|
|
if err != nil {
|
|
log.Printf("Failed to get my IP and hostname: %v", err)
|
|
return
|
|
}
|
|
for {
|
|
select {
|
|
case <-ticker.C: // 每隔一段时间检查并广播在线用户
|
|
var userList []UdpMessage
|
|
// 首先将自己的IP和主机名放入列表
|
|
myMsg := UdpMessage{
|
|
IP: myIP,
|
|
Hostname: myHostname,
|
|
Type: "online",
|
|
Message: time.Now().Format("2006-01-02 15:04:05"),
|
|
}
|
|
userList = append(userList, myMsg)
|
|
log.Printf("Online users: %v", OnlineUsers)
|
|
for ip, info := range OnlineUsers {
|
|
if ip != myIP { // 确保不重复添加自己
|
|
userList = append(userList, info)
|
|
}
|
|
}
|
|
res := UserList{Type: "user_list", Content: userList}
|
|
// 将用户列表转换为JSON字符串
|
|
jsonData, err := json.Marshal(res)
|
|
if err != nil {
|
|
log.Printf("Failed to marshal online users to JSON: %v", err)
|
|
continue
|
|
}
|
|
// 通过SSE发送JSON数据
|
|
fmt.Fprintf(w, "data: %s\n\n", string(jsonData))
|
|
flusher.Flush()
|
|
}
|
|
}
|
|
}
|
|
func HandleMessage(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
var msg Message
|
|
decoder := json.NewDecoder(r.Body)
|
|
if err := decoder.Decode(&msg); err != nil {
|
|
http.Error(w, "Invalid request body", http.StatusBadRequest)
|
|
return
|
|
}
|
|
defer r.Body.Close()
|
|
// 将消息放入messageChan
|
|
messageChan <- msg
|
|
log.Printf("Received text message from %s: %s", msg.SenderInfo.IP, msg.Content)
|
|
// 这里可以添加存储文本消息到数据库或其他处理逻辑
|
|
w.WriteHeader(http.StatusOK)
|
|
fmt.Fprintln(w, "Text message send successfully")
|
|
}
|
|
|