Go实现网络通信

Go 语言提供了强大的网络编程能力，包括 TCP、UDP、HTTP、WebSocket 等协议的支持。下面是 Go 语言中常用的网络操作：

TCP 通信

使用 net 包进行 TCP 通信，可以创建 TCP 客户端和服务器。
客户端使用 net.Dial 方法连接到指定的 TCP 地址，并使用返回的连接对象进行通信。
服务器使用 net.Listen 方法监听指定的 TCP 地址，接受客户端连接，并使用返回的连接对象进行通信。

TCP 客户端

package mainimport ("fmt""net"
)func main() {// 连接到 TCP 服务器conn, err := net.Dial("tcp", "localhost:8080")if err != nil {fmt.Println("无法连接到服务器:", err)return}defer conn.Close()// 发送数据message := "Hello, TCP server!"_, err = conn.Write([]byte(message))if err != nil {fmt.Println("发送数据失败:", err)return}// 接收服务器响应buffer := make([]byte, 1024)n, err := conn.Read(buffer)if err != nil {fmt.Println("接收响应失败:", err)return}fmt.Println("服务器响应:", string(buffer[:n]))
}

TCP 服务器

package mainimport ("fmt""net"
)func main() {// 监听 TCP 连接listener, err := net.Listen("tcp", "localhost:8080")if err != nil {fmt.Println("无法监听端口:", err)return}defer listener.Close()fmt.Println("等待客户端连接...")for {// 接受客户端连接conn, err := listener.Accept()if err != nil {fmt.Println("接受连接失败:", err)continue}// 处理客户端请求go handleClient(conn)}
}func handleClient(conn net.Conn) {defer conn.Close()// 接收客户端数据buffer := make([]byte, 1024)n, err := conn.Read(buffer)if err != nil {fmt.Println("读取数据失败:", err)return}fmt.Println("接收到客户端数据:", string(buffer[:n]))// 发送响应给客户端response := "Hello, TCP client!"_, err = conn.Write([]byte(response))if err != nil {fmt.Println("发送响应失败:", err)return}
}

UDP 通信

使用 net 包进行 UDP 通信，可以创建 UDP 客户端和服务器。
客户端使用 net.DialUDP 方法连接到指定的 UDP 地址，并使用返回的连接对象进行通信。
服务器使用 net.ListenUDP 方法监听指定的 UDP 地址，接收客户端发送的数据，并使用返回的连接对象进行通信。

UDP 客户端

package mainimport ("fmt""net"
)func main() {// 连接到 UDP 服务器conn, err := net.Dial("udp", "localhost:8080")if err != nil {fmt.Println("无法连接到服务器:", err)return}defer conn.Close()// 发送数据message := "Hello, UDP server!"_, err = conn.Write([]byte(message))if err != nil {fmt.Println("发送数据失败:", err)return}
}

UDP 服务器

package mainimport ("fmt""net"
)func main() {// 监听 UDP 连接addr, err := net.ResolveUDPAddr("udp", "localhost:8080")if err != nil {fmt.Println("无法解析地址:", err)return}conn, err := net.ListenUDP("udp", addr)if err != nil {fmt.Println("无法监听端口:", err)return}defer conn.Close()fmt.Println("等待客户端连接...")// 接收客户端数据buffer := make([]byte, 1024)n, addr, err := conn.ReadFromUDP(buffer)if err != nil {fmt.Println("读取数据失败:", err)return}fmt.Println("接收到客户端数据:", string(buffer[:n]))// 发送响应给客户端response := "Hello, UDP client!"_, err = conn.WriteToUDP([]byte(response), addr)if err != nil {fmt.Println("发送响应失败:", err)return}
}

HTTP通信

HTTP 客户端

使用 net/http 包进行 HTTP 客户端编程，可以发送 HTTP 请求并接收响应。
使用 http.Get、http.Post 等方法发送不同类型的 HTTP 请求。
可以通过设置请求头、请求方法、请求体等来定制请求，并使用响应对象获取服务器的响应数据。

1. 发起 HTTP GET 请求：

