[UMU 学 golang](3) TCP Echo Server

UMU · · 2072 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

  测试需要,以前用 C + libevent 写了一个 TCP Echo Server,返回服务器时间、客户端地址信息和客户端发送的原内容。为了水一篇,现在改为 golang 实现。

package main

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

const BUFFER_SIZE = 1024 * 4

var buffer = make([]byte, BUFFER_SIZE)

func handleConnect(tcpConn *net.TCPConn) {
	if tcpConn == nil {
		return
	}
	for {
		n, err := tcpConn.Read(buffer)
		if err == io.EOF {
			fmt.Printf("The RemoteAddr: %s is closed!\n", tcpConn.RemoteAddr().String())
			return
		}
		handleError(err)
		if n > 0 {
			//fmt.Printf("Read: %s", string(buffer[:n]))
			str := tcpConn.RemoteAddr().String() + " @ " +
				time.Now().Format("2006-01-02 15:04:05 Z07:00") + "\r\n" +
				string(buffer[:n])
			tcpConn.Write([]byte(str))
			fmt.Printf("Echo: %s", str)
		}
	}
}

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

func main() {
	if len(os.Args) < 2 {
		fmt.Println("Usage:", os.Args[0], "<port>")
		return
	}
	port := os.Args[1]
	tcpAddr, err := net.ResolveTCPAddr("tcp4", "0.0.0.0:"+port)
	handleError(err)
	tcpListener, err := net.ListenTCP("tcp4", tcpAddr)
	handleError(err)
	defer tcpListener.Close()

	fmt.Println("Listening on", tcpAddr, "...")
	for {
		tcpConn, err := tcpListener.AcceptTCP()
		fmt.Printf("The client: %s has connected!\n", tcpConn.RemoteAddr().String())
		handleError(err)
		defer tcpConn.Close()
		go handleConnect(tcpConn)
	}
}



有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:UMU

查看原文:[UMU 学 golang](3) TCP Echo Server

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

2072 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传