From d11f68c67efa2bf97a6f1b06380ba2b96e59d85f Mon Sep 17 00:00:00 2001
From: moicen <moicen@foxmail.com>
Date: Sat, 1 Jan 2022 11:41:23 +0800
Subject: [PATCH] use hty host header

---
 upload.lua       | 7 ++++---
 upload_audio.lua | 3 +++
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/upload.lua b/upload.lua
index 7836b41..58919db 100644
--- a/upload.lua
+++ b/upload.lua
@@ -16,11 +16,11 @@ local TaskTypes = {
 
 local task_server = ngx.var.task_server
 local htyuc = ngx.var.htyuc
-local host = ngx.var.host
+local htyhost = ngx.req.get_headers().HtyHost
 
 ngx.log(ngx.INFO, "TASK_SERVER -> ", task_server)
 ngx.log(ngx.INFO, "HTYUC -> ", htyuc)
-ngx.log(ngx.INFO, "HOST -> ", host)
+ngx.log(ngx.INFO, "HTY HOST -> ", htyhost)
 
 local file_dir = ngx.var.tmp_file_dir
 local file
@@ -68,7 +68,7 @@ local function verifyJwtToken(token)
                 ssl_verify = false, -- 设置参数 ssl_verify 为false 不校验ssl证书
                 method = "POST",
                 headers = {
-                    ["Host"] = host,
+                    ['HtyHost'] = htyhost,
                     ["Authorization"] = token,
                 },
             }
@@ -185,6 +185,7 @@ while true do
                         ["Content-Type"] = "application/json",
                         ["Authorization"] = authHeader,
                         ["HtySudoerToken"] = sudoerToken,
+                        ['HtyHost'] = htyhost
                     },
                     body = body_text,
                 }
diff --git a/upload_audio.lua b/upload_audio.lua
index c59c3d2..a2d034b 100644
--- a/upload_audio.lua
+++ b/upload_audio.lua
@@ -13,6 +13,7 @@ print('req_body...', ngx.req.get_body_data())
 local media_id = req_body['media_id']
 local authHeader = ngx.req.get_headers().Authorization
 local sudoerToken = ngx.req.get_headers().HtySudoerToken
+local htyHost = ngx.req.get_headers().HtyHost
 
 local function create_task()
     local task_server = ngx.var.task_server
@@ -20,6 +21,7 @@ local function create_task()
     ngx.log(ngx.INFO, 'remote_url -> ', remote_url)
     ngx.log(ngx.INFO, 'Authorization -> ', authHeader)
     ngx.log(ngx.INFO, 'HtySudoerToken -> ', sudoerToken)
+    ngx.log(ngx.INFO, 'HtyHost -> ', htyHost)
 
     local body_text = cjson.encode({ task_type = TaskTypes.CONVERT_AUDIO_FILE, data = { media_id = media_id } })
 
@@ -33,6 +35,7 @@ local function create_task()
                     ["Content-Type"] = "application/json",
                     ["Authorization"] = authHeader,
                     ["HtySudoerToken"] = sudoerToken,
+                    ["HtyHost"] = htyHost
                 },
                 body = body_text,
             }