File: test_fetch_multi.lua

package info (click to toggle)
trafficserver 9.2.5%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 53,008 kB
  • sloc: cpp: 345,484; ansic: 31,134; python: 24,200; sh: 7,271; makefile: 3,045; perl: 2,261; java: 277; pascal: 119; sql: 94; xml: 2
file content (64 lines) | stat: -rw-r--r-- 2,142 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
--  Licensed to the Apache Software Foundation (ASF) under one
--  or more contributor license agreements.  See the NOTICE file
--  distributed with this work for additional information
--  regarding copyright ownership.  The ASF licenses this file
--  to you under the Apache License, Version 2.0 (the
--  "License"); you may not use this file except in compliance
--  with the License.  You may obtain a copy of the License at
--
--  http://www.apache.org/licenses/LICENSE-2.0
--
--  Unless required by applicable law or agreed to in writing, software
--  distributed under the License is distributed on an "AS IS" BASIS,
--  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
--  See the License for the specific language governing permissions and
--  limitations under the License.

require 'os'

function process_combo(host)
    local url1 = string.format('http://%s/css/1.css', host)
    local url2 = string.format('http://%s/css/2.css', host)
    local url3 = string.format('http://%s/css/3.css', host)

    local hdr = {
        ['Host'] = host,
        ['User-Agent'] = 'blur blur',
    }

    local ct = {
        header = hdr,
        method = 'GET'
    }

    local arr = ts.fetch_multi(
            {
                {url1, ct},
                {url2, ct},
                {url3, ct},
            })

    local ctype = arr[1].header['Content-Type']
    local body = arr[1].body .. arr[2].body .. arr[3].body

    local resp =  'HTTP/1.1 200 OK\r\n' ..
                  'Server: ATS/5.2.0\r\n' ..
                  'Last-Modified: ' .. os.date("%a, %d %b %Y %H:%M:%S GMT", os.time()) .. '\r\n' ..
                  'Cache-Control: max-age=7200\r\n' ..
                  'Accept-Ranges: bytes\r\n' ..
                  'Content-Type: ' .. ctype .. '\r\n' ..
                  'Content-Length: ' .. string.format('%d', string.len(body)) .. '\r\n\r\n' ..
                  body

    ts.say(resp)
end

function do_remap()
    local inner =  ts.http.is_internal_request()
    if inner ~= 0 then
        return 0
    end

    local h = ts.client_request.header['Host']
    ts.http.intercept(process_combo, h)
end