当前位置：首页 > article >正文

第十二章标准库——7.net包

article 2025/1/13 1:19:42

一、引言

Go语言的net包提供了对网络和RPC（远程过程调用）编程的支持。通过net包，开发者可以创建HTTP服务、TCP服务器与客户端，以及实现RPC服务。本文将通过详细的示例，讲解如何使用net包进行这些操作。

二、创建HTTP服务

导入必要的包

package main
import "net/http"

定义处理器函数
处理器函数是一个接受http.ResponseWriter和*http.Request参数的函数。http.ResponseWriter用于向客户端发送响应，而*http.Request包含了客户端的请求信息。

var helloHandler = func(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("hello world"))
    return
}

注册处理器并启动服务
使用http.HandleFunc注册处理器函数，然后使用http.ListenAndServe启动HTTP服务。

func main() {
    http.HandleFunc("/hello", helloHandler)
    err := http.ListenAndServe("localhost:8080", nil)
    if err != nil {
        panic("error:" + err.Error())
    }
}

测试服务
启动服务后，可以使用curl命令测试：

curl localhost:8080/hello

输出应为hello world。

三、创建RPC服务

定义服务类型和方法
RPC服务需要定义一个服务类型，并为该类型定义可供远程调用的方法。

package main
import (
    "net"
    "net/rpc"
)
 
type UserService struct{}
 
func (s *UserService) Say(request string, reply *string) error {
    *reply = "hi," + request
    return nil
}

注册服务并启动RPC服务器
使用rpc.RegisterName注册服务，然后使用net.Listen和listener.Accept创建TCP监听器，并使用rpc.ServeConn处理连接。

func main() {
    var err error
    err = rpc.RegisterName("UserService", new(UserService))
    if err != nil {
        panic(err)
    }
    listener, err := net.Listen("tcp", ":8011")
    if err != nil {
        panic(err)
    }
    conn, err := listener.Accept()
    if err != nil {
        panic(err)
    }
    rpc.ServeConn(conn)
}

四、创建RPC客户端

连接到RPC服务器并调用远程方法
使用rpc.Dial连接到RPC服务器，然后使用client.Call调用远程方法。

package main
import (
    "fmt"
    "net/rpc"
    "testing"
)
 
func TestClient(t *testing.T) {
    client, err := rpc.Dial("tcp", "localhost:8011")
    if err != nil {
        panic(err)
    }
    var reply string
    err = client.Call("UserService.Say", "tom", &reply)
    if err != nil {
        panic(err)
    }
    fmt.Println(reply)
}