【Java转Go】快速上手学习笔记(六)之网络编程篇一

目录

  • TCP
    • 一个简单案例
      • server.go 服务端
      • client.go 客户端
  • HTTP
      • server.go 服务端
      • client.go 客户端
  • RPC
    • 一个很简单的示例
      • server.go 服务端
      • client.go 客户端
  • WebSocket
      • server.go 服务端
      • client.go 客户端
  • 完整代码
    • server.go 服务端
    • client.go 客户端

go往期文章笔记:

【Java转Go】快速上手学习笔记(一)之环境安装篇

【Java转Go】快速上手学习笔记(二)之基础篇一

【Java转Go】快速上手学习笔记(三)之基础篇二

【Java转Go】快速上手学习笔记(四)之基础篇三

【Java转Go】快速上手学习笔记(五)之Gorm篇


这篇记的是网络编程相关的笔记。

TCP

开启服务端,监听连接的请求。示意图:
【Java转Go】快速上手学习笔记(六)之网络编程篇一_第1张图片

服务端:启动后,监听连接过来的客户端,每连接过来一个客户端,开启一个协程去和客户端进行交互(一个协程代表一个客户端)

客户端:根据服务端的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")) // 发送给服务端
	}
}

写好之后,我们先启动服务端,然后再启动客户端。在客户端里面输入消息发送给服务端,服务端接收后要将消息显示出来。

【Java转Go】快速上手学习笔记(六)之网络编程篇一_第2张图片
【Java转Go】快速上手学习笔记(六)之网络编程篇一_第3张图片

客户端输入 exit 时,退出程序,关闭了连接,服务端也显示这个客户端已经退出了连接。
【Java转Go】快速上手学习笔记(六)之网络编程篇一_第4张图片
【Java转Go】快速上手学习笔记(六)之网络编程篇一_第5张图片

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,以上就是本篇笔记的全部内容了,就是一些简单的使用示例,关于理论概念的东西我没有记,因为太多了,不知道要怎么整理才能说明白说清楚,反正就是这么用的。

你可能感兴趣的:(Go,golang,go)