package com.xiaopeng.operation.controller;
|
|
|
import cn.afterturn.easypoi.excel.ExcelExportUtil;
|
import cn.afterturn.easypoi.excel.ExcelImportUtil;
|
import cn.afterturn.easypoi.excel.entity.ExportParams;
|
import cn.afterturn.easypoi.excel.entity.ImportParams;
|
import cn.hutool.json.JSONObject;
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
import com.xiaopeng.api.model.account.SysUser;
|
import com.xiaopeng.api.model.operation.XpInvitationCode;
|
import com.xiaopeng.api.model.operation.XpWhitelist;
|
import com.xiaopeng.api.model.operation.XpWhitelistOperlog;
|
import com.xiaopeng.common.core.constant.TokenConstants;
|
import com.xiaopeng.common.core.exception.ServiceException;
|
import com.xiaopeng.common.core.utils.DateUtils;
|
import com.xiaopeng.common.core.utils.WebUtils;
|
import com.xiaopeng.common.core.utils.ip.IpUtils;
|
import com.xiaopeng.common.core.web.domain.AjaxResult;
|
import com.xiaopeng.common.core.web.page.PageInfo;
|
import com.xiaopeng.common.log.annotation.Log;
|
import com.xiaopeng.common.log.enums.BusinessType;
|
import com.xiaopeng.common.security.service.TokenService;
|
import com.xiaopeng.operation.dto.WhitelistAddDto;
|
import com.xiaopeng.operation.exportExcle.XpWhitelistExportExcel;
|
import com.xiaopeng.operation.importExcel.InvitationCodeImportExcel;
|
import com.xiaopeng.operation.importExcel.WhitelistImportExcel;
|
import com.xiaopeng.operation.query.InvitationPageListQuery;
|
import com.xiaopeng.operation.query.WhitelistPageListQuery;
|
import com.xiaopeng.operation.service.IXpInvitationCodeService;
|
import com.xiaopeng.operation.service.IXpWhitelistOperlogService;
|
import com.xiaopeng.operation.service.IXpWhitelistService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiImplicitParam;
|
import io.swagger.annotations.ApiImplicitParams;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.poi.ss.usermodel.Workbook;
|
import org.springframework.transaction.annotation.Propagation;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.annotation.Resource;
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.net.URLEncoder;
|
import java.time.LocalDateTime;
|
import java.time.format.DateTimeFormatter;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 白名单管理 前端控制器
|
* </p>
|
*
|
* @author mitao
|
* @since 2026-02-03
|
*/
|
@RequestMapping("/api/v1/whitelist")
|
@RestController
|
@Api(tags = "白名单管理")
|
@Slf4j
|
public class XpWhitelistController {
|
@Resource
|
private IXpWhitelistService xpWhitelistService;
|
@Resource
|
private IXpWhitelistOperlogService xpWhitelistOperlogService;
|
@Resource
|
private TokenService tokenService;
|
|
@ApiOperation(value = "白名单管理分页列表", response = XpWhitelist.class)
|
@PostMapping("/pageList")
|
public AjaxResult<PageInfo<XpWhitelist>> pageList(@RequestBody WhitelistPageListQuery query) {
|
PageInfo<XpWhitelist> pageList = xpWhitelistService.pageList(query);
|
return AjaxResult.ok(pageList);
|
}
|
|
/**
|
* 新增白名单
|
*/
|
@PostMapping("/add")
|
@ResponseBody
|
@ApiOperation(value = "新增白名单")
|
@Transactional
|
public AjaxResult add(@RequestBody WhitelistAddDto dto) {
|
if (!StringUtils.hasLength(dto.getAccount())) {
|
return AjaxResult.error("请输入账号");
|
}
|
List<XpWhitelist> addList = new ArrayList<>();
|
List<String> accounts = Arrays.asList(dto.getAccount().split(","));
|
SysUser sysUser = tokenService.getLoginUser().getSysUser();
|
StringBuilder operationContent = new StringBuilder("新增账号:");
|
StringBuilder accountBuilder = new StringBuilder();
|
for (String account : accounts) {
|
XpWhitelist xpWhitelist = new XpWhitelist();
|
xpWhitelist.setAccount(account);
|
xpWhitelist.setStatus(true);
|
xpWhitelist.setIsPriorityDispatch(dto.getIsPriorityDispatch());
|
xpWhitelist.setWhitelistSource(1);
|
addList.add(xpWhitelist);
|
accountBuilder.append("\"").append(account).append("\"").append(",");
|
}
|
XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
|
xpWhitelistOperlog.setOperationName(sysUser.getNickName());
|
xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
|
xpWhitelistOperlog.setOperationType(1);
|
xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
|
xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
|
StringBuilder content = operationContent.append(accountBuilder.substring(0, accountBuilder.length() - 1));
|
xpWhitelistOperlog.setOperationContent(content.toString());
|
xpWhitelistOperlogService.save(xpWhitelistOperlog);
|
xpWhitelistService.saveBatch(addList);
|
return AjaxResult.success();
|
}
|
|
/**
|
* 编辑白名单
|
*/
|
@PostMapping("/edit")
|
@ResponseBody
|
@ApiOperation(value = "编辑白名单")
|
@Transactional
|
public AjaxResult edit(@RequestBody XpWhitelist dto) {
|
XpWhitelist whitelist = xpWhitelistService.getById(dto.getId());
|
SysUser sysUser = tokenService.getLoginUser().getSysUser();
|
|
StringBuilder operationContent = new StringBuilder("编辑账号:");
|
StringBuilder accountBuilder = new StringBuilder();
|
if (!whitelist.getAccount().equals(dto.getAccount())) {
|
accountBuilder.append("白名单账号" + "\"" + whitelist.getAccount() + "\"" + "为" + "\"" + dto.getAccount() + "\"");
|
}
|
if (StringUtils.hasLength(whitelist.getNickName()) || StringUtils.hasLength(dto.getNickName())) {
|
if (!StringUtils.hasLength(whitelist.getNickName())) {
|
if (!whitelist.getNickName().equals(dto.getNickName())) {
|
accountBuilder.append("用户名称" + "\"" + whitelist.getNickName() + "\"" + "为" + "\"" + dto.getNickName() + "\"");
|
}
|
} else {
|
if (!whitelist.getNickName().equals(dto.getNickName())) {
|
accountBuilder.append("用户名称" + "\"" + "为" + "\"" + dto.getNickName() + "\"");
|
}
|
}
|
}
|
if (whitelist.getStatus() != dto.getStatus()) {
|
accountBuilder.append("状态" + "\"" + (whitelist.getIsPriorityDispatch() ? "正常" : "冻结") + "\"" + "为" + "\"" + (dto.getIsPriorityDispatch() ? "正常" : "冻结") + "\"");
|
}
|
if (whitelist.getIsPriorityDispatch() != dto.getIsPriorityDispatch()) {
|
accountBuilder.append("优先派单" + "\"" + (whitelist.getIsPriorityDispatch() ? "是" : "否") + "\"" + "为" + "\"" + (dto.getIsPriorityDispatch() ? "是" : "否") + "\"");
|
}
|
XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
|
xpWhitelistOperlog.setOperationName(sysUser.getNickName());
|
xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
|
xpWhitelistOperlog.setOperationType(2);
|
xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
|
xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
|
accountBuilder.append("\"").append(dto.getAccount()).append("\"");
|
StringBuilder content = operationContent.append(accountBuilder);
|
xpWhitelistOperlog.setOperationContent(content.toString());
|
xpWhitelistOperlogService.save(xpWhitelistOperlog);
|
whitelist.setNickName(dto.getNickName());
|
whitelist.setIsPriorityDispatch(dto.getIsPriorityDispatch());
|
whitelist.setAccount(dto.getAccount());
|
whitelist.setStatus(dto.getStatus());
|
xpWhitelistService.updateById(whitelist);
|
return AjaxResult.success();
|
}
|
|
/**
|
* 冻结解冻
|
*/
|
@GetMapping("/freezeUnfreeze")
|
@ResponseBody
|
@ApiOperation(value = "冻结解冻")
|
@Transactional
|
public AjaxResult freezeUnfreeze(@RequestParam("id") Long id) {
|
XpWhitelist whitelist = xpWhitelistService.getById(id);
|
|
SysUser sysUser = tokenService.getLoginUser().getSysUser();
|
StringBuilder operationContent = new StringBuilder("编辑账号:"+"\""+whitelist.getAccount()+"\"");
|
StringBuilder accountBuilder = new StringBuilder();
|
accountBuilder.append("状态" + "\"" + (whitelist.getIsPriorityDispatch() ? "正常" : "冻结") + "\"" + "为" + "\"" + (!whitelist.getIsPriorityDispatch() ? "正常" : "冻结") + "\"");
|
XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
|
xpWhitelistOperlog.setOperationName(sysUser.getNickName());
|
xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
|
xpWhitelistOperlog.setOperationType(2);
|
xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
|
xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
|
StringBuilder content = operationContent.append(accountBuilder);
|
xpWhitelistOperlog.setOperationContent(content.toString());
|
xpWhitelistOperlogService.save(xpWhitelistOperlog);
|
whitelist.setStatus(!whitelist.getStatus());
|
xpWhitelistService.updateById(whitelist);
|
return AjaxResult.success();
|
}
|
|
/**
|
* 批量删除
|
*/
|
@PutMapping("/deleteByIds")
|
@ResponseBody
|
@ApiOperation(value = "批量删除")
|
public AjaxResult deleteByIds(List<Long> ids) {
|
List<String> accounts = xpWhitelistService.lambdaQuery().in(XpWhitelist::getId, ids).list()
|
.stream().map(XpWhitelist::getAccount).collect(Collectors.toList());
|
xpWhitelistService.removeBatchByIds(ids);
|
SysUser sysUser = tokenService.getLoginUser().getSysUser();
|
StringBuilder operationContent = new StringBuilder("删除账号:");
|
StringBuilder accountBuilder = new StringBuilder();
|
for (String account : accounts) {
|
accountBuilder.append("\"" + account + "\""+",");
|
}
|
XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
|
xpWhitelistOperlog.setOperationName(sysUser.getNickName());
|
xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
|
xpWhitelistOperlog.setOperationType(3);
|
xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
|
xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
|
StringBuilder content = operationContent.append(accountBuilder.substring(0, accountBuilder.length() - 1));
|
xpWhitelistOperlog.setOperationContent(content.toString());
|
xpWhitelistOperlogService.save(xpWhitelistOperlog);
|
return AjaxResult.success();
|
}
|
|
/**
|
* 批量冻结
|
*/
|
@PutMapping("/freezeByIds")
|
@ResponseBody
|
@ApiOperation(value = "批量冻结")
|
public AjaxResult freezeByIds(List<Long> ids) {
|
List<XpWhitelist> list = xpWhitelistService.lambdaQuery().in(XpWhitelist::getId, ids).list();
|
for (XpWhitelist xpWhitelist : list) {
|
xpWhitelist.setStatus(false);
|
}
|
xpWhitelistService.updateBatchById(list);
|
SysUser sysUser = tokenService.getLoginUser().getSysUser();
|
StringBuilder operationContent = new StringBuilder("共计冻结"+ids+"条数据");
|
XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
|
xpWhitelistOperlog.setOperationName(sysUser.getNickName());
|
xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
|
xpWhitelistOperlog.setOperationType(6);
|
xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
|
xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
|
xpWhitelistOperlog.setOperationContent(operationContent.toString());
|
xpWhitelistOperlogService.save(xpWhitelistOperlog);
|
return AjaxResult.success();
|
}
|
|
@ApiOperation(value = "白名单导入模板下载")
|
@GetMapping("/import-template")
|
public void importTemplate() {
|
List<WhitelistImportExcel> whitelistImportExcels = new ArrayList<>();
|
WhitelistImportExcel whitelistImportExcel = new WhitelistImportExcel();
|
whitelistImportExcels.add(whitelistImportExcel);
|
Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), WhitelistImportExcel.class, whitelistImportExcels);
|
HttpServletResponse response = WebUtils.response();
|
ServletOutputStream outputStream = null;
|
try {
|
String fileName = URLEncoder.encode("白名单导入模板.xls", "utf-8");
|
response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
|
response.setHeader("content-Type", "application/vnd.ms-excel");
|
response.setHeader("Pragma", "no-cache");
|
response.setHeader("Cache-Control", "no-cache");
|
outputStream = response.getOutputStream();
|
workbook.write(outputStream);
|
} catch (IOException e) {
|
e.printStackTrace();
|
System.out.println("白名单导入模板下载失败!");
|
} finally {
|
try {
|
outputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
@Log(title = "白名单导入", businessType = BusinessType.IMPORT)
|
@ApiOperation(value = "白名单导入")
|
@Transactional(rollbackFor = Exception.class, propagation = Propagation.REQUIRES_NEW)
|
@PostMapping("/importExcel")
|
public AjaxResult<String> importExcel(@RequestPart("file") MultipartFile file) {
|
ImportParams params = new ImportParams();
|
// params.setTitleRows(1); // 标题行数
|
params.setHeadRows(1); //表头行数
|
InputStream inputStream = null;
|
// List<CustomerImportFailedData> failedData = new ArrayList<>();
|
List<WhitelistImportExcel> whitelistImportExcels;
|
try {
|
inputStream = file.getInputStream();
|
whitelistImportExcels = ExcelImportUtil.importExcel(inputStream,
|
WhitelistImportExcel.class, params);
|
} catch (Exception e) {
|
e.printStackTrace();
|
log.error("白名单导入失败:{}", e.getMessage());
|
throw new ServiceException("白名单导入失败!");
|
} finally {
|
try {
|
inputStream.close();
|
} catch (IOException e) {
|
throw new ServiceException(e.getMessage());
|
}
|
}
|
if (CollectionUtils.isEmpty(whitelistImportExcels)) {
|
throw new ServiceException("白名单数据为空!");
|
}
|
// 校验手机号白名单是否有重复的
|
long count = whitelistImportExcels.stream().map(WhitelistImportExcel::getAccount)
|
.distinct().count();
|
if (count != whitelistImportExcels.size()) {
|
return AjaxResult.error("白名单手机号存在重复!");
|
}
|
Map<String, Long> codeMap = xpWhitelistService.list().stream()
|
.collect(Collectors.toMap(XpWhitelist::getAccount, XpWhitelist::getId));
|
JSONObject result = new JSONObject();
|
List<XpWhitelist> addList = new ArrayList<>();
|
for (WhitelistImportExcel whitelistImportExcel : whitelistImportExcels) {
|
Long id = codeMap.get(whitelistImportExcel.getAccount());
|
if (id != null) {
|
result.append("白名单：[", whitelistImportExcel.getAccount() + "]已存在");
|
}
|
XpWhitelist xpWhitelist = new XpWhitelist();
|
xpWhitelist.setNickName(whitelistImportExcel.getNickName());
|
xpWhitelist.setAccount(whitelistImportExcel.getAccount());
|
xpWhitelist.setStatus(true);
|
xpWhitelist.setIsPriorityDispatch(true);
|
xpWhitelist.setWhitelistSource(1);
|
addList.add(xpWhitelist);
|
}
|
if (!result.isEmpty()) {
|
return AjaxResult.success(result.toString());
|
}
|
xpWhitelistService.saveBatch(addList);
|
SysUser sysUser = tokenService.getLoginUser().getSysUser();
|
StringBuilder operationContent = new StringBuilder("共计导入"+addList.size()+"条数据");
|
XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
|
xpWhitelistOperlog.setOperationName(sysUser.getNickName());
|
xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
|
xpWhitelistOperlog.setOperationType(4);
|
xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
|
xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
|
xpWhitelistOperlog.setOperationContent(operationContent.toString());
|
xpWhitelistOperlogService.save(xpWhitelistOperlog);
|
return AjaxResult.success();
|
}
|
|
@ApiOperation(value = "白名单导出")
|
@PostMapping(value = "/exportExcel")
|
public void exportExcel(@RequestBody WhitelistPageListQuery query) {
|
List<XpWhitelist> exportList = xpWhitelistService.exportList(query);
|
List<XpWhitelistExportExcel> xpWhitelistExportExcels = new ArrayList<>();
|
for (XpWhitelist xpWhitelist : exportList) {
|
XpWhitelistExportExcel xpWhitelistExportExcel = new XpWhitelistExportExcel();
|
xpWhitelistExportExcel.setId(xpWhitelist.getId());
|
xpWhitelistExportExcel.setAccount(xpWhitelist.getAccount());
|
xpWhitelistExportExcel.setNickName(xpWhitelist.getNickName());
|
xpWhitelistExportExcel.setStatus(xpWhitelist.getStatus());
|
xpWhitelistExportExcel.setIsPriorityDispatch(xpWhitelist.getIsPriorityDispatch());
|
xpWhitelistExportExcel.setCreateBy(xpWhitelist.getCreateBy());
|
LocalDateTime createTime = xpWhitelist.getCreateTime();
|
// 将LocalDateTime转化为yyyy-MM-dd HH:mm:ss格式字符串
|
xpWhitelistExportExcel.setCreateTime(createTime.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
|
xpWhitelistExportExcels.add(xpWhitelistExportExcel);
|
}
|
SysUser sysUser = tokenService.getLoginUser().getSysUser();
|
StringBuilder operationContent = new StringBuilder("共计导出"+exportList.size()+"条数据");
|
XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
|
xpWhitelistOperlog.setOperationName(sysUser.getNickName());
|
xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
|
xpWhitelistOperlog.setOperationType(5);
|
xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
|
xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
|
xpWhitelistOperlog.setOperationContent(operationContent.toString());
|
xpWhitelistOperlogService.save(xpWhitelistOperlog);
|
//1.获取excel模板
|
Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), XpWhitelistExportExcel.class, xpWhitelistExportExcels);
|
HttpServletResponse response = WebUtils.response();
|
response.setContentType("application/vnd.ms-excel");
|
response.setCharacterEncoding("utf-8");
|
ServletOutputStream outputStream = null;
|
try {
|
String fileName = URLEncoder.encode("白名单列表.xls", "utf-8");
|
response.setHeader("Content-dispodition", "attachment;filename=" + fileName);
|
outputStream = response.getOutputStream();
|
workbook.write(outputStream);
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
outputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
|
}
|
|
}
|
