Golang一对一聊天

Question

我想在golang上进行一对一的聊天，我发现这个带有websocket的简单脚本工作得很好，它是一个房间，里面有多少用户想要。但我想将它转换为像facebook一样一个这是脚本如果有人可以帮助，因为我不知道我需要使用更多的连接或筛选用户。

package main 

import (
    "log" 
    "net/http" 

    "github.com/gorilla/websocket" 
) 

var clients = make(map[*websocket.Conn]bool) // connected clients 
var broadcast = make(chan Message)   // broadcast channel 

// Configure the upgrader 
var upgrader = websocket.Upgrader{ 
    CheckOrigin: func(r *http.Request) bool { 
     return true 
    }, 
} 

// Define our message object 
type Message struct { 
    Email string `json:"email"` 
    Username string `json:"username"` 
    Message string `json:"message"` 
    Created string `json:"created"` 
} 

func main() { 
    // Create a simple file server 
    fs := http.FileServer(http.Dir("public")) 
    http.Handle("/", fs) 

    // Configure websocket route 
    http.HandleFunc("/ws", handleConnections) 

    // Start listening for incoming chat messages 
    go handleMessages() 

    // Start the server on localhost port 8000 and log any errors 
    log.Println("http server started on :8090") 
    err := http.ListenAndServe(":8090", nil) 
    if err != nil { 
     log.Fatal("ListenAndServe: ", err) 
    } 
} 

func handleConnections(w http.ResponseWriter, r *http.Request) { 
    // Upgrade initial GET request to a websocket 
    ws, err := upgrader.Upgrade(w, r, nil) 
    if err != nil { 
     log.Fatal(err) 
    } 
    // Make sure we close the connection when the function returns 
    defer ws.Close() 

    // Register our new client 
    clients[ws] = true 

    for { 
     var msg Message 
     // Read in a new message as JSON and map it to a Message object 
     err := ws.ReadJSON(&msg) 
     if err != nil { 
      log.Printf("error: %v", err) 
      delete(clients, ws) 
      break 
     } 
     // Send the newly received message to the broadcast channel 
     broadcast <- msg 
    } 
} 

func handleMessages() { 
    for { 
     // Grab the next message from the broadcast channel 
     msg := <-broadcast 
     // Send it out to every client that is currently connected 
     for client := range clients { 
      err := client.WriteJSON(msg) 
      if err != nil { 
       log.Printf("error: %v", err) 
       client.Close() 
       delete(clients, client) 
      } 
     } 
    } 
} 

是我需要改变这一部分

clients[ws] = true 

Answer 1

你需要做几件事情：

1. 摆脱broadcast通道
2. 某种方式传递&从请求得到它您的用户想要连接的客户端。一些房间号/名称，密码？例如URL参数/ws?chat=abc。您可能需要维护一个map[chatid][]*websocket.Conn
3. 匹配2（或更多）客户端。
4. 维护地图，可能类型为map[*websocket.Conn]*websocket.Conn
5. 在从客户端接收到消息查找地图并将消息发送到匹配的客户端。与handleMessages()类似，但只是一次。

请注意StackOverflow不是要求为您编写代码的地方。

Answer 2

到这里看看：https://github.com/tinode/chat

它已经做了你想要的东西。