Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Go net/http包

详解Go语言中net/http包的使用

作者：242030

Go语言内置的 net/http 包十分的优秀，提供了HTTP客户端和服务端的实现，本文主要就来和大家聊聊net/http包的使用，感兴趣的可以了解一下

Http 协议(Hyper Text Transfer Protocol，超文本传输协议)是一个简单的请求-响应协议，它通常运行在 TCP 之上。超文本传输协议是互联网上应用最为广泛的一种网络传输协议，所有的WWW文件都必须遵守这个标准。

Http 协议是基于客户端 Cilent /服务器 Server 模式，且面向连接的。简单的来说就是客户端 Cilent 向服务器Server 发送 http 请求 Request，服务器 Server 接收到 http 服务请求 Request 后会在 http 响应Response 中回送所请求的数据。

Go语言内置的 net/http 包十分的优秀，提供了HTTP客户端和服务端的实现。

官网地址：https://pkg.go.dev/net/http

1、HTTP客户端

通过 Get、Head、Post 和 PostForm 函数发出 HTTP/HTTPS 请求。

resp, err := http.Get("http://example.com/")
resp, err := http.Post("http://example.com/upload", "image/jpeg", &buf)
resp, err := http.PostForm("http://example.com/form",url.Values{"key": {"Value"}, "id": {"123"}})

1.1 发送Get请求

不带请求参数：

package main
import (
	"fmt"
	"io/ioutil"
	"net/http"
)
func main() {
	resp, err := http.Get("http://www.baidu.com")
	if err != nil {
		fmt.Printf("get failed, err:%v\n", err)
		return
	}
	// 程序在使用完response后必须关闭回复的主体
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Printf("read from resp.Body failed, err:%v\n", err)
		return
	}
	fmt.Print(string(body))
}

带参数，参数需要使用到 net/url 来处理。

package main
import (
	"fmt"
	"io/ioutil"
	"net/http"
	"net/url"
)
func main() {
	apiUrl := "http://127.0.0.1:9090/get"
	// URL param
	data := url.Values{}
	data.Set("name", "mi")
	data.Set("age", "18")
	u, err := url.ParseRequestURI(apiUrl)
	if err != nil {
		fmt.Printf("parse url requestUrl failed, err:%v\n", err)
	}
	// URL encode
	u.RawQuery = data.Encode()
	// http://127.0.0.1:9090/get?age=18&name=mi
	fmt.Println(u.String())
	resp, err := http.Get(u.String())
	if err != nil {
		fmt.Printf("post failed, err:%v\n", err)
		return
	}
	defer resp.Body.Close()
	b, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Printf("get resp failed, err:%v\n", err)
		return
	}
	// {"status": "ok"}
	fmt.Println(string(b))
}

对应 server 端的处理：

package main
import (
	"fmt"
	"net/http"
)
func getHandler(w http.ResponseWriter, r *http.Request) {
	defer r.Body.Close()
	data := r.URL.Query()
    // mi
	fmt.Println(data.Get("name"))
    // 18
	fmt.Println(data.Get("age"))
	answer := `{"status": "ok"}`
	w.Write([]byte(answer))
}
func main() {
	http.HandleFunc("/get", getHandler)
	err := http.ListenAndServe(":9090", nil)
	if err != nil {
		fmt.Printf("http server failed, err:%v\n", err)
		return
	}
}

带 header 头部信息：

package main
import (
	"fmt"
	"net/http"
	"net/http/httputil"
)
func main() {
	url := "http://127.0.0.1:9090/get"
	request, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		panic(err)
	}
	request.Header.Add("Authorization", "jhs8723sd2dshd2")
	request.Header.Add("User-Agent", "User-Agent: Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/13.0.3 Mobile/15E148 Safari/604.1")
	resp, err := http.DefaultClient.Do(request)
	if err != nil {
		panic(err)
	}
	defer resp.Body.Close()
	// //获取网页内容
	s, err := httputil.DumpResponse(resp, true)
	if err != nil {
		panic(err)
	}
	fmt.Printf("%s", s)
}

1.2 发送POST请求

package main
import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)
func main() {
	url := "http://127.0.0.1:9090/post"
	// 表单数据
	// contentType := "application/x-www-form-urlencoded"
	// data := "name=mi&age=18"
	// json
	contentType := "application/json"
	data := `{"name":"mi","age":18}`
	resp, err := http.Post(url, contentType, strings.NewReader(data))
	if err != nil {
		fmt.Printf("post failed, err:%v\n", err)
		return
	}
	defer resp.Body.Close()
	b, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Printf("get resp failed, err:%v\n", err)
		return
	}
	// {"status": "ok"}
	fmt.Println(string(b))
}

