无关风月
2024-09-10 caa2b51bb4133f5a2e59eac242e744fb57f864a0
ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TCompanyController.java
@@ -1,13 +1,20 @@
package com.ruoyi.other.controller;
import org.springframework.web.bind.annotation.RequestMapping;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.account.api.dto.UnitListQueryDto;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.other.api.domain.TCompany;
import com.ruoyi.other.service.TCompanyService;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.bind.annotation.RestController;
import javax.annotation.Resource;
import java.time.LocalDateTime;
/**
 * <p>
 *  前端控制器
 * 前端控制器
 * </p>
 *
 * @author 无关风月
@@ -16,6 +23,32 @@
@RestController
@RequestMapping("/t-company")
public class TCompanyController {
   @Resource
   private TCompanyService companyService;
   //单位分页
   @PostMapping(value = "/unit/page")
   public R<Page<TCompany>> unitPage(@RequestBody UnitListQueryDto unitListQueryDto) {
      Page<TCompany> page = companyService.lambdaQuery()
            .orderByDesc(TCompany::getCreateTime)
            .like(unitListQueryDto.getCompanyName() != null && unitListQueryDto.getCompanyName().isEmpty(), TCompany::getName, unitListQueryDto.getCompanyName())
            .page(Page.of(unitListQueryDto.getPageCurr(), unitListQueryDto.getPageSize()));
      return R.ok(page);
   }
   //单位添加
   @PostMapping(value = "/unit/addOrUpdate")
   public R add(@RequestBody TCompany tCompany) {
      if (tCompany.getId() == null) {
         tCompany.setCreateTime(LocalDateTime.now());
      }
      companyService.saveOrUpdate(tCompany);
      return R.ok();
   }
   @PostMapping(value = "/unit/delete")
   public R delete(@RequestParam("id") Integer id) {
      return R.ok(companyService.removeById(id));
   }
}