feat: TCP supports custom header packing and unpacking format

swoft-cloud · Sep 17, 2019 · a6fbe93 · a6fbe93
1 parent c2c4b93
commit a6fbe93
Showing 1 changed file with 48 additions and 2 deletions.
diff --git a/src/tcp/src/Protocol.php b/src/tcp/src/Protocol.php
@@ -124,6 +124,20 @@ class Protocol
      */
     private $openLengthCheck = false;
 
+    /**
+     * Header pack format
+     *
+     * @var string
+     */
+    private $headerPackFormat = self::HEADER_PACK_FORMAT;
+
+    /**
+     * Header unpack format
+     *
+     * @var string
+     */
+    private $headerUnpackFormat = self::HEADER_UNPACK_FORMAT;
+
     /**
      * @link https://wiki.swoole.com/wiki/page/463.html
      * @link https://www.php.net/manual/en/function.pack.php
@@ -239,7 +253,7 @@ public function packBody(string $body): string
         }
 
         // Use length check
-        $format = self::HEADER_PACK_FORMAT;
+        $format = $this->headerPackFormat;
 
         // TODO
         // Args sort please see self::HEADER_UNPACK_FORMAT
@@ -263,7 +277,7 @@ public function unpackData(string $data): array
         }
 
         // Use length check
-        $format  = self::HEADER_UNPACK_FORMAT;
+        $format  = $this->headerUnpackFormat;
         $headLen = $this->packageBodyOffset;
 
         // Like: ['type' => 'json', 'len' => 254, ]
@@ -541,4 +555,36 @@ public function setPackageMaxLength(int $packageMaxLength): void
     {
         $this->packageMaxLength = $packageMaxLength;
     }
+
+    /**
+     * @return string
+     */
+    public function getHeaderPackFormat(): string
+    {
+        return $this->headerPackFormat;
+    }
+
+    /**
+     * @param string $headerPackFormat
+     */
+    public function setHeaderPackFormat(string $headerPackFormat): void
+    {
+        $this->headerPackFormat = $headerPackFormat;
+    }
+
+    /**
+     * @return string
+     */
+    public function getHeaderUnpackFormat(): string
+    {
+        return $this->headerUnpackFormat;
+    }
+
+    /**
+     * @param string $headerUnpackFormat
+     */
+    public function setHeaderUnpackFormat(string $headerUnpackFormat): void
+    {
+        $this->headerUnpackFormat = $headerUnpackFormat;
+    }
 }