Go语言简单的TCP编程

前期准备

需要import "net"包
IP类型，其中一个重要的方法是IP.ParseIP(ipaddr string)来判断是否是合法的IP地址

TCP Client

func (c *TCPConn) Write(b []byte) (n int, err os.Error)用于发送数据，返回发送的数据长度或者返回错误，是TCPConn的方法
func (c *TCPConn) Read(b []byte) (n int, err os.Error)用于接收数据，返回接收的长度或者返回错误，是TCPConn的方法
TCPAddr类型，保存TCP的地址信息，包括地址和端口 type TCPAddr struct { IP IP Port int }
func ResolveTCPAddr(net, addr string) (*TCPAddr, os.Error)获取一个TCPAddr，参数都是string类型，net是个const string,包括tcp4,tcp6,tcp一般使用tcp，兼容v4和v6，addr表示ip地址，包括端口号，如www.google.com:80之类的
func DialTCP(net string, laddr, raddr *TCPAddr) (c *TCPConn, err os.Error)用来连接(connect)到远程服务器上，net表示协议方式，tcp,tcp4或者tcp6，laddr表示本机地址，一般为nil，raddr表示远程地址，这里的laddr和raddr都是TCPAddr类型的，一般是上一个函数的返回值。
作为一个TCP的客户端，基本的操作流程如下： service="www.google.com:80" tcpAddr, err := net.ResolveTCPAddr("tcp4", service) conn, err := net.DialTCP("tcp", nil, tcpAddr) _, err = conn.Write([]byte("HEAD / HTTP/1.0rnrn")) _, err = conn.Read(b) / result, err := ioutil.ReadAll(conn)

TCP Server

func ListenTCP(net string, laddr *TCPAddr) (l *TCPListener, err os.Error)用来监听端口，net表示协议类型，laddr表示本机地址,是TCPAddr类型，注意，此处的laddr包括端口，返回一个*TCPListener类型或者错误
func (l *TCPListener) Accept() (c Conn, err os.Error)用来返回一个新的连接，进行后续操作，这是TCPListener的方法，一般TCPListener从上一个函数返回得来。
服务器的基本操作流程为： service:=":9090" tcpAddr, err := net.ResolveTCPAddr("tcp4", service) l,err := net.ListenTCP("tcp",tcpAddr) conn,err := l.Accept() go Handler(conn) //此处使用go关键字新建线程处理连接，实现并发

一个例子

需求

实现一个公共聊天服务器。

服务器接收客户端的信息
接收完以后将客户端的信息发送到所有的客户端上
客户端使用/quit退出聊天
只使用一套代码，通过命令行参数启动服务器还是客户端

实现：

package main  
 
import(  
 "fmt" 
 "os" 
 "net" 
)  
 
//////////////////////////////////////////////////////// 
// 
//错误检查 
// 
//////////////////////////////////////////////////////// 
func checkError(err error,info string) (res bool) {  
 
 if(err != nil){  
        fmt.Println(info+"  " + err.Error())  
 return false 
    }  
 return true 
}  
 
//////////////////////////////////////////////////////// 
// 
//服务器端接收数据线程 
//参数： 
//      数据连接 conn 
//      通讯通道 messages 
// 
//////////////////////////////////////////////////////// 
func Handler(conn net.Conn,messages chan string){  
 
    fmt.Println("connection is connected from ...",conn.RemoteAddr().String())  
 
    buf := make([]byte,1024)  
 for{  
        lenght, err := conn.Read(buf)  
 if(checkError(err,"Connection")==false){  
            conn.Close()  
 break 
        }  
 if lenght > 0{  
            buf[lenght]=0  
        }  
 //fmt.Println("Rec[",conn.RemoteAddr().String(),"] Say :" ,string(buf[0:lenght])) 
        reciveStr :=string(buf[0:lenght])  
        messages <- reciveStr  
 
    }  
 
}  
 
