rollouts/lua_configuration/trafficrouting_ingress/mse.lua at e5b5c059681bd343b1c75b2b7b1309a32f5cb68a · openkruise/rollouts · GitHub

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
65
66
67
68
69
70
function split(input, delimiter)
    local arr = {}
    string.gsub(input, '[^' .. delimiter ..']+', function(w) table.insert(arr, w) end)
    return arr
end

annotations = {}
if ( obj.annotations )
then
    annotations = obj.annotations
end
annotations["nginx.ingress.kubernetes.io/canary"] = "true"
annotations["nginx.ingress.kubernetes.io/canary-by-cookie"] = nil
annotations["nginx.ingress.kubernetes.io/canary-by-header"] = nil
annotations["nginx.ingress.kubernetes.io/canary-by-header-pattern"] = nil
annotations["nginx.ingress.kubernetes.io/canary-by-header-value"] = nil
-- MSE extended annotations
annotations["mse.ingress.kubernetes.io/canary-by-query"] = nil
annotations["mse.ingress.kubernetes.io/canary-by-query-pattern"] = nil
annotations["mse.ingress.kubernetes.io/canary-by-query-value"] = nil
annotations["nginx.ingress.kubernetes.io/canary-weight"] = nil
if ( obj.weight ~= "-1" )
then
    annotations["nginx.ingress.kubernetes.io/canary-weight"] = obj.weight
end
if ( annotations["mse.ingress.kubernetes.io/service-subset"] )
then
    annotations["mse.ingress.kubernetes.io/service-subset"] = "gray"
end

if ( obj.requestHeaderModifier )
then
    local str = ''
    for _,header in ipairs(obj.requestHeaderModifier.set) do
        str = str..string.format("%s %s", header.name, header.value)
    end
    annotations["mse.ingress.kubernetes.io/request-header-control-update"] = str
end
if ( not obj.matches )
then
    return annotations
end
for _,match in ipairs(obj.matches) do
    if match.headers and next(match.headers) ~= nil then
        header = match.headers[1]
        if ( header.name == "canary-by-cookie" )
        then
            annotations["nginx.ingress.kubernetes.io/canary-by-cookie"] = header.value
        else
            annotations["nginx.ingress.kubernetes.io/canary-by-header"] = header.name
            if ( header.type == "RegularExpression" )
            then
                annotations["nginx.ingress.kubernetes.io/canary-by-header-pattern"] = header.value
            else
                annotations["nginx.ingress.kubernetes.io/canary-by-header-value"] = header.value
            end
        end
    end
    if match.queryParams and next(match.queryParams) ~= nil then
        queryParam = match.queryParams[1]
        annotations["nginx.ingress.kubernetes.io/canary-by-query"] = queryParam.name
        if ( queryParam.type == "RegularExpression" )
        then
            annotations["nginx.ingress.kubernetes.io/canary-by-query-pattern"] = queryParam.value
        else
            annotations["nginx.ingress.kubernetes.io/canary-by-query-value"] = queryParam.value
        end
    end
end
return annotations