1、前言
其实我前面一篇笔记的例子就是socket的一个例子,但是由于大部分的笔记说明都是在整理基础的东西,所以socket的笔记单独列在这里。
server.go
package socket
import (
"fmt"
"net"
)
func StartServer() {
service := ":3338"
tcpAddr, err := net.ResolveTCPAddr("tcp4", service)
checkError(err)
listener, err := net.ListenTCP("tcp", tcpAddr)
checkError(err)
for {
conn, err := listener.Accept()
if err != nil {
continue
}
fmt.Println("新连接:", conn.RemoteAddr().String())
go handleConn(conn)
}
}
func handleConn(conn net.Conn) {
defer conn.Close()
for {
buffer := make([]byte, 1024)
length, _:= conn.Read(buffer)
if length > 12 {
data := buffer[:length]
switch data[11] & 0xff {
case 0x80:
//桌子
fmt.Println("桌子")
case 0x90:
//椅子
case 0xA0:
//台灯
default:
//其它
}
//写数据
// conn.Write(data)
}
}
}
func isProtocol(data []byte) bool {
if (data[0]&0xff) == 0xC0 && (data[len(data)-1]&0xff) == 0xC1 {
return true
}
return false
}
func checkError(err error) {
if err != nil {
fmt.Println(err.Error())
}
}
2、TCP Socket Server
2.1 导包
和java一样,golang提供了比较强大的网络编程的基础库net。
当然我们想使用golang的网络相关的库,先导入net包。而fmt包,则是为了测试一些结果,作为终端输出的工具包。
import (
"fmt"
"net"
)
2.2 声明地址
通过net包的ResolveTCPAddr函数声明一个TCPAddr。
// TCPAddr represents the address of a TCP end point.
type TCPAddr struct {
IP IP
Port int
Zone string // IPv6 scoped addressing zone
}
func ResolveTCPAddr(net, addr string) (*TCPAddr, error)
- net参数是”tcp4”、”tcp6”、”tcp”中的任意一个,分别表示TCPv4、TCPv6或者任意。查看ResolveTCPAddr的源码后,net参数甚至可以是一个空字符串,估计是为了兼容之前的Go版本的原因。
switch net {
case "tcp", "tcp4", "tcp6":
case "": // a hint wildcard for Go 1.0 undocumented behavior
net = "tcp"
default:
return nil, UnknownNetworkError(net)
}
- addr表示域名或者IP地址,例如”www.baidu.com:80” 或者”127.0.0.1:22”。
从源码上看,ip是可以缺省的,但端口不能缺省。ip缺省的话,默认使用的是本地地址。ip和端口使用英文的冒号(:)分开,net包是通过冒号分隔字符串的方式来处理ResolveTCPAddr的net参数的。
ResolveTCPAddr可供服务器及客户端共用的,不像java那样区分Socket和ServerSocket。所以缺省ip的逻辑很好理解。
我们这里写的是服务器端的逻辑,只输入端口号。
// SplitHostPort splits a network address of the form "host:port",
// "[host]:port" or "[ipv6-host%zone]:port" into host or
// ipv6-host%zone and port. A literal address or host name for IPv6
// must be enclosed in square brackets, as in "[::1]:80",
// "[ipv6-host]:http" or "[ipv6-host%zone]:80".
func SplitHostPort(hostport string) (host, port string, err error) {
j, k := 0, 0
// The port starts after the last colon.
i := last(hostport, ':')
if i < 0 {
goto missingPort
}
if hostport[0] == '[' {
// Expect the first ']' just before the last ':'.
end := byteIndex(hostport, ']')
if end < 0 {
err = &AddrError{Err: "missing ']' in address", Addr: hostport}
return
}
switch end + 1 {
case len(hostport):
// There can't be a ':' behind the ']' now.
goto missingPort
case i:
// The expected result.
default:
// Either ']' isn't followed by a colon, or it is
// followed by a colon that is not the last one.
if hostport[end+1] == ':' {
goto tooManyColons
}
goto missingPort
}
host = hostport[1:end]
j, k = 1, end+1 // there can't be a '[' resp. ']' before these positions
} else {
host = hostport[:i]
if byteIndex(host, ':') >= 0 {
goto tooManyColons
}
if byteIndex(host, '%') >= 0 {
goto missingBrackets
}
}
if byteIndex(hostport[j:], '[') >= 0 {
err = &AddrError{Err: "unexpected '[' in address", Addr: hostport}
return
}
if byteIndex(hostport[k:], ']') >= 0 {
err = &AddrError{Err: "unexpected ']' in address", Addr: hostport}
return
}
port = hostport[i+1:]
return
missingPort:
err = &AddrError{Err: "missing port in address", Addr: hostport}
return
tooManyColons:
err = &AddrError{Err: "too many colons in address", Addr: hostport}
return
missingBrackets:
err = &AddrError{Err: "missing brackets in address", Addr: hostport}
return
}
2.3 启动监听器
- 通过net包的ListenTCP之前的TCPAddr启动一个监听器。
func ListenTCP(net string, laddr *TCPAddr) (*TCPListener, error)
这里的net参数也是支付”tcp”, “tcp4”, “tcp6”。但不支持空字符串。
- 通过TCPListener的Accept方法阻塞式接收客户端的连接
func (l *TCPListener) Accept() (Conn, error)
整个过程和java socket server的思想基本是一至的。
2.4 Conn
type Conn interface {
// Read reads data from the connection.
// Read can be made to time out and return a Error with Timeout() == true
// after a fixed time limit; see SetDeadline and SetReadDeadline.
Read(b []byte) (n int, err error)
// Write writes data to the connection.
// Write can be made to time out and return a Error with Timeout() == true
// after a fixed time limit; see SetDeadline and SetWriteDeadline.
Write(b []byte) (n int, err error)
// Close closes the connection.
// Any blocked Read or Write operations will be unblocked and return errors.
Close() error
// LocalAddr returns the local network address.
LocalAddr() Addr
// RemoteAddr returns the remote network address.
RemoteAddr() Addr
// SetDeadline sets the read and write deadlines associated
// with the connection. It is equivalent to calling both
// SetReadDeadline and SetWriteDeadline.
//
// A deadline is an absolute time after which I/O operations
// fail with a timeout (see type Error) instead of
// blocking. The deadline applies to all future I/O, not just
// the immediately following call to Read or Write.
//
// An idle timeout can be implemented by repeatedly extending
// the deadline after successful Read or Write calls.
//
// A zero value for t means I/O operations will not time out.
SetDeadline(t time.Time) error
// SetReadDeadline sets the deadline for future Read calls.
// A zero value for t means Read will not time out.
SetReadDeadline(t time.Time) error
// SetWriteDeadline sets the deadline for future Write calls.
// Even if write times out, it may return n > 0, indicating that
// some of the data was successfully written.
// A zero value for t means Write will not time out.
SetWriteDeadline(t time.Time) error
}
- Conn接口的Read和Write方法进行IO操作。
- 通过SetDeadline、SetReadDeadline、SetWriteDeadline方法设置IO超时时间
- RemoteAddr方法可以获取到客户端的ip及端口信息。
fmt.Println("新连接:", conn.RemoteAddr().String())
注意:Conn接口的Read也是阻塞的。
2.5 defer
func handleConn(conn net.Conn) {
defer conn.Close()
for {
...
}
}
关键字 defer 的用法类似于面向对象编程语言 Java 和 C# 的 finally 语句块,它一般用于释放某些已分配的资源,比如常见的关闭文件、释放缓存。
这里是操作完写操作后,就把conn连接关掉。
3、Goroutine
在Go中,每个并发处理的部分被称作 goroutines(协程),它可以进行更有效的并发运算。在协程和操作系统线程之间并无一对一的关系:协程是根据一个或多个线程的可用性,映射(多路复用,执行于)在他们之上的;协程调度器在 Go 运行时很好的完成了这个工作。
golang想使用协程的方式很简单,只要在需要使用协程的方法调用的前面加上go关键字就可以了。
如:
go handleConn(conn)
这种方式相比java的线程,太优雅了。
有疑问加站长微信联系(非本文作者)