//////////////////////////////////////////////////////// 
// 
//服务器发送数据的线程 
// 
//参数 
//      连接字典 conns 
//      数据通道 messages 
// 
//////////////////////////////////////////////////////// 
func echoHandler(conns *map[string]net.Conn,messages chan string){  
 
 
 for{  
        msg:= <- messages  
        fmt.Println(msg)  
 
 for key,value := range *conns {  
 
            fmt.Println("connection is connected from ...",key)  
            _,err :=value.Write([]byte(msg))  
 if(err != nil){  
                fmt.Println(err.Error())  
 delete(*conns,key)  
            }  
 
        }  
    }  
 
}  
 
 
//////////////////////////////////////////////////////// 
// 
//启动服务器 
//参数 
//  端口 port 
// 
//////////////////////////////////////////////////////// 
func StartServer(port string){  
        service:=":"+port //strconv.Itoa(port); 
        tcpAddr, err := net.ResolveTCPAddr("tcp4", service)  
        checkError(err,"ResolveTCPAddr")  
        l,err := net.ListenTCP("tcp",tcpAddr)  
        checkError(err,"ListenTCP")  
        conns:=make(map[string]net.Conn)  
        messages := make(chan string,10)  
 //启动服务器广播线程 
        go echoHandler(&conns,messages)  
 
 for  {  
            fmt.Println("Listening ...")  
            conn,err := l.Accept()  
            checkError(err,"Accept")  
            fmt.Println("Accepting ...")  
            conns[conn.RemoteAddr().String()]=conn  
 //启动一个新线程 
            go Handler(conn,messages)   
 
        }  
 
}  
 
 
//////////////////////////////////////////////////////// 
// 
//客户端发送线程 
//参数 
//      发送连接 conn 
// 
//////////////////////////////////////////////////////// 
func chatSend(conn net.Conn){  
 
    var input string  
    username := conn.LocalAddr().String()  
 for {  
 
        fmt.Scanln(&input)  
 if input == "/quit"{  
            fmt.Println("ByeBye..")  
            conn.Close()  
            os.Exit(0);  
        }  
 
 
        lens,err :=conn.Write([]byte(username + " Say :::" + input))  
        fmt.Println(lens)  
 if(err != nil){  
            fmt.Println(err.Error())  
            conn.Close()  
 break 
        }  
 
    }  
 
}  
 
//////////////////////////////////////////////////////// 
// 
//客户端启动函数 
//参数 
//      远程ip地址和端口 tcpaddr 
// 
//////////////////////////////////////////////////////// 
func StartClient(tcpaddr string){  
 
    tcpAddr, err := net.ResolveTCPAddr("tcp4", tcpaddr)  
    checkError(err,"ResolveTCPAddr")  
    conn, err := net.DialTCP("tcp", nil, tcpAddr)  
    checkError(err,"DialTCP")  
 //启动客户端发送线程 
    go chatSend(conn)     
 
 //开始客户端轮训 
    buf := make([]byte,1024)  
 for{  
 
        lenght, err := conn.Read(buf)  
 if(checkError(err,"Connection")==false){  
            conn.Close()  
            fmt.Println("Server is dead ...ByeBye")  
            os.Exit(0)  
        }  
        fmt.Println(string(buf[0:lenght]))  
 
    }  
}  
 
 
//////////////////////////////////////////////////////// 
// 
//主程序 
// 
//参数说明： 
//  启动服务器端：  Chat server [port]             eg: Chat server 9090 
//  启动客户端：    Chat client [Server Ip Addr]:[Server Port]    eg: Chat client 192.168.0.74:9090 
// 
//////////////////////////////////////////////////////// 
func main(){  
 
 
 if len(os.Args)!=3  {     
        fmt.Println("Wrong pare")  
        os.Exit(0)  
    }  
 
 if os.Args[1]=="server" && len(os.Args)==3 {  
 
        StartServer(os.Args[2])  
    }  
 
 
 if os.Args[1]=="client" && len(os.Args)==3 {  
 
        StartClient(os.Args[2])  
    }  
 
 
}