对应 server 端：

package main
import (
	"fmt"
	"io/ioutil"
	"net/http"
)
func postHandler(w http.ResponseWriter, r *http.Request) {
	defer r.Body.Close()
	// 1. 请求类型是application/x-www-form-urlencoded时解析form数据
	r.ParseForm()
	// 打印form数据
	// map[]
	fmt.Println(r.PostForm)
	fmt.Println(r.PostForm.Get("name"), r.PostForm.Get("age"))
	// 2. 请求类型是application/json时从r.Body读取数据
	b, err := ioutil.ReadAll(r.Body)
	if err != nil {
		fmt.Printf("read request.Body failed, err:%v\n", err)
		return
	}
	// {"name":"mi","age":18}
	fmt.Println(string(b))
	answer := `{"status": "ok"}`
	w.Write([]byte(answer))
}
func main() {
	http.HandleFunc("/post", postHandler)
	err := http.ListenAndServe(":9090", nil)
	if err != nil {
		fmt.Printf("http server failed, err:%v\n", err)
		return
	}
}

1.3 PostForm方式

package main
import (
	"fmt"
	"io/ioutil"
	"net/http"
	"net/url"
)
func main() {
	resp, err := http.PostForm("http://127.0.0.1:9090/post",
		url.Values{"key": {"Value"}, "id": {"123"}})
	if err != nil {
		fmt.Println(err)
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println(err)
	}
	fmt.Println(string(body))
}

json：

package main
import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
)
func main() {
	data := make(map[string]interface{})
	data["site"] = "www.baidu.com"
	data["name"] = "tom"
	bytesData, _ := json.Marshal(data)
	resp, _ := http.Post("http://httpbin.org/post", "application/json", bytes.NewReader(bytesData))
	body, _ := ioutil.ReadAll(resp.Body)
	fmt.Println(string(body))
}

带有 headers 参数：

package main
import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)
func main() {
	client := &http.Client{}
	req, err := http.NewRequest("POST", "http://www.01happy.com/demo/accept.php", strings.NewReader("name=cjb"))
	if err != nil {
		fmt.Println(err)
	}
	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	req.Header.Set("Cookie", "name=anny")
	resp, err := client.Do(req)
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println(err)
	}
	fmt.Println(string(body))
}

1.4 自定义Client

要管理 HTTP 客户端的头域、重定向策略和其他设置，创建一个 Client：

package main
import (
	"fmt"
	"net/http"
	"net/http/httputil"
)
func main() {
	request, err := http.NewRequest(http.MethodGet, "http://www.baidu.com", nil)
	if err != nil {
		panic(err)
	}
	client := http.Client{
		CheckRedirect: func(req *http.Request, via []*http.Request) error {
			fmt.Println("Redirect:", req)
			return nil
		},
	}
	request.Header.Add("User-Agent", "User-Agent: Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/13.0.3 Mobile/15E148 Safari/604.1")
	resp, err := client.Do(request)
	if err != nil {
		panic(err)
	}
	defer resp.Body.Close()
	s, err := httputil.DumpResponse(resp, true)
	fmt.Printf("%s", s)
}

1.5 自定义Transport

要管理代理、TLS 配置、keep-alive、压缩和其他设置，创建一个Transport：

package main
import (
	"crypto/tls"
	"crypto/x509"
	"fmt"
	"io/ioutil"
	"net/http"
)
// 使用匹配的证书方法
func main() {
	// *.pem文件的内容
	rootCA := ""
	roots := x509.NewCertPool()
	ok := roots.AppendCertsFromPEM([]byte(rootCA))
	if !ok {
		panic("failed to parse root certificate")
	}
	client := &http.Client{
		Transport: &http.Transport{
			TLSClientConfig:    &tls.Config{RootCAs: roots},
			DisableCompression: true,
		},
	}
	resp, err := client.Get("http://baidu.com")
	if err != nil {
		panic(err)
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		panic(err)
	}
	if err != nil {
		fmt.Printf("read from resp.Body failed, err:%v\n", err)
		return
	}
	fmt.Print(string(body))
}

Client 和 Transport 类型都可以安全的被多个 goroutine 同时使用。出于效率考虑，应该一次建立、尽量重用。