package mainimport ("fmt""io/ioutil""net/http"
)func main() {// 发送 GET 请求response, err := http.Get("http://localhost:8080")if err != nil {fmt.Println("发送请求失败:", err)return}defer response.Body.Close()// 读取响应内容body, err := ioutil.ReadAll(response.Body)if err != nil {fmt.Println("读取响应失败:", err)
return}fmt.Println("服务器响应:", string(body))
}

2. 发起 HTTP POST 请求：

package mainimport ("bytes""fmt""net/http""io/ioutil"
)func main() {url := "https://jsonplaceholder.typicode.com/posts" // 要发送 POST 请求的网址payload := []byte(`{"title": "foo", "body": "bar", "userId": 1}`)resp, err := http.Post(url, "application/json", bytes.NewBuffer(payload))if err != nil {fmt.Println("HTTP POST 请求失败:", err)return}defer resp.Body.Close()body, err := ioutil.ReadAll(resp.Body)if err != nil {fmt.Println("读取响应体失败:", err)return}fmt.Println("POST 响应：", string(body))
}

3. 发起 HTTP PUT 请求：

package mainimport ("bytes""fmt""net/http""io/ioutil"
)func main() {url := "https://jsonplaceholder.typicode.com/posts/1" // 要更新的资源网址payload := []byte(`{"id": 1, "title": "foo", "body": "bar", "userId": 1}`)req, err := http.NewRequest("PUT", url, bytes.NewBuffer(payload))req.Header.Set("Content-Type", "application/json")client := &http.Client{}resp, err := client.Do(req)if err != nil {fmt.Println("HTTP PUT 请求失败:", err)return}defer resp.Body.Close()body, err := ioutil.ReadAll(resp.Body)if err != nil {fmt.Println("读取响应体失败:", err)return}fmt.Println("PUT 响应：", string(body))
}

HTTP 服务器

使用 net/http 包进行 HTTP 服务器编程，可以创建一个 HTTP 服务器来处理客户端的请求。
使用 http.HandleFunc 方法来注册处理不同路由的处理函数，或者使用自定义的 http.Handler 对象处理请求。
可以通过设置路由和处理逻辑来实现 RESTful API、静态文件服务等功能。

package mainimport ("fmt""net/http"
)func main() {// 注册处理函数http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {fmt.Fprint(w, "Hello, HTTP client!")})// 启动 HTTP 服务器err := http.ListenAndServe(":8080", nil)if err != nil {fmt.Println("启动服务器失败:", err)return}
}

WebSocket 通信

使用 net/http 包进行 WebSocket 通信，可以创建一个 WebSocket 服务器和客户端。
服务器可以使用 http.HandleFunc 方法来注册 WebSocket 的处理函数，处理客户端的连接和消息。
客户端可以使用 gorilla/websocket 等第三方库来建立 WebSocket 连接，并进行消息的发送和接收。

WebSocket 服务器

package mainimport ("fmt""log""net/http""github.com/gorilla/websocket"
)func main() {// 注册 WebSocket 处理函数http.HandleFunc("/", handleWebSocket)// 启动 WebSocket 服务器err := http.ListenAndServe(":8080", nil)if err != nil {log.Fatal("启动服务器失败:", err)}
}func handleWebSocket(w http.ResponseWriter, r *http.Request) {// 升级 HTTP 连接为 WebSocket 连接upgrader := websocket.Upgrader{}conn, err := upgrader.Upgrade(w, r, nil)if err != nil {log.Println("升级连接失败:", err)return}defer conn.Close()fmt.Println("客户端已连接")// 处理 WebSocket 消息for {// 读取消息_, message, err := conn.ReadMessage()if err != nil {log.Println("读取消息失败:", err)break}fmt.Println("接收到消息:", string(message))// 发送消息err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, WebSocket client!"))if err != nil {log.Println("发送消息失败:", err)break}}fmt.Println("客户端已断开连接")
}

WebSocket 客户端（使用 JavaScript）

const socket = new WebSocket("ws://localhost:8080");// 连接建立时触发
socket.onopen = function() {console.log("连接已建立");// 发送消息socket.send("Hello, WebSocket server!");
};// 接收消息时触发
socket.onmessage = function(event) {console.log("接收到消息:", event.data);
};// 连接关闭时触发
socket.onclose = function(event) {console.log("连接已关闭");
};