|
|
@@ -25,10 +25,8 @@ import org.springframework.stereotype.Service;
|
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
-import java.util.ArrayList;
|
|
|
-import java.util.HashMap;
|
|
|
-import java.util.List;
|
|
|
-import java.util.Map;
|
|
|
+import java.time.format.DateTimeFormatter;
|
|
|
+import java.util.*;
|
|
|
|
|
|
|
|
|
/**
|
|
|
@@ -246,4 +244,60 @@ public class CostProjectMemoManagerImpl extends BaseManagerImpl<CostProjectMemoD
|
|
|
Page<CostProjectMemo> page =baseMapper.getIndexList(objectPage,userId,req.getContent(),req.getMemoDate());
|
|
|
return page;
|
|
|
}
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public Map<String, List<CostProjectMemo>> getDataMonth(String month) {
|
|
|
+ // 获取本月的所有项目,装到Map,key为日期,value为项目
|
|
|
+ IUser iUser = ContextUtil.getCurrentUser();
|
|
|
+ User user = null;
|
|
|
+ try {
|
|
|
+ user = userService.getByAccount(iUser.getAccount());
|
|
|
+ } catch (Exception e) {
|
|
|
+ throw new RuntimeException(e);
|
|
|
+ }
|
|
|
+ List<CostProjectMemo> costProjectMemos = baseMapper.getDataMonth(user.getDataScope()
|
|
|
+ ,user.getProvinceCode()
|
|
|
+ ,user.getCityCode()
|
|
|
+ ,user.getCountyCode()
|
|
|
+ ,month);
|
|
|
+ Map<String, List<CostProjectMemo>> planNameMap = new HashMap<>();
|
|
|
+ if (costProjectMemos != null && costProjectMemos.size() > 0) {
|
|
|
+ //填充到planNameMap中,key为日期,value为项目、
|
|
|
+ for (CostProjectMemo costProjectMemo : costProjectMemos) {
|
|
|
+ if(planNameMap.containsKey(costProjectMemo.getMemoDate().format(DateTimeFormatter.ofPattern("yyyy-MM-dd")))){
|
|
|
+ planNameMap.get(costProjectMemo.getMemoDate().format(DateTimeFormatter.ofPattern("yyyy-MM-dd"))).add(costProjectMemo);
|
|
|
+ }else{
|
|
|
+ List<CostProjectMemo> list = new ArrayList<>();
|
|
|
+ list.add(costProjectMemo);
|
|
|
+ planNameMap.put(costProjectMemo.getMemoDate().format(DateTimeFormatter.ofPattern("yyyy-MM-dd")),list);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // 再为map排下序,按日期
|
|
|
+ Map<String, List<CostProjectMemo>> sortedPlanNameMap = new TreeMap<>(planNameMap);
|
|
|
+ return sortedPlanNameMap;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public Map<String, List<CostProjectMemo>> getIndexDataMonth(String month) {
|
|
|
+ IUser iUser = ContextUtil.getCurrentUser();
|
|
|
+ String userId = iUser.getUserId();
|
|
|
+ List<CostProjectMemo> costProjectMemos = baseMapper.getIndexDataMonth(userId,month);
|
|
|
+ Map<String, List<CostProjectMemo>> planNameMap = new HashMap<>();
|
|
|
+ if (costProjectMemos != null && costProjectMemos.size() > 0) {
|
|
|
+ //填充到planNameMap中,key为日期,value为项目、
|
|
|
+ for (CostProjectMemo costProjectMemo : costProjectMemos) {
|
|
|
+ if(planNameMap.containsKey(costProjectMemo.getMemoDate().format(DateTimeFormatter.ofPattern("yyyy-MM-dd")))){
|
|
|
+ planNameMap.get(costProjectMemo.getMemoDate().format(DateTimeFormatter.ofPattern("yyyy-MM-dd"))).add(costProjectMemo);
|
|
|
+ }else{
|
|
|
+ List<CostProjectMemo> list = new ArrayList<>();
|
|
|
+ list.add(costProjectMemo);
|
|
|
+ planNameMap.put(costProjectMemo.getMemoDate().format(DateTimeFormatter.ofPattern("yyyy-MM-dd")),list);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // 再为map排下序,按日期
|
|
|
+ Map<String, List<CostProjectMemo>> sortedPlanNameMap = new TreeMap<>(planNameMap);
|
|
|
+ return sortedPlanNameMap;
|
|
|
+ }
|
|
|
}
|