package main
import (
	"crypto/tls"
	"fmt"
	"io/ioutil"
	"net/http"
)
// 跳过安全检查
func main() {
	tr := &http.Transport{
		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
	}
	client := &http.Client{Transport: tr}
	resp, err := client.Get("https://localhost:8081")
	if err != nil {
		fmt.Println("error:", err)
		return
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	fmt.Println(string(body))
}

2、服务端

2.1 默认的Server

ListenAndServe 使用指定的监听地址和处理器启动一个 HTTP 服务端，处理器参数通常是 nil，这表示采用包变量

DefaultServeMux 作为处理器。

Handle 和 HandleFunc 函数可以向 DefaultServeMux 添加处理器。

package main
import (
	"log"
	"net/http"
)
type httpServer struct {
}
func (server httpServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	w.Write([]byte(r.URL.Path))
}
func main() {
	var server httpServer
	http.Handle("/foo", server)
	http.HandleFunc("/bar", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte(r.URL.Path))
	})
	log.Fatal(http.ListenAndServe(":9090", nil))
}

事件处理器的 Handler 接口定义如下：

type Handler interface {
	ServeHTTP(ResponseWriter, *Request)
}

只要实现了这个接口，就可以实现自己的 handler 处理器。

如果 ListenAndServe() 传入的第一个参数地址为空，则服务器在启动后默认使用 http://127.0.0.1:8080 地址进行访问。

如果这个函数传入的第二个参数为 nil，则服务器在启动后将使用默认的多路复用器DefaultServeMux。

2.2 默认的Server示例

使用 Go 语言中的 net/http 包来编写一个简单的接收 HTTP 请求的 Server 端示例，net/http 包是对 net 包的

进一步封装，专门用来处理 HTTP 协议的数据。具体的代码如下：

package main
import (
	"fmt"
	"net/http"
)
func sayHello(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintln(w, "Hello shanghai！")
}
func main() {
	http.HandleFunc("/", sayHello)
	err := http.ListenAndServe(":9090", nil)
	if err != nil {
		fmt.Printf("http server failed, err:%v\n", err)
		return
	}
}

将上面的代码编译之后执行，打开你电脑上的浏览器在地址栏输入127.0.0.1:9090回车：

输出

Hello shanghai！

package main
import (
	"encoding/json"
	_ "encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/url"
	"strings"
)
var (
	api   = "http://127.0.0.1:9090/"
	token = "9adf92861"
)
type Result struct {
	Code int    `json:"code"`
	Msg  string `json:"msg"`
}
type Resp struct {
	Code string `json:"code"`
	Msg  string `json:"msg"`
}
func wechatUrlHandle(w http.ResponseWriter, r *http.Request) {
	// 获取链接地址
	var wechatUrl string
	values := r.URL.Query()
	wechatUrl = values.Get("url")
	// 验证URL的正确性
	targetUrl, err := url.ParseRequestURI(wechatUrl)
	if err != nil {
		fmt.Printf("parse url failed, err:%v\n", err)
	}
	fmt.Printf(fmt.Sprintf("url: %v", targetUrl))
	// 拼接参数发起请求
	data := fmt.Sprintf("token=%s&url=%s", token, targetUrl.String())
	headers := "application/x-www-form-urlencoded"
	res, err := http.Post(api, headers, strings.NewReader(data))
	if err != nil {
		fmt.Printf("post failed, err: %v\n", err)
		return
	}
	// 关闭请求资源
	defer res.Body.Close()
	body, err := ioutil.ReadAll(res.Body)
	if err != nil {
		fmt.Fprintf(w, fmt.Sprintf("get resp failed. err: %v\n", err))
		return
	}
	var info Result
	var ress Resp
	w.Header().Set("Content-Type", "application/json")
	if err := json.Unmarshal(body, &info); err == nil {
		fmt.Println(info.Code, info.Msg)
		ress.Code = "200"
		ress.Msg = "Success"
		t, _ := json.Marshal(ress)
		w.Write(t)
	} else {
		ress.Code = "200"
		ress.Msg = "failure"
		t, _ := json.Marshal(ress)
		fmt.Println("err: ", err)
		w.Write(t)
	}
}
func main() {
	http.HandleFunc("/wechat/url", wechatUrlHandle)
	// 处理器参数一般为nil
	err := http.ListenAndServe(":9090", nil)
	if err != nil {
		fmt.Printf("Http server failed. err: %v\n", err)
		return
	}
}

