src/service/agw.lua · 小鱼/proxygateway

代码拉取完成,页面将自动刷新

richgiteeai

Watch

不关注关注所有动态仅关注版本发行动态关注但不提醒动态

1 Star 0 Fork 37

小鱼/proxygateway

forked from 小飞/proxygateway

代码 Issues 0 Pull Requests 0 Wiki 统计流水线

服务

加入 Gitee

与超过 1400万开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)

免费加入

已有帐号? 立即登录

文件

master

分支 (1)

管理

master

克隆/下载

HTTPS SSH SVN SVN+SSH 下载ZIP

提示

下载代码请复制以下命令到终端执行

为确保你提交的代码身份被 Gitee 正确识别,请执行以下命令完成配置

git config --global user.name userName 
git config --global user.email userEmail

初次使用 SSH 协议进行代码克隆、推送等操作时,需按下述提示完成 SSH 配置

1 生成 RSA 密钥

2 获取 RSA 公钥内容,并配置到 SSH公钥中

在 Gitee 上使用 SVN,请访问使用指南

使用 HTTPS 协议时,命令行会出现如下账号密码验证步骤。基于安全考虑,Gitee 建议配置并使用私人令牌替代登录密码进行克隆、推送等操作

Username for 'https://gitee.com': userName

Password for 'https://userName@gitee.com': # 私人令牌

分支 1

标签 0

proxygateway

src

service

agw.lua

agw.lua 3.37 KB

local cjson = require "cjson"
local api_model = require "model.api"
local service_model = require "model.module"
local server_model = require "model.server"
local domain_model = require "model.domain"

local cache = ngx.shared.cache

local log = ngx.log
local ERR = ngx.ERR
local DEBUG = ngx.DEBUG
local agw_service = {}

function check_state(ip, port)
 local sock = ngx.socket.tcp()
 sock:settimeout(5000)
 local ok, err = sock:connect(ip, port)
 sock:close()
 if not ok then
 log(ERR, "server connect err:", err)
 return false
 end
 return true
end

function set_domain_api_info(domainId, domainName)
 local services, err = service_model.getServices(domainId)
 local service_api_uri_map = {}
 local service_api_uri_array = {}
 for k, service in pairs(services) do
 local servers=server_model.getServiceServersWithState(service["id"], 1)
 local apis=api_model.getServiceApis(service["id"])
 for k, api in pairs(apis) do
 api["servers"] = servers
 api["host"] = service["host"]
 if service["host"] == "" then
 api["host"] = domainName
 end
 local original_uri, index=string.gsub(api["original_uri"], "%$([0-9]+)", "%%%1")
 api["original_uri"] = original_uri
 service_api_uri_map[api["request_uri"]] = api
 table.insert(service_api_uri_array, api["request_uri"])
 end
 end
 table.sort(service_api_uri_array, function(a,b) return string.len(a) > string.len(b) end)
 local config = {api_uri_map = service_api_uri_map, api_uri_array = service_api_uri_array}
 cache:set(domainName, cjson.encode(config))
end

function agw_service.syncApiInfo()
 local domains, err = domain_model.getDomains()
 for k, domain in pairs(domains) do
 set_domain_api_info(domain["id"], domain["name"])
 end
end

function agw_service.checkState()
 local servers = server_model.getAllServers()
 for k, server in pairs(servers) do
 if check_state(server["ip"], server["port"]) == false then
 log(ERR, "server down:", server["ip"]..":"..server["port"])
 server_model.updateState(server["id"], 0)
 else
 server_model.updateState(server["id"], 1)
 end
 end
end

function agw_service.getServices(domainId)
 local services, err = service_model.getServices(domainId)
 for k, service in pairs(services) do
 local up_servers_count = table.getn(server_model.getServiceServersWithState(service["id"], 1))
 local down_servers_count = table.getn(server_model.getServiceServersWithState(service["id"], 0))
 local status = "no backend servers"
 if up_servers_count ~= 0 or down_servers_count ~= 0 then
 status = "up "..up_servers_count..",down "..down_servers_count
 end
 service["status"] = status 
 end
 return services, err
