|
|
@@ -1,9 +1,12 @@
|
|
|
package com.kym.admin.controller;
|
|
|
|
|
|
+import cn.dev33.satoken.stp.StpUtil;
|
|
|
import com.kym.common.R;
|
|
|
import com.kym.common.annotation.SysLog;
|
|
|
import com.kym.entity.admin.Station;
|
|
|
import com.kym.service.admin.StationService;
|
|
|
+import com.kym.service.utils.KymCache;
|
|
|
+import org.springframework.context.annotation.Lazy;
|
|
|
import org.springframework.format.annotation.DateTimeFormat;
|
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
@@ -26,8 +29,11 @@ public class StationController {
|
|
|
|
|
|
private final StationService stationService;
|
|
|
|
|
|
- public StationController(StationService stationService) {
|
|
|
+ private final KymCache kymCache;
|
|
|
+
|
|
|
+ public StationController(StationService stationService, @Lazy KymCache kymCache) {
|
|
|
this.stationService = stationService;
|
|
|
+ this.kymCache = kymCache;
|
|
|
}
|
|
|
|
|
|
@SysLog("充电站数据统计")
|
|
|
@@ -44,7 +50,12 @@ public class StationController {
|
|
|
|
|
|
@GetMapping("listStation")
|
|
|
R listStation(@RequestParam(required = false) String stationName) {
|
|
|
- var res = stationService.lambdaQuery().like(isNotNull(stationName), Station::getStationName, stationName).orderByDesc(Station::getStationId).list();
|
|
|
+ // 判断数据权限
|
|
|
+ var adminStationId = kymCache.getAdminUserStationId(StpUtil.getLoginIdAsLong());
|
|
|
+ var res = stationService.lambdaQuery()
|
|
|
+ .eq(isNotNull(adminStationId), Station::getStationId, adminStationId)
|
|
|
+ .like(isNotNull(stationName), Station::getStationName, stationName)
|
|
|
+ .orderByDesc(Station::getStationId).list();
|
|
|
return R.success(res);
|
|
|
}
|
|
|
|