2.3 自定义Server

要管理服务端的行为，可以创建一个自定义的 Server。

用户可以通过 Server 结构体对服务器进行更详细的配置，包括设置地址，为请求读取操作设置超过时间等等。

package main
import (
	"log"
	"net/http"
	"time"
)
type httpServer struct {
}
func (server httpServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	w.Write([]byte(r.URL.Path))
}
func main() {
	s := &http.Server{
		Addr:           ":9090",
		Handler:        httpServer{},
		ReadTimeout:    10 * time.Second,
		WriteTimeout:   10 * time.Second,
		MaxHeaderBytes: 1 << 20,
	}
	log.Fatal(s.ListenAndServe())
}

2.4 多路复用器

多路复用器的基本原理：根据请求的 URL 地址找到对应的处理器，调用处理器对应的 ServeHTTP() 方法处理请求。

DefaultServeMux 是 net/http 包的默认多路复用器，其实就是 ServeMux 的一个实例。

HandleFunc() 函数用于为指定的 URL 注册一个处理器，HandleFunc() 处理器函数会在内部调用DefaultServeMux 对象对应的 HandleFunc 方法。

2.4.1 ServeMux与DefaultServeMux

我们可以使用默认多路复用器注册多个处理器，达到与处理器一样的作用。

下面演示如何通过默认多路复用器创建自己的服务器。

package main
import (
	"fmt"
	"net/http"
)
//定义多个处理器
type handle1 struct{}
func (h1 *handle1) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "handle one")
}
type handle2 struct{}
func (h2 *handle2) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "handle two")
}
func main() {
	handle1 := handle1{}
	handle2 := handle2{}
	// Handler:nil表明服务器使用默认的多路复用器DefaultServeMux
	s := &http.Server{
		Addr:    "127.0.0.1:8080",
		Handler: nil,
	}
	// Handle函数调用的是多路复用器DefaultServeMux.Handle方法
	http.Handle("/handle1", &handle1)
	http.Handle("/handle2", &handle2)
	s.ListenAndServe()
}

我们通过使用自己的 handle1 和 handle2 来指定两个处理器，http.Handle() 函数可以调用DefaultServeMux.Handle() 方法来处理请求。

Handle: nil 对应的是处理器是DefaultServeMux。

在ServeMux对象的ServeHTTP()方法中，根据URL查找我们注册的服务器然后请求交给它处理。

虽然默认的多路复用器很好用，但仍然不推荐使用，因为它是一个全局变量，所有的代码都可以修改它。有些第三方库中可能与默认复用器产生冲突。所以推荐的做法是自定义。

2.4.2 自定义多路复用器

package main
import (
	"fmt"
	"net/http"
)
func newservemux(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "NewServeMux")
}
func main() {
	mux := http.NewServeMux()
	mux.HandleFunc("/", newservemux)
	s := &http.Server{
		Addr:    ":8081",
		Handler: mux,
	}
	s.ListenAndServe()
}

NewServeMux 实质上还是 ServeMux。

2.4.3 ServeMux的路由匹配

我们现在需要绑定三个URL分别为/，/happy，/bad。

package main
import (
	"fmt"
	"net/http"
)
func newservemux(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "NewServeMux")
}
func newservemuxhappy(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "Newservemuxhappy")
}
func newservemuxbad(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "NewServeMuxbad")
}
func main() {
	mux := http.NewServeMux()
	mux.HandleFunc("/", newservemux)
	mux.HandleFunc("/happy", newservemuxhappy)
	mux.HandleFunc("/bad", newservemuxbad)
	s := &http.Server{
		Addr:    ":8080",
		Handler: mux,
	}
	s.ListenAndServe()
}

2.5 HttpRouter

ServeMux 的一个缺陷是无法使用变量实现 URL 模式匹配，而 HttpRouter 可以，HttpRouter 是一个高性能的第

三方 HTTP 路由包，弥补了 net/http 包中的路由不足问题。

$ go get github.com/julienschmidt/httprouter

httprouter 的使用首先得使用 New() 函数，生成一个 *Router 路由对象，然后使用 GET()，方法去注册匹配的函

数，最后再将这个参数传入 http.ListenAndServe 函数就可以监听服务。

