目录
go往期文章笔记:
这篇记的是网络编程相关的笔记。
TCP
开启服务端,监听连接的请求。示意图:
服务端:启动后,监听连接过来的客户端,每连接过来一个客户端,开启一个协程去和客户端进行交互(一个协程代表一个客户端)
客户端:根据服务端的IP和端口号去访问服务端,连接成功后可以向服务端发送消息。
连接的时候通过IP和端口号去连接,但是和客户端进行交流的是协程。
一个简单案例
创建两个文件夹,一个是server、一个是client,然后两个文件夹里面分别创建 server.go、client.go
文件,如下图:
server.go 服务端
package main
import (
"fmt"
"io"
"net"
)
// tcp处理客户端发送过来的信息
func tcpHandler(conn net.Conn) {
defer conn.Close()
for {
var buf []byte = make([]byte, 1024)
// 等待客户端通过conn发送信息,如果客户端没有wrtie[发送],那么协程就阻塞在这里
n, err := conn.Read(buf)
if err == io.EOF {
fmt.Println(conn.RemoteAddr().String() + " 退出连接")
break
}
// 客户端发送的内容显示到服务器的终端
fmt.Print(string(buf[:n]))
}
}
func main() {
TCP服务端()
}
func TCP服务端() {
fmt.Println("服务器开始监听")
listen, err := net.Listen("tcp", "0.0.0.0:8888")
if err != nil {
fmt.Println(err)
return
}
defer listen.Close()
for { // 循环等待客户端来连接
conn, err := listen.Accept()
if err != nil {
fmt.Println(err)
} else {
fmt.Printf("客户端 %v 连接成功\n", conn.RemoteAddr().String())
}
// 这里开启协程
go tcpHandler(conn)
}
}
client.go 客户端
package main
import (
"fmt"
"net"
)
func main() {
TCP客户端()
}
func TCP客户端() {
conn, err := net.Dial("tcp", "127.0.0.1:8888")
if err != nil {
fmt.Println(err)
return
}
defer conn.Close()
fmt.Println(conn.RemoteAddr().String(), "连接成功")
var s string
for { // 循环输入,给服务端发送消息
// 从终端读取用户输入,并发送给服务器
fmt.Scanln(&s)
if s == "exit" {
fmt.Println(conn.RemoteAddr().String(), "退出连接")
break
}
conn.Write([]byte(s + "\r\n")) // 发送给服务端
}
}
写好之后,我们先启动服务端,然后再启动客户端。在客户端里面输入消息发送给服务端,服务端接收后要将消息显示出来。
客户端输入 exit 时,退出程序,关闭了连接,服务端也显示这个客户端已经退出了连接。
HTTP
server.go 服务端
// http处理请求
func httpHandler(res http.ResponseWriter, req *http.Request) {
switch req.Method {
case "GET":
fmt.Println("这里是get请求")
res.Write([]byte("这里是get请求"))
case "POST":
defer req.Body.Close()
// 1、 请求类型是aplication/x-www-form-urlencode时解析form数据
//req.ParseForm()
//fmt.Println(req.PostForm) //打印form数数据
//fmt.Println(req.PostForm.Get("username"), req.PostForm.Get("password"))
// 2、请求类型是application/json时从req.Body读取数据
b, err := io.ReadAll(req.Body)
if err != nil {
fmt.Println("获取请求数据错误", err)
return
}
fmt.Println(string(b))
result := `{"code":0,"message":"ok","data":{}}` // 请求成功,返回数据
res.Write([]byte(result))
}
}
func main() {
HTTP服务端()
}
func HTTP服务端() {
http.HandleFunc("/index", httpHandler) // 回调函数
fmt.Println("http://localhost:8080")
http.ListenAndServe(":8080", nil) // 绑定服务
}
client.go 客户端
func main() {
HTTP客户端()
}
func HTTP客户端() {
data := `{"username":"符华","password":"123456"}`
//resp, err := http.Get("http://localhost:8080/index")
resp, err := http.Post("http://localhost:8080/index", "application/json", strings.NewReader(data))
//resp, err := http.PostForm("http://localhost:8080/index", url.Values{"username": {"符华"}, "password": {"123456"}})
if err != nil {
fmt.Println("请求错误", err)
return
}
defer resp.Body.Close()
b, err := io.ReadAll(resp.Body)
if err != nil {
fmt.Println("获取返回数据错误", err)
return
}
fmt.Println(string(b))
}
RPC
一个很简单的示例
server.go 服务端
import (
"net"
"net/http"
"net/rpc"
)
type Server struct{}
type Req struct {
Num1 int
Num2 int
}
type Res struct {
Num int
}
func (s Server) Add(req Req, res *Res) error {
res.Num = req.Num1 + req.Num2
return nil
}
func RPC服务端() {
// 注册rpc服务
rpc.Register(new(Server))
rpc.HandleHTTP()
listen, err := net.Listen("tcp", ":8080")
if err != nil {
fmt.Println(err)
return
}
http.Serve(listen, nil)
}
func main() {
RPC服务端()
}
client.go 客户端
import (
"fmt"
"net"
"net/rpc"
)
type Req struct {
Num1 int
Num2 int
}
type Res struct {
Num int
}
func RPC客户端() {
req := Req{1, 2}
client, err := rpc.DialHTTP("tcp", ":8080")
if err != nil {
fmt.Println(err)
return
}
var res Res
client.Call("Server.Add", req, &res)
fmt.Println(res)
}
func main() {
RPC客户端()
}
WebSocket
websocket是socket连接和http协议的结合体,可以实现网页和服务端的长连接。
要使用websocket我们要先去下载依赖
go get github.com/gorilla/websocket
server.go 服务端
// websocke 连接的升级器。升级器是一个http.HandlerFunc,它将HTTP连接升级为WebSocket连接
var UP = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
var conns []*websocket.Conn // 客户端连接切片,存储所有的客户端连接
// websocket处理请求
func websocketHandler(res http.ResponseWriter, req *http.Request) {
conn, err := UP.Upgrade(res, req, nil) // 服务升级
if err != nil {
fmt.Println(err)
return
}
defer conn.Close()
conns = append(conns, conn)
for {
// 消息类型,消息,错误
t, p, err := conn.ReadMessage()
if err != nil {
break
}
for index := range conns {
conns[index].WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("你说的是:%s吗?", string(p))))
}
fmt.Println(t, string(p))
}
fmt.Println("服务关闭")
}
func websocket服务端() {
http.HandleFunc("/", websocketHandler)
http.ListenAndServe(":8080", nil)
}
client.go 客户端
// websocket 发送消息
func send(conn *websocket.Conn) {
for {
reader := bufio.NewReader(os.Stdin)
l, _, _ := reader.ReadLine()
conn.WriteMessage(websocket.TextMessage, l)
}
}
func websocket客户端() {
dl := websocket.Dialer{}
conn, _, err := dl.Dial("ws://127.0.0.1:8080", nil)
if err != nil {
fmt.Println(err)
return
}
go send(conn)
for {
t, p, err := conn.ReadMessage()
if err != nil {
break
}
fmt.Println(t, string(p))
}
}
完整代码
server.go 服务端
package main
import (
"fmt"
"github.com/gorilla/websocket"
"io"
"net"
"net/http"
"net/rpc"
)
// tcp处理客户端发送过来的信息
func tcpHandler(conn net.Conn) {
defer conn.Close()
for {
var buf []byte = make([]byte, 1024)
// 等待客户端通过conn发送信息,如果客户端没有wrtie[发送],那么协程就阻塞在这里
n, err := conn.Read(buf)
if err == io.EOF {
fmt.Println(conn.RemoteAddr().String() + " 退出连接")
break
}
// 客户端发送的内容显示到服务器的终端
fmt.Print(string(buf[:n]))
}
}
// http处理请求
func httpHandler(res http.ResponseWriter, req *http.Request) {
switch req.Method {
case "GET":
fmt.Println("这里是get请求")
res.Write([]byte("这里是get请求"))
case "POST":
defer req.Body.Close()
// 1、 请求类型是aplication/x-www-form-urlencode时解析form数据
//req.ParseForm()
//fmt.Println(req.PostForm) //打印form数数据
//fmt.Println(req.PostForm.Get("username"), req.PostForm.Get("password"))
// 2、请求类型是application/json时从req.Body读取数据
b, err := io.ReadAll(req.Body)
if err != nil {
fmt.Println("获取请求数据错误", err)
return
}
fmt.Println(string(b))
result := `{"code":0,"message":"ok","data":{}}` // 请求成功,返回数据
res.Write([]byte(result))
}
}
// RPC
type Server struct{}
type Req struct {
Num1 int
Num2 int
}
type Res struct {
Num int
}
func (s Server) Add(req Req, res *Res) error {
res.Num = req.Num1 + req.Num2
return nil
}
// websocke 连接的升级器。升级器是一个http.HandlerFunc,它将HTTP连接升级为WebSocket连接
var UP = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
var conns []*websocket.Conn // 客户端连接切片,存储所有的客户端连接
// websocket处理请求
func websocketHandler(res http.ResponseWriter, req *http.Request) {
conn, err := UP.Upgrade(res, req, nil) // 服务升级
if err != nil {
fmt.Println(err)
return
}
defer conn.Close()
conns = append(conns, conn)
for {
// 消息类型,消息,错误
t, p, err := conn.ReadMessage()
if err != nil {
break
}
for index := range conns {
conns[index].WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("你说的是:%s吗?", string(p))))
}
fmt.Println(t, string(p))
}
fmt.Println("服务关闭")
}
func main() {
//TCP服务端()
//HTTP服务端()
//RPC服务端()
websocket服务端()
}
func TCP服务端() {
fmt.Println("服务器开始监听")
listen, err := net.Listen("tcp", "0.0.0.0:8080")
if err != nil {
fmt.Println(err)
return
}
defer listen.Close()
for { // 循环等待客户端来连接
conn, err := listen.Accept()
if err != nil {
fmt.Println(err)
} else {
fmt.Printf("客户端 %v 连接成功\n", conn.RemoteAddr().String())
}
// 这里开启协程
go tcpHandler(conn)
}
}
func HTTP服务端() {
http.HandleFunc("/index", httpHandler) // 回调函数
fmt.Println("http://localhost:8080")
http.ListenAndServe(":8080", nil) // 绑定服务
}
func RPC服务端() {
// 注册rpc服务
rpc.Register(new(Server))
rpc.HandleHTTP()
listen, err := net.Listen("tcp", ":8080")
if err != nil {
fmt.Println(err)
return
}
http.Serve(listen, nil)
}
func websocket服务端() {
http.HandleFunc("/", websocketHandler)
http.ListenAndServe(":8080", nil)
}
client.go 客户端
package main
import (
"bufio"
"fmt"
"github.com/gorilla/websocket"
"io"
"net"
"net/http"
"net/rpc"
"os"
"strings"
)
func main() {
//TCP客户端()
//HTTP客户端()
//RPC客户端()
websocket客户端()
}
func TCP客户端() {
conn, err := net.Dial("tcp", "127.0.0.1:8080")
if err != nil {
fmt.Println(err)
return
}
defer conn.Close()
fmt.Println(conn.RemoteAddr().String(), "连接成功")
var s string
for { // 循环输入,给服务端发送消息
// 从终端读取用户输入,并发送给服务器
fmt.Scanln(&s)
if s == "exit" {
fmt.Println(conn.RemoteAddr().String(), "退出连接")
break
}
conn.Write([]byte(s + "\r\n")) // 发送给服务端
}
}
func HTTP客户端() {
data := `{"username":"符华","password":"123456"}`
//resp, err := http.Get("http://localhost:8080/index")
resp, err := http.Post("http://localhost:8080/index", "application/json", strings.NewReader(data))
//resp, err := http.PostForm("http://localhost:8080/index", url.Values{"username": {"符华"}, "password": {"123456"}})
if err != nil {
fmt.Println("请求错误", err)
return
}
defer resp.Body.Close()
b, err := io.ReadAll(resp.Body)
if err != nil {
fmt.Println("获取返回数据错误", err)
return
}
fmt.Println(string(b))
}
// RPC
type Req struct {
Num1 int
Num2 int
}
type Res struct {
Num int
}
func RPC客户端() {
req := Req{1, 2}
client, err := rpc.DialHTTP("tcp", ":8080")
if err != nil {
fmt.Println(err)
return
}
var res Res
client.Call("Server.Add", req, &res)
fmt.Println(res)
}
// websocket 发送消息
func send(conn *websocket.Conn) {
for {
reader := bufio.NewReader(os.Stdin)
l, _, _ := reader.ReadLine()
conn.WriteMessage(websocket.TextMessage, l)
}
}
func websocket客户端() {
dl := websocket.Dialer{}
conn, _, err := dl.Dial("ws://127.0.0.1:8080", nil)
if err != nil {
fmt.Println(err)
return
}
go send(conn)
for {
t, p, err := conn.ReadMessage()
if err != nil {
break
}
fmt.Println(t, string(p))
}
}
ok,以上就是本篇笔记的全部内容了,就是一些简单的使用示例,关于理论概念的东西我没有记,因为太多了,不知道要怎么整理才能说明白说清楚,反正就是这么用的。