2026-04-23 c247f5bf64c98595cc7e4efc9e1ae7ff1df79c11
src/main/java/com/ruoyi/equipmentenergyconsumption/controller/ElectricityConsumptionAreaController.java
@@ -2,33 +2,39 @@
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.device.pojo.DeviceLedger;
import com.ruoyi.equipmentenergyconsumption.dto.ElectricityConsumptionAreaTreeDto;
import com.ruoyi.equipmentenergyconsumption.pojo.ElectricityConsumptionArea;
import com.ruoyi.equipmentenergyconsumption.service.ElectricityConsumptionAreaService;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.AjaxResult;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import io.swagger.v3.oas.annotations.tags.Tag;
import io.swagger.v3.oas.annotations.Operation;
import lombok.AllArgsConstructor;
import org.springframework.util.CollectionUtils;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletResponse;
import java.util.List;
@RestController
@Api(tags = "用电区域")
@Tag(name = "用电区域")
@RequestMapping("/electricityConsumptionArea")
@AllArgsConstructor
public class ElectricityConsumptionAreaController extends BaseController {
    @Autowired
    private ElectricityConsumptionAreaService electricityConsumptionAreaService;
    private final ElectricityConsumptionAreaService electricityConsumptionAreaService;
    /**
     * 查询树结构
     */
    @GetMapping("/list")
    public List<ElectricityConsumptionAreaTreeDto> selectElectricityConsumptionAreaList() {
        return electricityConsumptionAreaService.selectElectricityConsumptionAreaList();
    }
    @GetMapping("/listPage")
    @ApiOperation("用电区域-分页查询")
    @Operation(summary = "用电区域-分页查询")
    @Log(title = "用电区域-分页查询", businessType = BusinessType.OTHER)
    public AjaxResult listPage(Page page, ElectricityConsumptionArea electricityConsumptionArea) {
        IPage<ElectricityConsumptionArea> listPage = electricityConsumptionAreaService.listPage(page, electricityConsumptionArea);
@@ -36,40 +42,20 @@
    }
    @PostMapping("/add")
    @ApiOperation("用电区域-新增")
    @Operation(summary = "用电区域-新增")
    @Log(title = "用电区域-新增", businessType = BusinessType.INSERT)
    public AjaxResult add(@RequestBody ElectricityConsumptionArea electricityConsumptionArea) {
        boolean save = electricityConsumptionAreaService.save(electricityConsumptionArea);
        boolean save = electricityConsumptionAreaService.saveOrUpdate(electricityConsumptionArea);
        return save ? AjaxResult.success() : AjaxResult.error();
    }
    @PostMapping("/update")
    @ApiOperation("用电区域-修改")
    @Log(title = "用电区域-修改", businessType = BusinessType.UPDATE)
    public AjaxResult update(@RequestBody ElectricityConsumptionArea electricityConsumptionArea) {
        boolean update = electricityConsumptionAreaService.updateById(electricityConsumptionArea);
        return update ? AjaxResult.success() : AjaxResult.error();
    }
    @DeleteMapping("/delete")
    @ApiOperation("用电区域-删除")
    @Operation(summary = "用电区域-删除")
    @Log(title = "用电区域-删除", businessType = BusinessType.DELETE)
    public AjaxResult delete(@RequestBody List<Long> ids) {
        if(CollectionUtils.isEmpty(ids)) return AjaxResult.error("请选择至少一条数据");
        boolean remove = electricityConsumptionAreaService.removeBatchByIds(ids);
        return remove ? AjaxResult.success() : AjaxResult.error();
    }
    /**
     * 导出用电区域
     */
    @Log(title = "导出用电区域", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    @ApiOperation("导出用电区域")
    public void export(HttpServletResponse response) {
        ExcelUtil<ElectricityConsumptionArea> util = new ExcelUtil<ElectricityConsumptionArea>(ElectricityConsumptionArea.class);
        util.exportExcel(response, null , "用电区域数据");
    }
}