package main
import (
	"net/http"
	"github.com/julienschmidt/httprouter"
)
func Hello(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	w.Write([]byte("Hello,httprouter!"))
}
func main() {
	router := httprouter.New()
	router.GET("/", Hello)
	http.ListenAndServe(":8080", router)
}

更为重要的是，它为 URL 提供了两种匹配模式：

Pattern: /user/:user
 /user/gordon              match
 /user/you                 match
 /user/gordon/profile      no match
 /user/                    no match
Pattern: /src/*filepath
 /src/                     match
 /src/somefile.go          match
 /src/subdir/somefile.go   match
package main
import (
	"fmt"
	"log"
	"net/http"
	"github.com/julienschmidt/httprouter"
)
func Index(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	fmt.Fprint(w, "Welcome!\n")
}
func Hello(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
	fmt.Fprintf(w, "hello, %s!\n", ps.ByName("name"))
}
func main() {
	router := httprouter.New()
	router.GET("/", Index)
	router.GET("/hello/:name", Hello)
	log.Fatal(http.ListenAndServe(":8080", router))
}

当然这里还有 POST()，DELETE() 函数的详情，就不一一介绍了。

3、其它应用

3.1 DNS

package main
import (
	"fmt"
	"log"
	"net"
)
func main() {
	// 根据域名返回ip地址
	addr, err := net.ResolveIPAddr("ip", "devops.miliantech.com")
	if err != nil {
		return
	}
	// 2023/07/24 20:45:43 devops.miliantech.com 对应地址ip地址----> 172.21.6.96
	log.Println("devops.miliantech.com 对应地址ip地址---->", addr)
	// 查找DNS A记录
	iprecords, _ := net.LookupIP("devops.miliantech.com")
	for _, ip := range iprecords {
		// LookupIP -----> 172.21.6.96
		fmt.Println("LookupIP ----->", ip)
	}
	//查找DNS CNAME记录
	canme, _ := net.LookupCNAME("devops.miliantech.com")
	// LookupCNAME -----> devops.miliantech.com.
	fmt.Println("LookupCNAME ----->", canme)
	//查找DNS PTR记录
	ptr, e := net.LookupAddr("172.21.6.96")
	if e != nil {
		// lookup 172.21.6.96: dnsquery: DNS name does not exist.
		fmt.Println(e)
	} else {
		fmt.Println(ptr)
	}
	for _, ptrval := range ptr {
		fmt.Println(ptrval)
	}
	//查找DNS NS记录 名字服务器记录
	nameserver, _ := net.LookupNS("baidu.com")
	for _, ns := range nameserver {
		/*
			ns记录 &{ns3.baidu.com.}
			ns记录 &{ns2.baidu.com.}
			ns记录 &{ns7.baidu.com.}
			ns记录 &{ns4.baidu.com.}
			ns记录 &{dns.baidu.com.}
		*/
		fmt.Println("ns记录", ns)
	}
	//查找DNS MX记录 邮件服务器记录
	mxrecods, _ := net.LookupMX("google.com")
	for _, mx := range mxrecods {
		fmt.Println("mx:", mx)
	}
	//查找DNS TXT记录 域名对应的文本信息
	txtrecords, _ := net.LookupTXT("baidu.com")
	for _, txt := range txtrecords {
		/*
			txt: _globalsign-domain-verification=qjb28W2jJSrWj04NHpB0CvgK9tle5JkOq-EcyWBgnE
			txt: google-site-verification=GHb98-6msqyx_qqjGl5eRatD3QTHyVB6-xQ3gJB5UwM
			txt: v=spf1 include:spf1.baidu.com include:spf2.baidu.com include:spf3.baidu.com include:spf4.baidu.com mx ptr -all
		*/
		fmt.Println("txt:", txt)
	}
	//查看本地IP地址
	addrs, err := net.InterfaceAddrs()
	if err != nil {
		return
	}
	for _, address := range addrs {
		// 检查ip地址判断是否回环地址
		if ipnet, ok := address.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
			if ipnet.IP.To4() != nil {
				/*
					本地地址 -- > 169.254.124.27
					本地地址 -- > 169.254.27.66
					本地地址 -- > 169.254.158.118
					本地地址 -- > 192.168.223.1
					本地地址 -- > 192.168.132.1
					本地地址 -- > 192.168.226.185
				*/
				fmt.Println("本地地址 -- >", ipnet.IP.String())
			}
		}
	}
}

3.2 简单的tcp请求实现

3.2.1 服务端

