Go語言網(wǎng)絡庫net/http

Http 協(xié)議(Hyper Text Transfer Protocol，超文本傳輸協(xié)議)是一個簡單的請求-響應協(xié)議，它通常運行在 TCP 之

上。超文本傳輸協(xié)議是互聯(lián)網(wǎng)上應用最為廣泛的一種網(wǎng)絡傳輸協(xié)議，所有的WWW文件都必須遵守這個標準。

Http 協(xié)議是基于客戶端 Cilent /服務器 Server 模式，且面向連接的。簡單的來說就是客戶端 Cilent 向服務器

Server 發(fā)送 http 請求 Request，服務器 Server 接收到 http 服務請求 Request 后會在 http 響應Response 中回

送所請求的數(shù)據(jù)。

Go語言內(nèi)置的 net/http 包十分的優(yōu)秀，提供了HTTP客戶端和服務端的實現(xiàn)。

官網(wǎng)地址：https://pkg.go.dev/net/http

1、HTTP客戶端

通過 Get、Head、Post 和 PostForm 函數(shù)發(fā)出 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 發(fā)送Get請求

不帶請求參數(shù)：

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))
}

帶參數(shù)，參數(shù)需要使用到 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()
	// //獲取網(wǎng)頁內(nèi)容
	s, err := httputil.DumpResponse(resp, true)
	if err != nil {
		panic(err)
	}
	fmt.Printf("%s", s)
}

1.2 發(fā)送POST請求

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)

func main() {
	url := "http://127.0.0.1:9090/post"
	// 表單數(shù)據(jù)
	// 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數(shù)據(jù)
	r.ParseForm()
	// 打印form數(shù)據(jù)
	// map[]
	fmt.Println(r.PostForm)
	fmt.Println(r.PostForm.Get("name"), r.PostForm.Get("age"))
	// 2. 請求類型是application/json時從r.Body讀取數(shù)據(jù)
	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 參數(shù)：

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 客戶端的頭域、重定向策略和其他設置，創(chuàng)建一個 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、壓縮和其他設置，創(chuàng)建一個Transport：

package main

import (
	"crypto/tls"
	"crypto/x509"
	"fmt"
	"io/ioutil"
	"net/http"
)

// 使用匹配的證書方法
func main() {
	// *.pem文件的內(nèi)容
	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 使用指定的監(jiān)聽地址和處理器啟動一個 HTTP 服務端，處理器參數(shù)通常是 nil，這表示采用包變量

DefaultServeMux 作為處理器。

Handle 和 HandleFunc 函數(shù)可以向 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)
}

只要實現(xiàn)了這個接口，就可以實現(xiàn)自己的 handler 處理器。

如果 ListenAndServe() 傳入的第一個參數(shù)地址為空，則服務器在啟動后默認使用 http://127.0.0.1:8080 地址

進行訪問。

如果這個函數(shù)傳入的第二個參數(shù)為 nil，則服務器在啟動后將使用默認的多路復用器DefaultServeMux。

2.2 默認的Server示例

使用 Go 語言中的 net/http 包來編寫一個簡單的接收 HTTP 請求的 Server 端示例，net/http 包是對 net 包的

進一步封裝，專門用來處理 HTTP 協(xié)議的數(shù)據(jù)。具體的代碼如下：

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
	}
}

將上面的代碼編譯之后執(zhí)行，打開你電腦上的瀏覽器在地址欄輸入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))
	// 拼接參數(shù)發(fā)起請求
	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)
	// 處理器參數(shù)一般為nil
	err := http.ListenAndServe(":9090", nil)
	if err != nil {
		fmt.Printf("Http server failed. err: %v\n", err)
		return
	}

}

2.3 自定義Server

要管理服務端的行為，可以創(chuàng)建一個自定義的 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 多路復用器

多路復用器的基本原理：根據(jù)請求的 URL 地址找到對應的處理器，調(diào)用處理器對應的 ServeHTTP() 方法處理請

求。

DefaultServeMux 是 net/http 包的默認多路復用器，其實就是 ServeMux 的一個實例。

HandleFunc() 函數(shù)用于為指定的 URL 注冊一個處理器，HandleFunc() 處理器函數(shù)會在內(nèi)部調(diào)用

DefaultServeMux 對象對應的 HandleFunc 方法。

2.4.1 ServeMux與DefaultServeMux

我們可以使用默認多路復用器注冊多個處理器，達到與處理器一樣的作用。

下面演示如何通過默認多路復用器創(chuàng)建自己的服務器。

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函數(shù)調(diào)用的是多路復用器DefaultServeMux.Handle方法
	http.Handle("/handle1", &handle1)
	http.Handle("/handle2", &handle2)
	s.ListenAndServe()
}

我們通過使用自己的 handle1 和 handle2 來指定兩個處理器，http.Handle() 函數(shù)可以調(diào)用

DefaultServeMux.Handle() 方法來處理請求。

Handle: nil 對應的是處理器是DefaultServeMux。

在ServeMux對象的ServeHTTP()方法中，根據(jù)URL查找我們注冊的服務器然后請求交給它處理。

雖然默認的多路復用器很好用，但仍然不推薦使用，因為它是一個全局變量，所有的代碼都可以修改它。有些第三

方庫中可能與默認復用器產(chǎn)生沖突。所以推薦的做法是自定義。

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 實質(zhì)上還是 ServeMux。

2.4.3 ServeMux的路由匹配

我們現(xiàn)在需要綁定三個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 的一個缺陷是無法使用變量實現(xiàn) URL 模式匹配，而 HttpRouter 可以，HttpRouter 是一個高性能的第

三方 HTTP 路由包，彌補了 net/http 包中的路由不足問題。

$ go get github.com/julienschmidt/httprouter

httprouter 的使用首先得使用 New() 函數(shù)，生成一個 *Router 路由對象，然后使用 GET()，方法去注冊匹配的函

數(shù)，最后再將這個參數(shù)傳入 http.ListenAndServe 函數(shù)就可以監(jiān)聽服務。

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 提供了兩種匹配模式：

• /user/:pac：精準匹配 /user/pac
• /user/*pac：匹配所有模式 /user/hello

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() 函數(shù)的詳情，就不一一介紹了。

3、其它應用

3.1 DNS

package main

import (
	"fmt"
	"log"
	"net"
)

func main() {
	// 根據(jù)域名返回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地址判斷是否回環(huán)地址
		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請求實現(xiàn)

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("客戶端發(fā)送消息為", 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("服務端發(fā)送消息為", string(p[:n]))
	}

}

3.3 粘包問題

一次只能讀取固定長度的數(shù)據(jù)，解決方案：文章來源地址http://www.zghlxwxcb.cn/news/detail-606203.html

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請求實現(xiàn)

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語言網(wǎng)絡庫net/http的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章，希望大家以后多多支持TOY模板網(wǎng)！