Go实现简单的多并发服务器

 互联网通信本质上是应用进程之间的通信、连接,必须知道通信双方的IP以及端口Port;

image

本文选用流式套接字的模式,利用Go语言实现简单的多并发任务,原理的流程图大致如下:

服务器的建立

建立socket,申明自身的端口port和IP,并绑定到socket,使用listen监听,然后不断用accept去查看是否有连接。如果有,捕获socket,并通过recv获取消息的内容,通信完成后调用closeSocket关闭这个对应accept到的socket。如果不需要等待任何客户端连接,那么用closeSocket直接关闭自身的socket。
package main

import (
	"fmt"
	"net"
	"strings"
)


func HandleConn(conn net.Conn){
	//调用完毕,自动关闭conn
	defer conn.Close()

	//获取客户端的网络地址信息
	addr:=conn.RemoteAddr().String()
	fmt.Printf("[%s] connect successful\n",addr)

	buf:=make([]byte,2048)  //切片存储数据

    for{
		//读取用户数据
		n,err:=conn.Read(buf)   //返回长度
		if err!=nil{
			fmt.Println("err=",err)
			return
		}
		fmt.Printf("[%s]: %s\n",addr,string(buf[:n]))

		//fmt.Println("len=",len(string(buf[:n])))
		if "exit"==string(buf[:n-1]){ //客户端退出输入exit,键盘输入多了1个字符,‘\n’
			fmt.Println(addr,"exit")
			return
		}

		//把数据转换为大写,再给用户发送
		conn.Write([]byte(strings.ToUpper(string(buf[:n]))))  //[]byte隐式赋值
	}

}
func main(){
	//监听,可以申明自身端口和IP,或者任意IP地址和端口
	listener,err:=net.Listen("tcp","127.0.0.1:8000")
	if err!=nil{
		fmt.Println("err=",err)
		return    //不执行下面的
	}
	defer listener.Close()  //主线程退出前关闭

	//接受多个用户
	for{
		conn,err1:=listener.Accept()
		if err1!=nil{
			fmt.Println("err1=",err1)
			return
		}
		//处理用户请求,多用户,创建新的协程
		go HandleConn(conn)
	}
}

客户端的建立 

建立socket,通过端口号和地址确定目标服务器,使用Connect连接到服务器,send发送消息,等待处理,通信完成后调用closeSocket关闭socket。 
package main

import "fmt"
import (
	"net"
	"os"
)

func main(){
	//主动连接服务器
	conn,err:=net.Dial("tcp","127.0.0.1:8000")

	if err!=nil{
		fmt.Println("net.Dial err=",err)
		return
	}

	//main调用完毕后,关闭连接
	defer conn.Close()

	//创建协程,用户端输入数据
	go func(){
		//客户端输入内容,发给服务器
		//切片缓冲
		str:=make([]byte,1024)
		for{
			n1,err2:=os.Stdin.Read(str)  //从键盘读取内容,存在str
			if err2!=nil{
				fmt.Println("os.Stdin.Read err2=",err2)
				return
			}
			//内容传给服务器
			conn.Write(str[:n1])
		}

	}()

	//接受服务器回复的数据
	//切片缓冲
	buf:=make([]byte,1024)
	for{
		n,err1:=conn.Read(buf) //接受服务器的请求
		if err1!=nil{
			fmt.Println("conn.Read err1=",err1)
			return
		}
		fmt.Println(string(buf[:n]))  //打印收到的内容,转换为字符串
	}
}

 

 

 

 


版权声明:本文为Jiangtagong原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。