package main
import (
	"bufio"
	"io"
	"log"
	"net"
)
func pes(conn net.Conn) {
	defer conn.Close()
	read := bufio.NewReader(conn)
	for {
		var p [512]byte
		n, err := read.Read(p[:])
		if err != nil {
			if err == io.EOF {
				log.Println("客户端关闭链接", conn.RemoteAddr().String())
				return
			}
			log.Println("读取失败", err.Error())
			return
		}
		log.Println("客户端发送消息为", string(p[:n]))
		conn.Write([]byte("ok"))
	}
}
func main() {
	lis, err := net.Listen("tcp", "127.0.0.1:20003")
	if err != nil {
		log.Println("listen ,err ", err.Error())
		return
	}
	for {
		conn, err := lis.Accept()
		if err != nil {
			log.Println("建立链接异常", err.Error())
		}
		go pes(conn)
	}
}

3.2.2 客户端

package main
import (
	"bufio"
	"io"
	"log"
	"net"
	"os"
	"strings"
)
func main() {
	conn, err := net.Dial("tcp", "127.0.0.1:20003")
	if err != nil {
		log.Println("conn err", err.Error())
		return
	}
	defer conn.Close()
	inputreader := bufio.NewReader(os.Stdin)
	for {
		input, _ := inputreader.ReadString('\n')
		input = strings.Trim(input, "\r\n")
		if input == "q" || input == "Q" {
			log.Println("退出链接")
			return
		}
		conn.Write([]byte(input))
		var p [512]byte
		n, err := conn.Read(p[:])
		if err != nil {
			if err == io.EOF {
				log.Println("服务端关闭链接", conn.RemoteAddr().String())
				return
			}
			log.Println("读取失败", err.Error())
			return
		}
		log.Println("服务端发送消息为", string(p[:n]))
	}
}

3.3 粘包问题

一次只能读取固定长度的数据，解决方案：

package main
import (
	"bufio"
	"bytes"
	"encoding/binary"
	"fmt"
)
func Encode(massge string) ([]byte, error) {
	len_for_msaaget := int32(len(massge))
	pkg := new(bytes.Buffer)
	err := binary.Write(pkg, binary.LittleEndian, len_for_msaaget)
	if err != nil {
		return nil, err
	}
	err = binary.Write(pkg, binary.LittleEndian, []byte(massge))
	if err != nil {
		return nil, err
	}
	return pkg.Bytes(), nil
}
func Decode(reader *bufio.Reader) (string, error) {
	peek, err := reader.Peek(4)
	if err != nil {
		return "", err
	}
	lengthBuff := bytes.NewBuffer(peek)
	var length int32
	err = binary.Read(lengthBuff, binary.LittleEndian, &length)
	if err != nil {
		return "", err
	}
	if int32(reader.Buffered()) <= 4+length {
		return "", fmt.Errorf("err")
	}
	p := make([]byte, int(4+length))
	_, err = reader.Read(p)
	if err != nil {
		return "", err
	}
	return string(p[4:]), nil
}

3.4 简单的udp请求实现

3.4.1 服务端

package main
import (
	"log"
	"net"
)
func main() {
	udp, err := net.ListenUDP("udp", &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		Port: 30000,
	})
	if err != nil {
		return
	}
	defer udp.Close()
	for {
		var p [1024]byte
		n, addr, err := udp.ReadFromUDP(p[:])
		if err != nil {
			log.Println("udp 请求处理异常 ", err.Error())
			return
		}
		log.Println("请求信息", string(p[:n]), addr.String())
		_, err = udp.WriteToUDP([]byte("get"), addr)
		if err != nil {
			log.Println("返回信息失败", err.Error())
			return
		}
	}
}

3.4.2 客户端

package main
import (
	"log"
	"net"
)
func main() {
	udp, err := net.DialUDP("udp", nil, &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		Port: 30000,
	})
	if err != nil {
		log.Println(err.Error())
		return
	}
	defer udp.Close()
	_, err = udp.Write([]byte("我是服务端， hello server"))
	if err != nil {
		log.Println(err.Error())
		return
	}
	data := make([]byte, 4096)
	fromUDP, u, err := udp.ReadFromUDP(data)
	if err != nil {
		log.Println("接收失败", err)
		return
	}
	log.Println("接收信息返回成功 ", string([]byte(data[:fromUDP])), u)
}

到此这篇关于详解Go语言中net/http包的使用的文章就介绍到这了,更多相关Go net/http包内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:
阅读全文