From 7be401fb57f0fc27276f0ecceba15b5ed7ddb1e8 Mon Sep 17 00:00:00 2001 From: calyrex <2213354201@qq.com> Date: Tue, 4 Mar 2025 19:51:57 +0800 Subject: [PATCH] calyrex 3.4 --- .../controller/GetUserInfo_controller.java | 2 +- .../HTGD_IT13_GETIP_CONTROLLER.java | 42 ++++++++++++++++++- 2 files changed, 42 insertions(+), 2 deletions(-) diff --git a/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/GetUserInfo_controller.java b/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/GetUserInfo_controller.java index 147272c..232a70f 100644 --- a/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/GetUserInfo_controller.java +++ b/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/GetUserInfo_controller.java @@ -45,7 +45,7 @@ public class GetUserInfo_controller { List> records = (List>)datas.get("records"); if (records.size()>0){ Map map = records.get(0); - json.put("username",map.get("username")); + json.put("username",map.get("USERNAME")); return WeaResult.success(json); }else { log.error("getCustomId 相关字段未找到"); diff --git a/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/HTGD_IT13_GETIP_CONTROLLER.java b/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/HTGD_IT13_GETIP_CONTROLLER.java index ff6d132..b1a9c40 100644 --- a/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/HTGD_IT13_GETIP_CONTROLLER.java +++ b/secondev-history-action/src/main/java/com/weaver/seconddev/interfaces/workflow/controller/HTGD_IT13_GETIP_CONTROLLER.java @@ -5,12 +5,20 @@ import com.alibaba.fastjson.JSONObject; import com.weaver.common.authority.annotation.WeaPermission; import com.weaver.common.base.entity.result.WeaResult; import com.weaver.common.security.json.JSON; +import com.weaver.ebuilder.common.util.TenantContext; +import com.weaver.ebuilder.datasource.api.entity.ExecuteSqlEntity; +import com.weaver.ebuilder.datasource.api.enums.SourceType; +import com.weaver.ebuilder.datasource.api.service.DataSetService; +import com.weaver.framework.rpc.context.impl.TenantRpcContext; +import com.weaver.teams.api.tenant.Tenant; import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.util.HashMap; +import java.util.List; import java.util.Map; /** @@ -25,6 +33,9 @@ import java.util.Map; @RestController @RequestMapping({"/papi/secondev/workflow/controller" }) public class HTGD_IT13_GETIP_CONTROLLER { + + @Autowired + private DataSetService dataSetService; @GetMapping("/getip") @ResponseBody @WeaPermission(publicPermission = true) @@ -32,11 +43,40 @@ public class HTGD_IT13_GETIP_CONTROLLER { response.setContentType("application/x-www-form-urlencoded; charset=utf-8"); log.error("HTGD_IT13_GETIP_CONTROLLER"+ JSON.parseObjectToString(request)); String remoteAddr = request.getRemoteAddr(); + String remoteUser = request.getRemoteUser(); + log.error("remoteAddr:"+remoteAddr); Map data = new HashMap<>(); - data.put("ip",remoteAddr); + String sql = " select * from eteams.dbo.employee where id = " + remoteUser; + Map stringObjectMap = executeForQuery(null2String(SourceType.LOGIC), "weaver-hr-service", sql); + if(String.valueOf(stringObjectMap.get("status")).equals("OK")){ + List> records = (List>)stringObjectMap.get("records"); + if (records.size()>0){ + Map map = records.get(0); + data.put("JOB_NUM",map.get("JOB_NUM")); + }else { + log.error("getCustomId 相关字段未找到"); + } + }else { + log.error("getUserId status-->"+ stringObjectMap.get("status")); + } + data.put("ip",remoteAddr); return WeaResult.success(JSONObject.toJSONString(data)); } + public Map executeForQuery(String sourceType, String groupId, String sql) { +// log.error("executeForQuery sourceType-->" + sourceType + ",groupId-->" + groupId + ",sql-->" + sql); + final Tenant currentTenant = TenantContext.getCurrentTenant(); + TenantRpcContext.setTargetTenantKey(String.valueOf(currentTenant)); + ExecuteSqlEntity executeSqlEntity = new ExecuteSqlEntity(); + executeSqlEntity.setSql(cn.hutool.core.codec.Base64.encode(sql)); + executeSqlEntity.setGroupId(groupId); //groupid,可以访问 E10地址/api/datasource/ds/group?sourceType=LOGIC 获取 + executeSqlEntity.setSourceType(SourceType.valueOf(sourceType)); +// log.error("executeForQuery executeSqlEntity-->"+executeSqlEntity.getSql()+"-->"+executeSqlEntity.getGroupId()+"-->"+executeSqlEntity.getSourceType()); + Map datas = dataSetService.executeSql(executeSqlEntity); + log.error("executeForQuery datas-->" + datas); + TenantRpcContext.removeTargetTenantKey(); + return datas; + } public static String null2String(String s){ return s == null ? "" : s;