Code/sms_water/src/main/java/com/huaxu/controller/RevenueController.java

package com.huaxu.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.huaxu.common.StringUtils;
import com.huaxu.dto.AlarmSettingDto;
import com.huaxu.dto.MonthRevenueDto;
import com.huaxu.dto.MonthSellwaterDto;
import com.huaxu.entity.AlarmSetting;
import com.huaxu.entity.SceneEntity;
import com.huaxu.model.AjaxMessage;
import com.huaxu.model.LoginUser;
import com.huaxu.model.Pagination;
import com.huaxu.model.ResultStatus;
import com.huaxu.service.AlarmSettingService;
import com.huaxu.service.RevenueService;
import com.huaxu.service.SceneService;
import com.huaxu.util.OrgInfoUtil;
import com.huaxu.util.UserUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;

/**
 * 营收控制层
 *
 * @author yjy
 * @since 2021-2-24
 */
@RestController
@RequestMapping("/revenue")
@Api(tags = "营收信息")
public class RevenueController {
    /**
     * 服务对象
     */
    @Autowired
    private RevenueService revenueService;

    @Autowired
    private OrgInfoUtil orgInfoUtil;

    /**
     * 查询本月售水信息
     *
     * @param companyOrgId
     * @return
     */
    @RequestMapping(value = "selectMonthSell", method = RequestMethod.GET)
    @ApiOperation(value = "查询本月售水信息")
    public AjaxMessage<List<MonthSellwaterDto>> selectMonthSell(
            @ApiParam(value = "公司机构id", required = false)@RequestParam(required = false) Integer companyOrgId) {
        List<MonthSellwaterDto> result = revenueService.selectMonthSell(companyOrgId);
        return new AjaxMessage<>(ResultStatus.OK, result);
    }
    /**
     * 查询近6个月总售水量
     *
     * @param companyOrgId
     * @return
     */
    @RequestMapping(value = "selectMonthSellTotal", method = RequestMethod.GET)
    @ApiOperation(value = "查询近6个月总售水量")
    public AjaxMessage<List<MonthSellwaterDto>> selectMonthSellTotal(
            @ApiParam(value = "公司机构id", required = false)@RequestParam(required = false) Integer companyOrgId) {
        List<MonthSellwaterDto> result = revenueService.selectMonthSellTotal(companyOrgId);
        return new AjaxMessage<>(ResultStatus.OK, result);
    }
    /**
     * 查询月营收情况
     *
     * @param companyOrgId
     * @return
     */
    @RequestMapping(value = "selectMonthRevenue", method = RequestMethod.GET)
    @ApiOperation(value = "查询月营收情况")
    public AjaxMessage<List<MonthRevenueDto>> selectMonthRevenue(
            @ApiParam(value = "近几月", required = true)@RequestParam Integer months,
            @ApiParam(value = "公司机构id", required = false)@RequestParam(required = false) Integer companyOrgId) {
        List<MonthRevenueDto> result = revenueService.selectMonthRevenue(companyOrgId,months);
        return new AjaxMessage<>(ResultStatus.OK, result);
    }
    /**
     * 查询本月各分公司的营收情况
     *
     * @return
     */
    @RequestMapping(value = "selectCompanyRevenue", method = RequestMethod.GET)
    @ApiOperation(value = "查询本月各分公司的营收情况")
    public AjaxMessage<List<MonthRevenueDto>> selectCompanyRevenue() {
        List<MonthRevenueDto> result = revenueService.selectCompanyRevenue();
        for(MonthRevenueDto revenue : result){
            revenue.setCompanyOrgName(orgInfoUtil.getOrgName(revenue.getCompanyOrgId()));
        }
        return new AjaxMessage<>(ResultStatus.OK, result);
    }
}