data-service/duty-service/src/main/java/cn/com/taiji/duty/controller/DutyLoginLogController.java

package cn.com.taiji.duty.controller;


import cn.com.taiji.duty.model.*;
import cn.com.taiji.duty.service.IDutyAccountService;
import cn.com.taiji.duty.service.IDutyLoginLogService;
import cn.com.taiji.duty.service.IDutyOperLogService;
import cn.com.taiji.duty.utils.*;
import com.alibaba.excel.EasyExcel;
import com.alibaba.fastjson.JSON;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * <p>
 * 席位管理-登录日志 前端控制器
 * </p>
 *
 * @author yangyue
 * @since 2023-11-08
 */
@Api(tags = "登录日志")
@Slf4j
@RestController
@RequestMapping("/login/log")
public class DutyLoginLogController {

    @Autowired
    private IDutyLoginLogService loginLogService;

    @Autowired
    private IDutyOperLogService operLogService;

    @Autowired
    protected StringRedisTemplate redisTemplate;

    @Autowired
    private IDutyAccountService accountService;

    @ApiOperation("登录日志查询")
    @GetMapping("search")
    public AjaxResult search(Long id) {
        DutyLoginLog one = loginLogService.getById(id);
        AjaxResult ajaxResult = new AjaxResult(200, "请求成功!");
        ajaxResult.put("data", one);
        return ajaxResult;
    }

    @ApiOperation("登录日志列表")
    @PostMapping("list")
    public TableDataInfo list(HttpServletRequest request, @RequestBody DutyLoginLogVo vo) {
        String token = JwtUtils.getToken(request);
        String username = redisTemplate.opsForValue().get(token);
        DutyAccount account = accountService.findByUsername(username);
        if (account != null) {
            List<DutyRole> roles = account.getRoles();
            List<String> stringList = roles.parallelStream().map(DutyRole::getPermissionValue).filter(Objects::nonNull).distinct().collect(Collectors.toList());
            String join = StringUtils.join(stringList, ",");
            if (join.contains("ZBY")) {
                vo.setDeptId(account.getDept().getId());
            }
        }
        PageHelper.startPage(vo.getPageNum(), vo.getPageSize());
        List<DutyLoginLog> list = loginLogService.queryList(vo);
        TableDataInfo rspData = new TableDataInfo();
        rspData.setCode(HttpStatus.SUCCESS);
        rspData.setMsg("查询成功");
        rspData.setRows(list);
        rspData.setTotal(new PageInfo(list).getTotal());
        return rspData;
    }

    @ApiOperation("导出登录日志列表")
    @PostMapping("export")
    public void export(HttpServletRequest request, HttpServletResponse response, @RequestBody(required = false) DutyLoginLogVo vo) {
        String token = JwtUtils.getToken(request);
        String username = redisTemplate.opsForValue().get(token);
        DutyAccount account = accountService.findByUsername(username);
        if (account != null) {
            List<DutyRole> roles = account.getRoles();
            List<String> stringList = roles.parallelStream().map(DutyRole::getPermissionValue).filter(Objects::nonNull).distinct().collect(Collectors.toList());
            String join = StringUtils.join(stringList, ",");
            if (join.contains("ZBY")) {
                vo.setDeptId(account.getDept().getId());
            }
        }
        List<DutyLoginLog> list = loginLogService.queryList(vo);
        list.forEach(loginLog -> {
            if (loginLog.getDuration() != null) {
                String result="";
                long hours = TimeUnit.SECONDS.toHours(loginLog.getDuration());
                long minutes = TimeUnit.SECONDS.toMinutes(loginLog.getDuration()) % 60;
                long seconds = loginLog.getDuration() % 60;
                if(hours!=0){
                    result+=hours+"时";
                }
                if(minutes!=0){
                    result+=minutes+"分";
                }
                if(seconds!=0){
                    result+=seconds+"秒";
                }
                loginLog.setDurationValue(result);
            }
        });
        DutyOperLog operLog = new DutyOperLog();
        operLog.setOperModule("8");
        operLog.setOperType("3");
        operLogService.saveOperLog(operLog);
        exportExcel(response,DutyLoginLog.class,list,"登录日志列表");
    }

    // @ApiOperation("登录日志新增")
    @PostMapping("save")
    public AjaxResult save(@RequestBody DutyLoginLog loginLog) {
        boolean save = loginLogService.save(loginLog);
        AjaxResult ajaxResult = null;
        if (!save) {
            ajaxResult = new AjaxResult(500, "请求失败!");
            return ajaxResult;
        }
        ajaxResult = new AjaxResult(200, "请求成功!");
        ajaxResult.put("data", loginLog.getId());
        return ajaxResult;
    }

    // @ApiOperation("登录日志编辑")
    @PutMapping("edit")
    public AjaxResult update(@RequestBody DutyLoginLog loginLog) {
        boolean update = loginLogService.updateById(loginLog);
        AjaxResult ajaxResult = null;
        if (!update) {
            ajaxResult = new AjaxResult(500, "请求失败!");
            return ajaxResult;
        }
        ajaxResult = new AjaxResult(200, "请求成功!");
        return ajaxResult;
    }

    // @ApiOperation("登录日志删除")
    @DeleteMapping("delete")
    public AjaxResult delete(Long id) {
        boolean remove = loginLogService.removeById(id);
        AjaxResult ajaxResult = null;
        if (!remove) {
            ajaxResult = new AjaxResult(500, "请求失败!");
            return ajaxResult;
        }
        ajaxResult = new AjaxResult(200, "请求成功!");
        return ajaxResult;
    }

    /**
     * 导出表格数据
     * @param cl       表格字段实体类
     * @param data     查询数据
     * @param sheetName  表格名称
     */
    public void exportExcel(HttpServletResponse response, Class cl, List data, String sheetName){
        HashMap<String, String> map = new HashMap<>();
        try {
            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
            response.setCharacterEncoding("utf-8");
            String fileName = URLEncoder.encode(sheetName, "UTF-8").replaceAll("\\+", "%20");
            response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
            EasyExcel.write(response.getOutputStream(), cl).autoCloseStream(Boolean.FALSE).sheet(sheetName).doWrite(data);
        } catch (IOException e) {
            response.reset();
            response.setContentType("application/json");
            response.setCharacterEncoding("utf-8");
            map.put("status", "failure");
            map.put("message", "下载文件失败" + e.getMessage());
            try {
                response.getWriter().println(JSON.toJSONString(map));
            } catch (IOException ex) {
                ex.printStackTrace();
            }
        }
    }
}