end

function agw_service.deleteDomain(domainId)
 domain_model.delete(domainId)
 local services = service_model.getServices(domainId)
 for k, service in pairs(services) do
 api_model.deleteByServiceId(service["id"])
 server_model.deleteByServiceId(service["id"])
 end
 service_model.deleteByDomainId(domainId)
end

function agw_service.deleteService(serviceId)
 api_model.deleteByServiceId(serviceId)
 server_model.deleteByServiceId(serviceId)
 service_model.delete(serviceId)
end

return agw_service

一键复制编辑原始数据按行查看历史

fengfei 提交于 2016年05月08日 02:06 +08:00 . init

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 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99

local cjson = require "cjson"
local api_model = require "model.api"
local service_model = require "model.module"
local server_model = require "model.server"
local domain_model = require "model.domain"

local cache = ngx.shared.cache

local log = ngx.log
local ERR = ngx.ERR
local DEBUG = ngx.DEBUG
local agw_service = {}

function check_state(ip, port)
 local sock = ngx.socket.tcp()
 sock:settimeout(5000)
 local ok, err = sock:connect(ip, port)
 sock:close()
 if not ok then
 log(ERR, "server connect err:", err)
 return false
 end
 return true
end

function set_domain_api_info(domainId, domainName)
 local services, err = service_model.getServices(domainId)
 local service_api_uri_map = {}
 local service_api_uri_array = {}
 for k, service in pairs(services) do
 local servers=server_model.getServiceServersWithState(service["id"], 1)
 local apis=api_model.getServiceApis(service["id"])
 for k, api in pairs(apis) do
 api["servers"] = servers
 api["host"] = service["host"]
 if service["host"] == "" then
 api["host"] = domainName
 end
 local original_uri, index=string.gsub(api["original_uri"], "%$([0-9]+)", "%%%1")
 api["original_uri"] = original_uri
 service_api_uri_map[api["request_uri"]] = api
 table.insert(service_api_uri_array, api["request_uri"])
 end
 end
 table.sort(service_api_uri_array, function(a,b) return string.len(a) > string.len(b) end)
 local config = {api_uri_map = service_api_uri_map, api_uri_array = service_api_uri_array}
 cache:set(domainName, cjson.encode(config))
end

function agw_service.syncApiInfo()
 local domains, err = domain_model.getDomains()
 for k, domain in pairs(domains) do
 set_domain_api_info(domain["id"], domain["name"])
 end
end

function agw_service.checkState()
 local servers = server_model.getAllServers()
 for k, server in pairs(servers) do
 if check_state(server["ip"], server["port"]) == false then
 log(ERR, "server down:", server["ip"]..":"..server["port"])
 server_model.updateState(server["id"], 0)
 else
 server_model.updateState(server["id"], 1)
 end
 end
end

function agw_service.getServices(domainId)
 local services, err = service_model.getServices(domainId)
 for k, service in pairs(services) do
 local up_servers_count = table.getn(server_model.getServiceServersWithState(service["id"], 1))
 local down_servers_count = table.getn(server_model.getServiceServersWithState(service["id"], 0))
 local status = "no backend servers"
 if up_servers_count ~= 0 or down_servers_count ~= 0 then
 status = "up "..up_servers_count..",down "..down_servers_count
 end
 service["status"] = status 
 end
 return services, err
end

function agw_service.deleteDomain(domainId)
 domain_model.delete(domainId)
 local services = service_model.getServices(domainId)
 for k, service in pairs(services) do
 api_model.deleteByServiceId(service["id"])
 server_model.deleteByServiceId(service["id"])
 end
 service_model.deleteByDomainId(domainId)
end

function agw_service.deleteService(serviceId)
 api_model.deleteByServiceId(serviceId)
 server_model.deleteByServiceId(serviceId)
 service_model.delete(serviceId)
end

return agw_service