czg_team/cashier_server
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

钉钉考勤

Browse Source
This commit is contained in:
张松
2025-12-01 15:12:21 +08:00
parent 0f908da39f
commit f9d697ea58
1 changed files with 43 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
cash-service/market-service/src/main/java/com/czg/service/market/service/impl/DingService.java
View File

@@ -22,6 +22,9 @@ import org.springframework.stereotype.Service;
import java.math.BigDecimal;
import java.util.*;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.stream.Collectors;
@Service
@@ -80,6 +83,10 @@ public class DingService {
}
public Map<String, String> getColumns(String key, String secret, Long shopId) {
Object value = redisService.get("ding_columns:" + shopId);
if (value instanceof String && StrUtil.isNotBlank((String) value)) {
return JSONObject.parseObject((String) value, Map.class);
}
DingTalkClient client = new DefaultDingTalkClient("https://oapi.dingtalk.com/topapi/attendance/getattcolumns");
OapiAttendanceGetattcolumnsRequest req = new OapiAttendanceGetattcolumnsRequest();
OapiAttendanceGetattcolumnsResponse rsp = null;
@@ -102,6 +109,7 @@ public class DingService {
})
.collect(Collectors.toMap(item -> ((JSONObject) item).getLong("id").toString(), item -> ((JSONObject) item).getString("name")));
redisService.set("ding_columns:" + shopId, JSONObject.toJSONString(columMap));
return columMap;
}
@@ -177,39 +185,54 @@ public class DingService {
}
}
public ArrayList<DingUserVO> getUserList(String name,String key, String secret, Long shopId, String userId) {
public ArrayList<DingUserVO> getUserList(String name, String key, String secret, Long shopId, String userId) {
try {
DingTalkClient client = new DefaultDingTalkClient("https://oapi.dingtalk.com/topapi/v2/department/listsub");
OapiV2DepartmentListsubRequest req = new OapiV2DepartmentListsubRequest();
req.setDeptId(1L);
req.setLanguage("zh_CN");
OapiV2DepartmentListsubResponse rsp = client.execute(req, getToken(key, secret, shopId));
Set<Long> deptIdList = getData(rsp, true).getJSONArray("result").stream().map(item -> ((JSONObject) item)
.getLong("dept_id")).collect(Collectors.toSet());
ArrayList<DingUserVO> dingUserVOS = new ArrayList<>();
for (Long dptId : deptIdList) {
dingUserVOS.addAll(getUserByDeptId(dptId, key, secret, shopId));
}
if (StrUtil.isNotBlank(name)) {
return dingUserVOS.stream().filter(item -> {
boolean flag = true;
if (StrUtil.isNotBlank(userId)) {
flag = item.getUserid().equals(userId);
}
Set<Long> deptIdList = getData(rsp, true).getJSONArray("result")
.stream()
.map(item -> ((JSONObject) item).getLong("dept_id"))
.collect(Collectors.toSet());
if (StrUtil.isNotBlank(name)) {
flag = item.getName().contains(name);
}
return flag;
}).collect(Collectors.toCollection(ArrayList::new));
// 线程池：按 CPU 核数 * 2
ExecutorService pool = Executors.newFixedThreadPool(
Math.min(deptIdList.size(), Runtime.getRuntime().availableProcessors() * 2)
);
List<Future<List<DingUserVO>>> futures = new ArrayList<>();
// 并发拉取
for (Long deptId : deptIdList) {
futures.add(pool.submit(() -> getUserByDeptId(deptId, key, secret, shopId)));
}
return dingUserVOS;
}catch (Exception e) {
// 汇总结果
List<DingUserVO> allUsers = new ArrayList<>();
for (Future<List<DingUserVO>> f : futures) {
allUsers.addAll(f.get());
}
pool.shutdown();
// 正确过滤逻辑
return allUsers.stream().filter(item -> {
if (StrUtil.isNotBlank(userId) && !item.getUserid().equals(userId)) {
return false;
}
return !StrUtil.isNotBlank(name) || item.getName().contains(name);
}).collect(Collectors.toCollection(ArrayList::new));
} catch (Exception e) {
throw new RuntimeException(e);
}
}
public ArrayList<DingAttendanceStatsVO> getUserReport(Long shopId, String key, String secret, DateTime startTime, DateTime endTime, String name, String userId) {
Map<String, String> columns = getColumns(key, secret, shopId);
ArrayList<DingUserVO> userList = getUserList(name, key, secret, shopId, userId);