编写Go的TCP服务来替代PHP的Swoole

Tue, 30 Apr 2019 21:22:46 +0800

背景说明

目前公司采用微服务架构，主要开发语言为PHP，通过Swoole开启TCP服务供业务端调用。通过公司内部编写的PHP扩展封装客户端调用逻辑。

需求

暂定使用Go语言开发新的业务，并提供TCP服务。其中老的PHP项目要通过原有的客户端扩展实现无修改调用。

解决方案

通过阅读客户端扩展源码了解调用逻辑。编写简单的测试如下。

<?php

$_client = new \swoole_client(SWOOLE_SOCK_TCP | SWOOLE_KEEP);
$_client->set([
 'open_length_check' => true,
 'package_length_type' => 'N',
 'package_length_offset' => 0,
 'package_body_offset' => 4,
 'package_max_length' => 24657920,
]);
if (false == $_client->connect("127.0.0.1", 8880)) {
 printf("err_msg: %s err_code: %s" . PHP_EOL, var_export($_client->errMsg, true), var_export($_client->errCode, true));
}

// 随便测试个请求参数
$data = [
 'api' => 'getUserInfo',
 'params' => [
 'user_id' => 123
 ]
];
$data = json_encode($data);
$data = gzcompress($data, 9);
$_client->send(pack("N", strlen($data)) . $data);

$res = $_client->recv();
$end = getTime();

$data = json_decode($res, true);

其中前4个字节是head，表示body长度，采用二进制大端字节序编码。body先进行json编码再进行了zlib压缩。这都是编写Go的TCP服务时需要处理的。

写个简单的Go TCP服务试试，先不考虑过多的错误边界处理。

package main

import (
 "bytes"
 "compress/zlib"
 "encoding/binary"
 "fmt"
 "io"
 "net"
)

func main() {
 ln, err := net.Listen("tcp", ":8880")
 if err != nil {
 fmt.Printf("%s", err)
 }
 for {
 conn, err := ln.Accept()
 if err != nil {
 fmt.Printf("accept err:%s", err)
 }
 go handleConnection(conn)
 }
}

func handleConnection(conn net.Conn) {
 fmt.Println("on conn")

 var err error
 headLen := 4
 head := make([]byte, headLen)
 if _, err = conn.Read(head); err != nil {
 fmt.Println(err.Error())
 return
 }

 // 解码大端字节序获取body长度
 bodyLen := binary.BigEndian.Uint32(head)

 allBody := make([]byte, 0)
 readLen := 0
 for bodyLen > 0 {
 body := make([]byte, bodyLen)
 readLen, err = conn.Read(body)
 if err != nil {
 fmt.Println(err.Error())
 return
 }

 bodyLen = bodyLen - uint32(readLen)
 allBody = append(allBody, body[:readLen]...)
 }

 // 解压zlib压缩的数据 RFC 1950
 b := bytes.NewReader(allBody)
 uncompressRead, err := zlib.NewReader(b)
 if err != nil {
 fmt.Printf("uncompress data err:%s", err)
 }
 var uncompressData bytes.Buffer
 io.Copy(&uncompressData, uncompressRead)

 // 解出的json字符串
 fmt.Printf("Received:%s", uncompressData.Bytes())

 // 路由调用实际业务逻辑处理 ...
 // conn.Write()

 conn.Close()
 return
}

运行Go的TCP服务，跑一个PHP请求测试。

TCP on Nick's Blog

编写Go的TCP服务来替代PHP的Swoole

背景说明

需求

解决方案