[GoLang]GO语言编写的简易TCP服务端、客户端

客户端程序:

package main

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

//缓冲区大小
const BUFFSIZE = 1024

//定义消息缓冲区
var buff = make([]byte, BUFFSIZE)

//链接处理
func handleConn(tcpConn *net.TCPConn, addr string) {
	if tcpConn == nil {
		//fmt.Println()
		return
	}
	for{
		n, err := tcpConn.Read(buff)
		if err == io.EOF {
			fmt.Printf("远程链接:%s已经关闭!\n", tcpConn.RemoteAddr().String())
			return
		}
		handleError(err)
		if string(buff[:n]) == "exit" {
			fmt.Printf("远程链接:%s退出!\n", tcpConn.RemoteAddr().String())
		}
		if n > 0 {
			fmt.Printf("%s发来消息:%s", addr, string(buff[:n]))
		}
	}
}

func handleError(err error) {
	if err == nil {
		return
	}
	fmt.Printf("错误:%s\n", err.Error())
}

func main() {

	//判断命令行参数是否错误
	if len(os.Args) != 2 {
		fmt.Printf("调用参数错误,程序中断!\n")
		return
	}

	//获取通讯端口
	port := os.Args[1]

	//原型:func ResolveTCPAddr(net, addr string) (*TCPAddr, error) ResolveTCPAddr parses addr as a TCP address
	//分解ip为TCP地址
	tcpAddr, err := net.ResolveTCPAddr("tcp4", "localhost:" + port)
	handleError(err)
	if err != nil {
		return
	}

	//原型:func ListenTCP(net string, laddr *TCPAddr) (*TCPListener, error) ListenTCP announces on the TCP address laddr and returns a TCP listener.
	//通过TCP地址返回一个TCP监听
	tcpListener, err := net.ListenTCP("tcp4", tcpAddr)
	handleError(err)
	if err != nil {
		return
	}

	fmt.Printf("启动监听,等待链接!\n")

	//延迟,监听关闭
	defer tcpListener.Close()

	for {
		//func (l *TCPListener) AcceptTCP() (*TCPConn, error) AcceptTCP accepts the next incoming call and returns the new connection.
		//等待下一个访问并返回一个新链接
		tcpConn, err := tcpListener.AcceptTCP()
		handleError(err)
		if err != nil {
			return
		}

		fmt.Printf("客户端:%s已连接!\n", tcpConn.RemoteAddr().String())
		//延迟,关闭链接
		defer tcpConn.Close()
		//启动一个连接处理线程
		go handleConn(tcpConn, tcpConn.RemoteAddr().String())
	}
}

服务端程序:

package main

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

const BUFFSIZE = 1024
var buff = make([]byte, BUFFSIZE)

func handleError(err error) {
	if err == nil {
		return
	}
	fmt.Printf("错误:%s\n", err.Error())
}

func main() {
	if len(os.Args) != 2 {
		fmt.Printf("参数错误, 程序中断!\n")
		return
	}
	port := os.Args[1]
	tcpAddr, err := net.ResolveTCPAddr("tcp4", "localhost:" + port)
	handleError(err)
	tcpConn, err := net.DialTCP("tcp4", nil, tcpAddr)
	handleError(err)

	fmt.Printf("连接成功,请输入信息!\n")

	reader := bufio.NewReader(os.Stdin)
	continued := true
	inputStr := ""

	for(continued){
		n, err := reader.Read(buff)
		handleError(err)
		if n > 0 {
			k, _ := tcpConn.Write(buff[:n])
			if k > 0 {
				inputStr = string(buff[:n])
				fmt.Printf("发送消息:%s", inputStr)
				if inputStr == "exit\n" {
					continued = false
				}
			}
		}
	}
}

服务端:


客户端:


本文来自:CSDN博客

感谢作者:u011229962

查看原文:[GoLang]GO语言编写的简易TCP服务端、客户端

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。