Go TCP编程

系统管理员 2023-02-14 07:44 12阅读 0赞

**Go TCP编程**  
**一、服务端编程step：**  
1.1 创建套接字  
1.2 监听套接字  
1.3 处理请求

`server/tcp_server.go`

package server
    import (
    	"fmt"
    	"net"
    	"github.com/jstang007/gateway_demo/httpserver/handler"
    )
    // SrvForever 服务端启动函数
    func SrvForever() { 
    	//1、创建套接字
    	listenner, err := net.Listen("tcp", "0.0.0.0:9091")
    	if err != nil { 
    		fmt.Printf("listen fail, err:%v\n", err)
    		return
    	}
    	//2、监听套接字
    	for { 
    		conn, err := listenner.Accept() //接收请求, 阻塞态
    		if err != nil { 
    			fmt.Printf("accept fail, err:%v\n", err)
    			continue
    		}
    		//3、处理请求
    		go handler.Process(conn)
    	}
    }

**二、客户端编程step**  
2.1 创建套接字  
2.2 发送消息

package client
    import (
    	"fmt"
    	"net"
    	"github.com/jstang007/gateway_demo/httpserver/handler"
    )
    //TCPClient 客户端
    func TCPClient() { 
    	conn, err := net.Dial("tcp", "localhost:9091") //创建套接字
    	if err != nil { 
    		fmt.Printf("connect failed, error:%v", err)
    	}
    	defer conn.Close()
    	handler.Send(conn, "my name is TCP Client!")
    }

**三、消息处理函数**  
3.1 消息解码  
3.2 消息编码  
3.3 发送消息  
3.4 接收处理消息  
`handler/base_handler.go`

package handler
    import (
    	"encoding/binary"
    	"fmt"
    	"io"
    	"net"
    )
    //MsgHeader 消息头
    var MsgHeader = "12345678"
    
    // decode 解码，将字节流解码成utf-8
    func decode(bytesBuffer io.Reader) (bodyBuf []byte, err error) { 
    	//1.首先读取消息头数据存到MagicBuf
    	MagicBuf := make([]byte, len(MsgHeader))
    	if _, err = io.ReadFull(bytesBuffer, MagicBuf); err != nil { 
    		return nil, err
    	}
    
    	//2、读取消息内容长度到lengthBuf
    	lengthBuf := make([]byte, 4)
    	if _, err := io.ReadFull(bytesBuffer, lengthBuf); err != nil { 
    		return nil, err
    	}
    
    	//3.读取消息内容到bodyBuf
    	length := binary.BigEndian.Uint32(lengthBuf)
    	bodyBuf = make([]byte, length)
    	if _, err := io.ReadFull(bytesBuffer, bodyBuf); err != nil { 
    		return nil, err
    	}
    	return bodyBuf, err
    }
    
    // Process 处理请求
    func Process(conn net.Conn) { 
    	defer conn.Close()
    	for { 
    		bt, err := decode(conn)
    		if err != nil { 
    			fmt.Printf("read from connect failed, err:%v\n", err)
    			break
    		}
    		str := string(bt)
    		fmt.Printf("receive from client, data: %s\n", str)
    	}
    }
    
    // Send 发送消息
    func Send(conn net.Conn, content string) error { 
    	return encode(conn, content)
    }
    
    // encode 将string字符串转换成字节流
    func encode(bytesBuffer io.Writer, contetnt string) error { 
    	//1.写入消息头
    	bytesMsgHeader := []byte(MsgHeader)
    	if err := binary.Write(bytesBuffer, binary.BigEndian, bytesMsgHeader); err != nil { 
    		return err
    	}
    	//2.写入内容长度
    	clen := int32(len([]byte(contetnt)))
    	if err := binary.Write(bytesBuffer, binary.BigEndian, clen); err != nil { 
    		return err
    	}
    	//3.写入内容
    	bytesContent := []byte(contetnt)
    	if err := binary.Write(bytesBuffer, binary.BigEndian, bytesContent); err != nil { 
    		return err
    	}
    	return nil
    }

**四、测试函数**  
test/main.go

package main
    import (
    	"fmt"
    	"time"
    	"github.com/jstang007/gateway_demo/httpserver/client"
    	"github.com/jstang007/gateway_demo/httpserver/server"
    )
    func main() { 
    	//启动服务器
    	go server.SrvForever()
    	//启动客户端
    	for i := 0; i < 5; i++ { 
    		time.Sleep(time.Second * 2)
    		fmt.Printf("-------启动客户端[%v]--------\n", i)
    		client.TCPClient()
    	}
    }

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4OTAwNTY1_size_16_color_FFFFFF_t_70_pic_center]  
EOF表示读完数据了

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4OTAwNTY1_size_16_color_FFFFFF_t_70_pic_center]: https://img-blog.csdnimg.cn/2020060123011950.PNG?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4OTAwNTY1,size_16,color_FFFFFF,t_70#pic_center