diff options
Diffstat (limited to 'src')
3 files changed, 150 insertions, 2 deletions
diff --git a/src/main/java/net/geedge/confagent/controller/ConfigController.java b/src/main/java/net/geedge/confagent/controller/ConfigController.java index b56f826..3cea8a8 100644 --- a/src/main/java/net/geedge/confagent/controller/ConfigController.java +++ b/src/main/java/net/geedge/confagent/controller/ConfigController.java @@ -4,11 +4,13 @@ import cn.hutool.core.util.StrUtil; import cn.hutool.log.Log; import net.geedge.confagent.entity.LokiPushUrlEntity; import net.geedge.confagent.entity.LokiPushUrlsEntity; +import net.geedge.confagent.util.ConfagentUtil; import net.geedge.confagent.util.R; import net.geedge.confagent.util.RCode; import net.geedge.confagent.util.Tool; import net.geedge.confagent.util.Tool.YamlUtil; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.web.bind.annotation.*; @@ -25,6 +27,16 @@ public class ConfigController extends BaseController{ @Value("${confagent.lokiPushUrlsFile:lokiPushUrls.yml}") private String lokiFile; + + @Value("${server.port:10090}") + private Integer agentPort; + + @Value("${confagent.loki.config}") + private String lokiConfPath; + + @Autowired + private ConfagentUtil confagentUtil; + private static String rootPath = Tool.WebPathUtil.getRootPath(); @@ -38,11 +50,23 @@ public class ConfigController extends BaseController{ //进行验证：target和token均非空 if(lokiPushUrls!=null && lokiPushUrls.getLokiPushUrls()!=null) { + int lokiPort = 13100; + Map<String, Object> lokiMap = Tool.YamlUtil.readAsMap(lokiConfPath); + Map<String, Object> serverObj = (Map<String, Object>) lokiMap.get("server"); + if (serverObj != null) { + lokiPort = (Integer) serverObj.get("http_listen_port"); + } + List<LokiPushUrlEntity> lokiPushUrlList = lokiPushUrls.getLokiPushUrls(); for(LokiPushUrlEntity urlEntity:lokiPushUrlList) { if(!validLokiPushUrl(urlEntity)) { return R.error(RCode.LOKI_PUSH_URL_ISNULL); } + // mode = local , target 为 本地 loki地址 + if (StrUtil.equals("local", urlEntity.getMode())) { + urlEntity.setTarget(String.format("127.0.0.1:%s/loki/api/v1/push", lokiPort)); + urlEntity.setToken(confagentUtil.readToken()); + } } } @@ -53,10 +77,14 @@ public class ConfigController extends BaseController{ } /** - * @Description 验证：target和token均非空 + * @Description mode = remote 验证：target和token均非空 */ public Boolean validLokiPushUrl(LokiPushUrlEntity lokiPushUrlsEntity){ - return StrUtil.isNotBlank(lokiPushUrlsEntity.getTarget())&&StrUtil.isNotBlank(lokiPushUrlsEntity.getToken()); + // return StrUtil.isNotBlank(lokiPushUrlsEntity.getTarget())&&StrUtil.isNotBlank(lokiPushUrlsEntity.getToken()); + if (StrUtil.equals("remote", lokiPushUrlsEntity.getMode())) { + return StrUtil.isNotBlank(lokiPushUrlsEntity.getTarget()) && StrUtil.isNotBlank(lokiPushUrlsEntity.getToken()); + } + return true; } } diff --git a/src/main/java/net/geedge/confagent/controller/TalonProxyController.java b/src/main/java/net/geedge/confagent/controller/TalonProxyController.java new file mode 100644 index 0000000..ab954be --- /dev/null +++ b/src/main/java/net/geedge/confagent/controller/TalonProxyController.java @@ -0,0 +1,119 @@ +package net.geedge.confagent.controller; + +import cn.hutool.core.net.url.UrlBuilder; +import cn.hutool.log.Log; +import com.alibaba.fastjson.JSONObject; +import net.geedge.confagent.util.ConfagentUtil; +import net.geedge.confagent.util.R; +import net.geedge.confagent.util.RCode; +import net.geedge.confagent.util.Tool; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.OutputStream; +import java.util.ArrayList; +import java.util.LinkedHashMap; +import java.util.List; +import java.util.Map; +import java.util.concurrent.Callable; +import java.util.concurrent.Future; +import java.util.concurrent.TimeUnit; + +@RestController +@RequestMapping +@SuppressWarnings("all") +public class TalonProxyController extends BaseController { + + private final static Log log = Log.get(); + + @Autowired + private ConfagentUtil confagentUtil; + + /** + * talon 批量代理接口 + * + * @param mapList + * @return + */ + @PostMapping(value = "/talon/proxy/**") + public void overwriteConfig(@RequestBody List<Map<String, Object>> talonConfigList, HttpServletRequest request, HttpServletResponse response) { + String proxyPath = request.getServletPath().replace("/talon/proxy", ""); + String token = request.getHeader("Authorization"); + + // 校验 token + R checkToken = confagentUtil.checkToken(token); + if (!Tool.ObjectUtil.equal(RCode.SUCCESS.getCode(), checkToken.getCode())) { + ConfagentUtil.writeResponse(response, checkToken); + return; + } + + // promtail config 接口代理 + if ("/promtail/config".equals(proxyPath)) { + List<Map> resultMapList = new ArrayList<>(); + int successNum = 0; + for (Map<String, Object> talonConfig : talonConfigList) { + // ip:port + String target = (String) talonConfig.get("target"); + String talonToken = (String) talonConfig.get("token"); + Map requestParams = (Map) talonConfig.get("params"); + Callable<String> task = new Callable<String>() { + @Override + public String call() throws Exception { + UrlBuilder urlBuilder = UrlBuilder.create().setScheme("http").setHost(target.split(":")[0]).setPort(Integer.valueOf(target.split(":")[1])).appendPath("nz-talon").appendPath("promtail").appendPath("config"); + String result = null; + try { + result = Tool.HttpUtil.createPost(urlBuilder.build()).auth(talonToken).body(JSONObject.toJSONString(requestParams), "application/json;charset=UTF-8").execute().body(); + } catch (RuntimeException e) { + log.error(String.format("请求 %s 接口失败", urlBuilder.toString()), e); + result = JSONObject.toJSONString(R.error(500, e.getMessage())); + } + return result; + } + }; + Future<String> f = Tool.ThreadUtil.execAsync(task); + Map resultMap = new LinkedHashMap(4); + resultMap.put("assetId", talonConfig.get("assetId")); + try { + String resultStr = f.get(5, TimeUnit.SECONDS); + if (Tool.StrUtil.isNotEmpty(resultStr)) { + R parseObject = JSONObject.parseObject(resultStr, R.class); + if (RCode.SUCCESS.getCode().equals((Integer) parseObject.get("code"))) { + successNum++; + } + } + resultMap.put("result", resultStr); + } catch (Exception e) { + log.error(e); + resultMap.put("result", JSONObject.toJSONString(R.error(500, e.getMessage()))); + } + resultMapList.add(resultMap); + } + + OutputStream outputStream = null; + try { + log.info("talon 返回内容：" + resultMapList.toString()); + R r = R.ok(); + Map dataMap = new LinkedHashMap(); + dataMap.put("total", resultMapList.size()); + dataMap.put("success", successNum); + dataMap.put("fail", resultMapList.size() - successNum); + dataMap.put("list", resultMapList); + r.put("data", dataMap); + outputStream = response.getOutputStream(); + Tool.IoUtil.writeUtf8(outputStream, true, JSONObject.toJSONString(r)); + outputStream.flush(); + } catch (Exception e) { + log.error(e); + } finally { + Tool.IoUtil.close(outputStream); + } + } else { + // TODO + } + } +} diff --git a/src/main/java/net/geedge/confagent/entity/LokiPushUrlEntity.java b/src/main/java/net/geedge/confagent/entity/LokiPushUrlEntity.java index 9dd3192..83c6be7 100644 --- a/src/main/java/net/geedge/confagent/entity/LokiPushUrlEntity.java +++ b/src/main/java/net/geedge/confagent/entity/LokiPushUrlEntity.java @@ -4,6 +4,7 @@ import lombok.Data; @Data public class LokiPushUrlEntity { + private String mode; private String token; private String target; } |