实时聊天应用 #

type Client struct {
    ID     string
    Name   string
    Room   string
    Conn   *websocket.Conn
}

type Room struct {
    ID      string
    Clients map[string]*Client
}

type Hub struct {
    Rooms    map[string]*Room
    Register chan *Client
    Message  chan *Message
}

type Message struct {
    Room    string `json:"room"`
    Sender  string `json:"sender"`
    Content string `json:"content"`
    Time    string `json:"time"`
}

app.Get("/ws/:room/:user", websocket.New(func(c *websocket.Conn) {
    room := c.Params("room")
    user := c.Params("user")
    
    client := &Client{
        ID:   uuid.New().String(),
        Name: user,
        Room: room,
        Conn: c,
    }
    
    hub.Register <- client
    
    for {
        _, msg, err := c.ReadMessage()
        if err != nil {
            break
        }
        
        message := &Message{
            Room:    room,
            Sender:  user,
            Content: string(msg),
            Time:    time.Now().Format(time.RFC3339),
        }
        
        hub.Message <- message
    }
}))