gatewayServer.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. package gatewayServer
  2. import (
  3. "fmt"
  4. "io"
  5. "lcfns/global"
  6. "log"
  7. "net"
  8. "sync/atomic"
  9. )
  10. var ipAddress = global.Config.Foreign.GatewayServer
  11. var ConnMap = make(map[string]net.Conn, 50)
  12. var openDebug bool = false
  13. func GatewayServe() {
  14. listen, err := net.Listen("tcp", ipAddress)
  15. if err != nil {
  16. log.Fatalln("服务启动失败 ", err)
  17. }
  18. for {
  19. accept, err := listen.Accept()
  20. if err != nil {
  21. log.Fatalln("获取连接失败 ", err)
  22. }
  23. go handleConn(accept)
  24. }
  25. }
  26. var n int32
  27. func handleConn(c net.Conn) {
  28. defer func() {
  29. atomic.AddInt32(&n, -1)
  30. }()
  31. atomic.AddInt32(&n, 1)
  32. fmt.Printf("连接数 %d\n", atomic.LoadInt32(&n))
  33. fmt.Printf("当前连接 %s\n", c.RemoteAddr())
  34. //var filterChan = make(chan []byte, 32)
  35. var dataChan = make(chan []byte, 32)
  36. readData := func() {
  37. for {
  38. var buf = make([]byte, 1024)
  39. n, err := c.Read(buf)
  40. if err != nil {
  41. if err == io.EOF {
  42. log.Println("已到达文件结束")
  43. break
  44. }
  45. log.Println("读数据失败", err)
  46. break
  47. }
  48. if n == 0 {
  49. log.Println("未读取到数据")
  50. break
  51. }
  52. dataChan <- buf[:n]
  53. }
  54. }
  55. //filterData := func() {
  56. //
  57. //}
  58. handleData := func() {
  59. for {
  60. v, ok := <-dataChan
  61. if !ok {
  62. return
  63. }
  64. adu := UnPack(v)
  65. c.RemoteAddr()
  66. ParseAndResponse(adu, c)
  67. }
  68. }
  69. go readData()
  70. go handleData()
  71. }