huatai/src/main/java/com/izouma/awesomeadmin/web/FinancialDataController.java

package com.izouma.awesomeadmin.web;

import java.util.*;

import com.izouma.awesomeadmin.model.OperateHistory;
import com.izouma.awesomeadmin.model.UserInfo;
import com.izouma.awesomeadmin.service.OperateHistoryService;
import com.izouma.awesomeadmin.util.ExportExcelUtil;
import org.apache.commons.lang.StringUtils;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authz.annotation.RequiresAuthentication;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import com.izouma.awesomeadmin.constant.AppConstant;
import com.izouma.awesomeadmin.dto.Page;
import com.izouma.awesomeadmin.dto.Result;
import com.izouma.awesomeadmin.model.FinancialData;
import com.izouma.awesomeadmin.service.FinancialDataService;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
*  controller类
*/
@Controller
@RequestMapping("/financialData")
public class FinancialDataController {

    @Autowired
    private FinancialDataService financialDataService;

    @Autowired
    private OperateHistoryService operateHistoryService;


    /**
    * <p>获取全部记录。</p>
    */
    @RequiresAuthentication
    @RequestMapping(value = "/all", method = RequestMethod.GET)
    @ResponseBody
    public Result all(FinancialData record) {
        List<FinancialData> pp = financialDataService.getFinancialDataList(record);
        return new Result(true, pp);
    }

    /**
    * <p>根据Id。</p>
    */
    @RequestMapping(value = "/getFinancialData", method = RequestMethod.GET)
    @ResponseBody
    public Result getFinancialData(@RequestParam(required = false, value = "id") String id) {
        FinancialData data = financialDataService.getFinancialDataById(id);
        return new Result(true, data);
    }

    /**
    * <p>根据条件获取。</p>
    */
    @RequestMapping(value = "/getOne", method = RequestMethod.GET)
    @ResponseBody
    public Result getOne(FinancialData record) {
        FinancialData data = financialDataService.getFinancialData(record);
        return new Result(true, data);
    }


    /**
    * <p>分页查询。</p>
    */
    @RequestMapping(value = "/page", method = RequestMethod.GET)
    @ResponseBody
    public Result page(Page page, FinancialData record) {

        Subject subject = SecurityUtils.getSubject();
        UserInfo userInfo = (UserInfo) subject.getPrincipal();
        if(!"2".equals(userInfo.getDepartId()))
        {
            record.setStationId(Integer.parseInt(userInfo.getDepartId()));
        }

        Map<String, Object> result = new HashMap<>();

        List<FinancialData> pp =financialDataService.getFinancialDataByPage(page, record);

        result.put(AppConstant.PAGE, page);
        result.put("pp", pp);
        return new Result(true, result);
    }


    /**
    * <p>保存。</p>
    */
    @RequestMapping(value = "/save", method = RequestMethod.POST)
    @ResponseBody
    public Result save(FinancialData record) {

        Subject subject = SecurityUtils.getSubject();
        UserInfo userInfo = (UserInfo) subject.getPrincipal();
        if(!"2".equals(userInfo.getDepartId()))
        {
            record.setStationId(Integer.parseInt(userInfo.getDepartId()));
        }

        boolean num = financialDataService.createFinancialData(record);
        if (num) {
            OperateHistory oh = new OperateHistory();
            oh.setUsername(userInfo.getUsername());
            oh.setStationId(Integer.parseInt(userInfo.getDepartId()));
            oh.setContent("添加金融大数据: id="+record.getId());
            operateHistoryService.createOperateHistory(oh);

        return new Result(true, record.getId());
        }
        return new Result(false, "保存异常");
    }

    /**
    * <p>更新信息。</p>
    */
    @RequestMapping(value = "/update", method = RequestMethod.POST)
    @ResponseBody
    public Result updateFinancialData(FinancialData record) {

        Subject subject = SecurityUtils.getSubject();
        UserInfo userInfo = (UserInfo) subject.getPrincipal();
        boolean num = financialDataService.updateFinancialData(record);
        if (num) {
            OperateHistory oh = new OperateHistory();
            oh.setUsername(userInfo.getUsername());
            oh.setStationId(Integer.parseInt(userInfo.getDepartId()));
            oh.setContent("更新金融大数据: id="+record.getId());
            operateHistoryService.createOperateHistory(oh);

            return new Result(true, "保存成功");
        }
        return new Result(false, "保存异常");
    }

    /**
    * <p>删除。</p>
    */
    @RequestMapping(value = "/del", method = RequestMethod.POST)
    @ResponseBody
    public Result deleteFinancialData(FinancialData record) {

        boolean num = financialDataService.deleteFinancialData(record);
        if (num) {
        return new Result(true, "删除成功");
        }
        return new Result(false, "删除异常");
    }

    /**
    * 导出Excel
    * @param request
    * @param response
    * @param record
    * @throws Exception
    */
    @RequestMapping(value = "/exportExcel", method = RequestMethod.GET)
    @ResponseBody
    public void exportExcel(HttpServletRequest request, HttpServletResponse response, FinancialData record) throws Exception {

    List<FinancialData> financialDatas = financialDataService.getFinancialDataList(record);


        String sheetName = "financial_data";
        String titleName = "金融大数据数据表";
        String fileName = "金融大数据表";
        int columnNumber = 7;
        int[] columnWidth = { 20,  20,  20,  20,  20,  20,  20 };
        String[] columnName = {  "" ,   "删除标识" ,   "更新时间" ,   "更新人" ,   "创建时间" ,   "创建人" ,   "运营点ID"  };
        String[][] dataList = new String[financialDatas.size()][7];

        for (int i = 0; i < financialDatas.size(); i++) {

                        dataList[i][0] = String.valueOf(financialDatas.get(i).getId());
                        dataList[i][1] = String.valueOf(financialDatas.get(i).getDelFlag());
                        dataList[i][2] = String.valueOf(financialDatas.get(i).getUpdateTime());
                        dataList[i][3] = String.valueOf(financialDatas.get(i).getUpdateUser());
                        dataList[i][4] = String.valueOf(financialDatas.get(i).getCreateTime());
                        dataList[i][5] = String.valueOf(financialDatas.get(i).getCreateUser());
                        dataList[i][6] = String.valueOf(financialDatas.get(i).getStationId());
                    }


        ExportExcelUtil.ExportWithResponse(sheetName, titleName, fileName,
        columnNumber, columnWidth, columnName, dataList, response);


        }
    }