[Golang] 从零开始写Socket Server(1): Socket-Client框架

时间:2023-03-09 20:59:33
[Golang] 从零开始写Socket Server(1): Socket-Client框架

版权声明:本文为博主原创文章,未经博主允许不得转载。

第一次跑到互联网公司实习 。。感觉自己进步飞快啊~第一周刚写了个HTTP服务器用于微信公共号的点餐系统~ 第二周就直接开始一边自学Go语言一边写用于Socket的服务器了。。。

因为发现Golang这一块资料挺少的,接下来我会在Blog里把整个Server的Coding,还有遇到的坑都记录下来~

在golang中,网络协议已经被封装的非常完好了,想要写一个Socket的Server,我们并不用像其他语言那样需要为socket、bind、listen、receive等一系列操作头疼,只要使用Golang中自带的net包即可很方便的完成连接等操作~

在这里,给出一个最最基础的基于Socket的Server的写法:

  1. package main
  2. import (
  3. "fmt"
  4. "net"
  5. "log"
  6. "os"
  7. )
  8. func main() {
  9. //建立socket,监听端口
  10. netListen, err := net.Listen("tcp", "localhost:1024")
  11. CheckError(err)
  12. defer netListen.Close()
  13. Log("Waiting for clients")
  14. for {
  15. conn, err := netListen.Accept()
  16. if err != nil {
  17. continue
  18. }
  19. Log(conn.RemoteAddr().String(), " tcp connect success")
  20. handleConnection(conn)
  21. }
  22. }
  23. //处理连接
  24. func handleConnection(conn net.Conn) {
  25. buffer := make([]byte, 2048)
  26. for {
  27. n, err := conn.Read(buffer)
  28. if err != nil {
  29. Log(conn.RemoteAddr().String(), " connection error: ", err)
  30. return
  31. }
  32. Log(conn.RemoteAddr().String(), "receive data string:\n", string(buffer[:n]))
  33. }
  34. }
  35. func Log(v ...interface{}) {
  36. log.Println(v...)
  37. }
  38. func CheckError(err error) {
  39. if err != nil {
  40. fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
  41. os.Exit(1)
  42. }
  43. }

唔,抛除go语言里面10行代码有5行error的蛋疼之处,你可以看到,Server想要建立并接受一个Socket,其核心流程就是

  1. netListen, err := net.Listen("tcp", "localhost:1024")
  1. conn, err := netListen.Accept()
  1. n, err := conn.Read(buffer)

这三步,通过Listen、Accept 和Read,我们就成功的绑定了一个端口,并能够读取从该端口传来的内容~

Server写好之后,接下来就是Client方面啦,我手写一个HelloWorld给大家:

  1. package main
  2. import (
  3. "fmt"
  4. "net"
  5. "os"
  6. )
  7. func sender(conn net.Conn) {
  8. words := "hello world!"
  9. conn.Write([]byte(words))
  10. fmt.Println("send over")
  11. }
  12. func main() {
  13. server := "127.0.0.1:1024"
  14. tcpAddr, err := net.ResolveTCPAddr("tcp4", server)
  15. if err != nil {
  16. fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
  17. os.Exit(1)
  18. }
  19. conn, err := net.DialTCP("tcp", nil, tcpAddr)
  20. if err != nil {
  21. fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
  22. os.Exit(1)
  23. }
  24. fmt.Println("connect success")
  25. sender(conn)
  26. }

可以看到,Client这里的关键在于

  1. tcpAddr, err := net.ResolveTCPAddr("tcp4", server)
  1. conn, err := net.DialTCP("tcp", nil, tcpAddr)

这两步,主要是负责解析端口和连接~

写好Server和Client之后,让我们运行一下看看:~~

成功运行,Console出现Server等待连接的提示:

[Golang] 从零开始写Socket Server(1): Socket-Client框架

然后我们运行Client:

[Golang] 从零开始写Socket Server(1): Socket-Client框架

成功连接Server啦,让我们看看Server那边的成果:

[Golang] 从零开始写Socket Server(1): Socket-Client框架

Server端成功的收到了我们的Hello-World啦,至于后面的那行红字,则是断开连接的提示~

到这里,一个最基础的使用Socket的Server-Client框架就出来啦~

如果想要让Server能够响应来自不同Client的请求,我们只要在Server端的代码的main入口中,

在 handleConnection(conn net.Conn) 这句代码的前面加上一个 go,就可以让服务器并发处理不同的Client发来的请求啦

关于Golang的并发处理,可以参考这里 Golang并行计算

下一篇,我会写一下如何在Server和Client中设计一个通讯协议,从而达到自己的一些特殊要求~