TPromoCodeController.java 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716
  1. package com.szwl.controller;
  2. import cn.afterturn.easypoi.excel.ExcelExportUtil;
  3. import cn.afterturn.easypoi.excel.entity.ExportParams;
  4. import com.alibaba.fastjson.JSONArray;
  5. import com.alibaba.fastjson.JSONObject;
  6. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  7. import com.baomidou.mybatisplus.core.metadata.IPage;
  8. import com.baomidou.mybatisplus.core.toolkit.Wrappers;
  9. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  10. import com.szwl.constant.JoinpayConstant;
  11. import com.szwl.constant.ResponseCodesEnum;
  12. import com.szwl.feign.bean.PayFeign;
  13. import com.szwl.model.bo.JsonMessage;
  14. import com.szwl.model.bo.R;
  15. import com.szwl.model.bo.ResponseModel;
  16. import com.szwl.model.dto.PromoCodeTarget;
  17. import com.szwl.model.dto.PromoCodeTargetEn;
  18. import com.szwl.model.entity.*;
  19. import com.szwl.model.utils.DateUtils;
  20. import com.szwl.service.*;
  21. import com.szwl.util.IDGenerator;
  22. import io.swagger.annotations.Api;
  23. import io.swagger.annotations.ApiOperation;
  24. import org.apache.commons.lang.StringUtils;
  25. import org.apache.poi.ss.usermodel.Workbook;
  26. import org.springframework.beans.factory.annotation.Autowired;
  27. import org.springframework.web.bind.annotation.*;
  28. import javax.servlet.http.HttpServletResponse;
  29. import java.io.IOException;
  30. import java.io.OutputStream;
  31. import java.math.BigDecimal;
  32. import java.net.URLEncoder;
  33. import java.text.SimpleDateFormat;
  34. import java.util.*;
  35. import java.util.stream.Collectors;
  36. import static com.szwl.constant.ResponseCodesEnum.A0001;
  37. import static com.szwl.constant.ResponseCodesEnum.A0002;
  38. import static com.szwl.model.utils.DateUtils.PATTERN_yyyy_MM_dd_HH_mm_ss;
  39. /**
  40. * <p>
  41. * 优惠码列表 前端控制器
  42. * </p>
  43. *
  44. * @author wuhs
  45. * @since 2022-06-17
  46. */
  47. @Api(value = "/tPromoCode", tags = {"优惠码"})
  48. @RestController
  49. @RequestMapping("/tPromoCode")
  50. public class TPromoCodeController {
  51. @Autowired
  52. TEquipmentService tEquipmentService;
  53. @Autowired
  54. TPromoCodeService promoCodeService;
  55. @Autowired
  56. TPriceService priceService;
  57. @Autowired
  58. TAdminService adminService;
  59. @Autowired
  60. TOrderService orderService;
  61. @Autowired
  62. TJoinpayMchService joinpayMchService;
  63. @Autowired
  64. PayFeign payFeign;
  65. @ApiOperation(value = "获取优惠码")
  66. @GetMapping("/getTPromoCode")
  67. public ResponseModel<IPage<TPromoCode>> getTPromoCode(@RequestParam String code) {
  68. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  69. query.eq(TPromoCode::getCode, code);
  70. Page<TPromoCode> page = new Page<>(1, 10, true);
  71. IPage<TPromoCode> iPage = promoCodeService.page(page, query);
  72. return R.ok(iPage);
  73. }
  74. @ApiOperation(value = "获取优惠码")
  75. @GetMapping("/getPromoCode")
  76. public ResponseModel<TPromoCode> getPromoCode(@RequestParam String code) {
  77. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  78. String[] split = code.split("-");
  79. query.eq(TPromoCode::getCode, split[0]);
  80. query.eq(TPromoCode::getAdminId, split[1]);
  81. List<TPromoCode> list = promoCodeService.list(query);
  82. if (list.size() > 0) {
  83. return R.ok(list.get(0));
  84. }
  85. return R.ok();
  86. }
  87. @ApiOperation(value = "获取优惠码")
  88. @GetMapping("/selectTPromoCode")
  89. public ResponseModel<TPromoCode> selectTPromoCode(@RequestParam String code, @RequestParam String adminId) {
  90. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  91. query.eq(TPromoCode::getCode, code);
  92. query.eq(TPromoCode::getAdminId, adminId);
  93. // List<TPromoCode> list = promoCodeService.list(query);
  94. TPromoCode promoCode = promoCodeService.getOne(query);
  95. return R.ok(promoCode);
  96. }
  97. @ApiOperation(value = "小程序获取优惠码")
  98. @GetMapping("/getCode")
  99. public ResponseModel<TPromoCode> getCode(String clientId, String code) {
  100. if (StringUtils.isEmpty(clientId)) {
  101. return null;
  102. }
  103. LambdaQueryWrapper<TEquipment> query = Wrappers.lambdaQuery();
  104. query.eq(TEquipment::getClientId, clientId);
  105. List<TEquipment> list = tEquipmentService.list(query);
  106. TEquipment equipment = list.get(0);
  107. LambdaQueryWrapper<TPromoCode> query1 = Wrappers.lambdaQuery();
  108. query1.eq(TPromoCode::getCode, code);
  109. query1.eq(TPromoCode::getAdminId, equipment.getAdminId());
  110. List<TPromoCode> list1 = promoCodeService.list(query1);
  111. if (list1.size() > 0) {
  112. TPromoCode tPromoCode = list1.get(0);
  113. String isUse = tPromoCode.getIsUse();
  114. if (isUse.equals("1")) {
  115. return R.fail(A0002, "已使用");
  116. }
  117. Date lastUseDate = tPromoCode.getLastUseDate();
  118. Date date = new Date();
  119. if (date.getTime() > lastUseDate.getTime()) {
  120. return R.fail(A0002, "已过期");
  121. }
  122. return R.ok(tPromoCode);
  123. } else {
  124. return R.fail(A0002, "没有找到优惠码");
  125. }
  126. }
  127. @ApiOperation(value = "修改优惠码状态")
  128. @GetMapping("/updateCode")
  129. public ResponseModel<?> updateCode(String clientId, String code) {
  130. if (StringUtils.isEmpty(clientId)) {
  131. return null;
  132. }
  133. LambdaQueryWrapper<TEquipment> query = Wrappers.lambdaQuery();
  134. query.eq(TEquipment::getClientId, clientId);
  135. List<TEquipment> list = tEquipmentService.list(query);
  136. TEquipment equipment = list.get(0);
  137. LambdaQueryWrapper<TPromoCode> query1 = Wrappers.lambdaQuery();
  138. query1.eq(TPromoCode::getCode, code);
  139. query1.eq(TPromoCode::getAdminId, equipment.getAdminId());
  140. List<TPromoCode> list1 = promoCodeService.list(query1);
  141. if (!list1.isEmpty()) {
  142. TPromoCode tPromoCode = list1.get(0);
  143. tPromoCode.setIsUse("1");
  144. if (StringUtils.isNotEmpty(equipment.getName())) {
  145. tPromoCode.setUseBy(equipment.getName());
  146. } else {
  147. tPromoCode.setUseBy(clientId);
  148. }
  149. SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  150. String time = sdf.format(new Date());
  151. tPromoCode.setUseDate(time);
  152. promoCodeService.updateById(tPromoCode);
  153. return R.ok(null, "修改成功");
  154. } else {
  155. return R.fail(A0002, "没有找到优惠码");
  156. }
  157. }
  158. @ApiOperation(value = "修改优惠码")
  159. @GetMapping("/updatePromoCode")
  160. public void updatePromoCode(@RequestParam(value = "id") String id, @RequestParam(value = "type") String type) {
  161. TPromoCode promoCode = promoCodeService.getById(id);
  162. promoCode.setIsUse(type);
  163. if (type.equals("1")) {
  164. promoCode.setUseDate(DateUtils.formatDate(new Date(), PATTERN_yyyy_MM_dd_HH_mm_ss));
  165. }
  166. promoCodeService.updateById(promoCode);
  167. }
  168. @ApiOperation(value = "生成0折优惠码")
  169. @GetMapping("/addPromoCode")
  170. public void addPromoCode(@RequestParam(value = "adminId") String adminId, @RequestParam(value = "lastUseDate") Date lastUseDate,
  171. @RequestParam(value = "number") String number, @RequestParam(value = "codeNum") String codeNum) {
  172. int num = Integer.parseInt(number);
  173. List<String> codes = new ArrayList<>();
  174. if (StringUtils.isEmpty(codeNum)) {
  175. for (int i = 0; i < num; i++) {
  176. String code = orderService.initSn(Long.valueOf(adminId));
  177. StringBuffer str = new StringBuffer();
  178. str.append(code.substring(0, 6));
  179. codes.add(str.toString());
  180. }
  181. } else {
  182. codes.add(codeNum);
  183. }
  184. TAdmin admin = adminService.getById(adminId);
  185. for (String code : codes) {
  186. TPromoCode promoCode = new TPromoCode();
  187. promoCode.setCreateDate(new Date());
  188. promoCode.setModifyDate(new Date());
  189. promoCode.setCode(Long.parseLong(code));
  190. promoCode.setAdminId(adminId);
  191. promoCode.setIsUse("0");
  192. promoCode.setDiscount(0f);
  193. promoCode.setLastUseDate(lastUseDate);
  194. // promoCode.setCreateDate(new Date());
  195. promoCode.setType("0");
  196. promoCode.setUserName(admin.getUsername());
  197. promoCodeService.save(promoCode);
  198. }
  199. }
  200. @ApiOperation(value = "生成n元优惠券")
  201. @GetMapping("/addYHJ")
  202. public ResponseModel<?> addYHJ(Long wxId, String flag) {
  203. if (wxId == null || StringUtils.isEmpty(flag)) {
  204. return R.fail(A0001);
  205. }
  206. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  207. query.eq(TPromoCode::getWxId, wxId);
  208. // if(flag.equals("YHJ5")||flag.equals("YHJ10")){
  209. // query.eq(TPromoCode::getType,"2");
  210. // List<TPromoCode> list = promoCodeService.list(query);
  211. // if(list.size()>0){
  212. // return R.ok(1,"此券每人限领一张,您已领取过,不能再领!");
  213. // }
  214. // }
  215. query.eq(TPromoCode::getType, "2");
  216. List<TPromoCode> list = promoCodeService.list(query);
  217. if (list.size() > 0) {
  218. return R.ok(1, "此券每人限领一张,您已领取过,不能再领!");
  219. }
  220. String code = orderService.initSn(Long.valueOf(wxId));
  221. TPromoCode promoCode = new TPromoCode();
  222. promoCode.setCreateDate(new Date());
  223. promoCode.setModifyDate(new Date());
  224. promoCode.setCode(Long.parseLong(code.substring(code.length() - 8, code.length())));
  225. promoCode.setAdminId(null);
  226. promoCode.setIsUse("0");
  227. // if(flag.equals("YHJ5")){
  228. // promoCode.setDiscount(5f);
  229. // promoCode.setType("2");
  230. // }
  231. // if(flag.equals("YHJ10")){
  232. // promoCode.setDiscount(10f);
  233. // promoCode.setType("2");
  234. // }
  235. String[] js = flag.split("J");
  236. String s = js[1];
  237. int price = Integer.parseInt(s);
  238. if (price > 100) {
  239. return R.fail(A0001);
  240. }
  241. promoCode.setDiscount(Float.valueOf(price));
  242. promoCode.setType("2");
  243. Long data = Calendar.getInstance().getTimeInMillis();
  244. Long t = 3l;
  245. Long mon = 30L * 24L * 60L * 60L * 1000L;
  246. Long d = data + (mon * t);
  247. Date lastUseDate = new Date(Long.parseLong(String.valueOf(d)));
  248. promoCode.setLastUseDate(lastUseDate);
  249. promoCode.setCreateDate(new Date());
  250. promoCode.setUserName(null);
  251. promoCode.setWxId(wxId);
  252. promoCodeService.save(promoCode);
  253. return R.ok(promoCode, "领取成功");
  254. }
  255. @ApiOperation(value = "核销优惠券")
  256. @GetMapping("/useYHJ")
  257. public void useYHJ(@RequestParam String code, @RequestParam String adminId, @RequestParam String clentId) {
  258. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  259. query.eq(TPromoCode::getCode, code);
  260. query.eq(TPromoCode::getIsUse, "0");
  261. List<TPromoCode> list = promoCodeService.list(query);
  262. if (list.size() > 0) {
  263. TPromoCode promoCode = list.get(0);
  264. promoCode.setUseDate(DateUtils.formatDate(new Date(), DateUtils.PATTERN_yyyy_MM_dd_HH_mm_ss));
  265. promoCode.setAdminId(adminId);
  266. promoCode.setUseBy(clentId);
  267. promoCode.setIsUse("1");
  268. promoCode.setModifyDate(new Date());
  269. promoCodeService.updateById(promoCode);
  270. }
  271. }
  272. @GetMapping("/getYHJList")
  273. public ResponseModel<?> getYHJList(Long wxId) {
  274. if (wxId == null) {
  275. return R.fail(A0001);
  276. }
  277. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  278. query.eq(TPromoCode::getWxId, wxId);
  279. query.eq(TPromoCode::getIsUse, "0");
  280. query.eq(TPromoCode::getType, "2");
  281. List<TPromoCode> list = promoCodeService.list(query);
  282. return R.ok(list, "成功");
  283. }
  284. @GetMapping("/checkYHJList")
  285. public ResponseModel<?> checkYHJList(Long wxId) {
  286. if (wxId == null) {
  287. return R.fail(A0001);
  288. }
  289. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  290. query.eq(TPromoCode::getWxId, wxId);
  291. query.eq(TPromoCode::getType, "2");
  292. List<TPromoCode> list = promoCodeService.list(query);
  293. if (list.size() > 0) {
  294. return R.ok(true, "已领取");
  295. } else {
  296. return R.ok(false, "未领取");
  297. }
  298. }
  299. @GetMapping("/getYHJ")
  300. public ResponseModel<TPromoCode> getYHJ(@RequestParam String code) {
  301. if (StringUtils.isEmpty(code)) {
  302. return R.fail(A0001);
  303. }
  304. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  305. query.eq(TPromoCode::getCode, code);
  306. query.eq(TPromoCode::getType, "2");
  307. List<TPromoCode> list = promoCodeService.list(query);
  308. if (list.size() > 0) {
  309. TPromoCode tPromoCode = list.get(0);
  310. // String isUse = tPromoCode.getIsUse();
  311. // if(isUse.equals("1")){
  312. // return R.fail(A0002,"已使用");
  313. // }
  314. // Date lastUseDate = tPromoCode.getLastUseDate();
  315. // Date date = new Date();
  316. // if(date.getTime()>lastUseDate.getTime()){
  317. // return R.fail(A0002,"已过期");
  318. // }
  319. return R.ok(tPromoCode);
  320. }
  321. return R.ok();
  322. }
  323. @ApiOperation(value = "添加优惠码")
  324. @GetMapping("/add")
  325. public ResponseModel<?> add(String addMode, String codeNum,
  326. Long adminId, int number, Float discount,
  327. Float month, String type, String frpCode) {
  328. if (discount == null || month == null || adminId == null || StringUtils.isEmpty(type)) {
  329. return R.fail(A0001);
  330. }
  331. if (addMode.equals("2")) {
  332. if (StringUtils.isEmpty(codeNum)) {
  333. return R.fail(A0001, "优惠码为空");
  334. }
  335. LambdaQueryWrapper<TPromoCode> query = new LambdaQueryWrapper<>();
  336. query.eq(TPromoCode::getCode, codeNum);
  337. query.eq(TPromoCode::getIsUse, 0);
  338. query.eq(TPromoCode::getAdminId, adminId);
  339. TPromoCode tPromoCode = promoCodeService.getOne(query);
  340. if (tPromoCode != null) {
  341. // 已存在优惠码
  342. return R.fail(A0002, "已存在该优惠码");
  343. }
  344. }
  345. TAdmin admin = adminService.getById(adminId);
  346. // LocalDateTime now = LocalDateTime.now();
  347. // LocalDateTime startOfToday = now.with(LocalTime.MIN);
  348. // LocalDateTime endOfToday = now.with(LocalTime.MAX);
  349. // LambdaQueryWrapper<TPromoCode> promoCodeWrapper = new LambdaQueryWrapper<>();
  350. // promoCodeWrapper.between(TPromoCode::getCreateDate, startOfToday, endOfToday);
  351. // promoCodeWrapper.eq(TPromoCode::getAdminId, adminId);
  352. // int countCode = promoCodeService.count(promoCodeWrapper);
  353. // if (countCode >= 5000) {
  354. // return R.fail(B0002, "今日优惠码申请已达到上限");
  355. // }
  356. if (String.valueOf(admin.getId()).equals("1")) {
  357. List<String> codes = new ArrayList<>();
  358. if (addMode.equals("1")) {
  359. for (int i = 0; i < number; i++) {
  360. StringBuilder num = new StringBuilder();
  361. Random random = new Random();
  362. for (int j = 0; j < 8; j++) {
  363. num.append(random.nextInt(10));
  364. }
  365. String code = num.toString();
  366. codes.add(code);
  367. }
  368. } else {
  369. codes.add(codeNum);
  370. }
  371. Long data = Calendar.getInstance().getTimeInMillis();
  372. Float mm = month * 10;
  373. String m = String.valueOf(mm);
  374. String[] split = m.split("\\.");
  375. String th = split[0];
  376. Long t = Long.valueOf(th);
  377. Long mon = 30L * 24L * 60L * 60L * 1000L;
  378. Long d = data + ((mon * t) / 10);
  379. Date lastUseDate = new Date(Long.parseLong(String.valueOf(d)));
  380. for (String code : codes) {
  381. TPromoCode promoCode = new TPromoCode();
  382. promoCode.setCode(Long.parseLong(code));
  383. promoCode.setAdminId(String.valueOf(admin.getId()));
  384. promoCode.setIsUse("0");
  385. promoCode.setDiscount(discount);
  386. promoCode.setLastUseDate(lastUseDate);
  387. promoCode.setCreateDate(new Date());
  388. promoCode.setUserName(admin.getUsername());
  389. promoCode.setType(type);
  390. promoCodeService.save(promoCode);
  391. }
  392. return R.ok(null, "生成成功");
  393. }
  394. LambdaQueryWrapper<TEquipment> query = Wrappers.lambdaQuery();
  395. query.eq(TEquipment::getAdminId, admin.getId());
  396. List<TEquipment> list = tEquipmentService.list(query);
  397. TEquipment equipment = list.get(0);
  398. if (list.size() == 0) {
  399. // "要有机器才能购买优惠码";
  400. return R.fail("没有机器");
  401. }
  402. //0折优惠码需要支付
  403. if (type.equals("0") && discount == 0) {
  404. if (admin.getPromoCodeOpen() == null || admin.getPromoCodeOpen().equals("1")) {
  405. Long equipmentId = equipment.getId();
  406. Double money = null;
  407. LambdaQueryWrapper<TPrice> query1 = Wrappers.lambdaQuery();
  408. query1.eq(TPrice::getName, "优惠码");
  409. List<TPrice> prices = priceService.list(query1);
  410. for (TPrice price : prices) {
  411. if (price.getName().equals("优惠码")) {
  412. money = price.getPrice();
  413. }
  414. }
  415. Long client6 = Long.parseLong(equipment.getClientId().substring(equipment.getClientId().length() - 6));
  416. String sn = orderService.initSn(client6);
  417. String orderNo = sn;
  418. String productName = "优惠码";
  419. BigDecimal price = BigDecimal.valueOf(money * number);
  420. BigDecimal cutPrice = price.multiply(new BigDecimal(100)).divide(new BigDecimal(100));
  421. BigDecimal amount = cutPrice;
  422. String productDesc = "";
  423. String commonParameter = "";
  424. if (addMode.equals("2") && StringUtils.isNotEmpty(codeNum)) {
  425. commonParameter = codeNum;
  426. }
  427. String returnUrl = null;
  428. String notifyUrl = JoinpayConstant.Notify_Url_PromoCode;
  429. String isShowPic = "1";
  430. String openId = null;
  431. String authCode = null;
  432. String appid = null;
  433. String transactionModel = null;
  434. String tradeMerchantNo = null;
  435. String buyerId = null;
  436. String isAlt = "0";
  437. String altType = null;
  438. String altUrl = null;
  439. BigDecimal marketingAmount = null;
  440. // frpCode = "WEIXIN_NATIVE";
  441. TOrder order1 = new TOrder();
  442. order1.setId(IDGenerator.orderID());
  443. order1.setSn(sn);
  444. order1.setType(0);
  445. order1.setAdminId(admin.getId());
  446. order1.setProductName(productName);
  447. order1.setPrice(price);
  448. order1.setEquipmentId(equipmentId);
  449. order1.setClientId(equipment.getClientId());
  450. order1.setStatus(0);
  451. order1.setFrpCode(frpCode);
  452. order1.setRefundTrxNo(String.valueOf(month));
  453. order1.setCreateDate(new Date());
  454. order1.setModifyDate(new Date());
  455. payFeign.addOrder(order1);
  456. JSONArray altInfo = new JSONArray();
  457. String result = null;
  458. try {
  459. result = joinpayMchService.uniPay(
  460. orderNo, amount, productName, productDesc,
  461. commonParameter, returnUrl, notifyUrl, frpCode,
  462. isShowPic, openId, authCode, appid, transactionModel, tradeMerchantNo,
  463. buyerId, isAlt, altType, altInfo, altUrl, marketingAmount
  464. );
  465. } catch (Exception e) {
  466. e.printStackTrace();
  467. return R.fail("申请支付失败");
  468. }
  469. JSONObject resultJson = JSONObject.parseObject(result);
  470. // 汇聚支付支付申请返回支付二维码图片
  471. String rd_Pic = resultJson.getString("rd_Pic");
  472. if (resultJson == null || StringUtils.isBlank(rd_Pic)) {
  473. return R.fail("找不到支付图片");
  474. }
  475. JSONObject kindData = new JSONObject();
  476. kindData.put("sn", sn);
  477. kindData.put("price", price);
  478. kindData.put("image", rd_Pic);
  479. return R.ok(kindData);
  480. }
  481. }
  482. Long client6 = Long.parseLong(equipment.getClientId().substring(equipment.getClientId().length() - 6));
  483. List<String> codes = new ArrayList<>();
  484. if (addMode.equals("1")) {
  485. for (int i = 0; i < number; i++) {
  486. String code = orderService.initSn(client6);
  487. StringBuffer str = new StringBuffer();
  488. str.append(code.substring(0, 6));
  489. codes.add(str.toString());
  490. }
  491. } else {
  492. codes.add(codeNum);
  493. }
  494. Long data = Calendar.getInstance().getTimeInMillis();
  495. Float mm = month * 10;
  496. String m = String.valueOf(mm);
  497. String[] split = m.split("\\.");
  498. String th = split[0];
  499. Long t = Long.valueOf(th);
  500. Long mon = 30L * 24L * 60L * 60L * 1000L;
  501. Long d = data + ((mon * t) / 10);
  502. Date lastUseDate = new Date(Long.parseLong(String.valueOf(d)));
  503. for (String code : codes) {
  504. TPromoCode promoCode = new TPromoCode();
  505. promoCode.setCode(Long.parseLong(code));
  506. promoCode.setAdminId(String.valueOf(admin.getId()));
  507. promoCode.setIsUse("0");
  508. promoCode.setDiscount(discount);
  509. promoCode.setLastUseDate(lastUseDate);
  510. promoCode.setCreateDate(new Date());
  511. promoCode.setUserName(admin.getUsername());
  512. promoCode.setType(type);
  513. promoCodeService.save(promoCode);
  514. }
  515. return R.ok("生成成功");
  516. }
  517. @ApiOperation(value = "获取优惠码列表")
  518. @GetMapping("/pagePromoCode")
  519. public ResponseModel<IPage<?>> pagePromoCode(String adminId, String code, String isUse, Date useStratDate, Date useEndDate, Date createStratDate, Date createEndDate, long current, long size) {
  520. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  521. if (StringUtils.isNotEmpty(adminId)) {
  522. TAdmin admin = adminService.getById(adminId);
  523. if (admin.getType() != 0 && admin.getType() != 1) {
  524. query.eq(TPromoCode::getAdminId, adminId);
  525. }
  526. if (admin.getId() == 2738) {
  527. LambdaQueryWrapper<TAdmin> adminQuery = Wrappers.lambdaQuery();
  528. adminQuery.isNull(TAdmin::getCompanyType).or()
  529. .eq(TAdmin::getCompanyType, "0");
  530. List<Long> adminIds = adminService.list(adminQuery).stream()
  531. .map(TAdmin::getId).collect(Collectors.toList());
  532. query.in(TPromoCode::getAdminId, adminIds);
  533. }
  534. // 七云平台管理员
  535. if (admin.getId() == 2739) {
  536. LambdaQueryWrapper<TAdmin> adminQuery = Wrappers.lambdaQuery();
  537. adminQuery.eq(TAdmin::getCompanyType, "1");
  538. List<Long> adminIds = adminService.list(adminQuery).stream()
  539. .map(TAdmin::getId).collect(Collectors.toList());
  540. query.in(TPromoCode::getAdminId, adminIds);
  541. }
  542. } else {
  543. return R.fail(A0001);
  544. }
  545. if (StringUtils.isNotEmpty(code)) {
  546. query.eq(TPromoCode::getCode, code);
  547. }
  548. if (StringUtils.isNotEmpty(isUse)) {
  549. query.like(TPromoCode::getIsUse, isUse);
  550. }
  551. if (useStratDate != null && useEndDate != null) {
  552. query.gt(TPromoCode::getUseDate, useStratDate);
  553. query.lt(TPromoCode::getUseDate, useEndDate);
  554. }
  555. if (createStratDate != null && createEndDate != null) {
  556. query.gt(TPromoCode::getCreateDate, createStratDate);
  557. query.lt(TPromoCode::getCreateDate, createEndDate);
  558. }
  559. query.orderByDesc(TPromoCode::getCreateDate);
  560. Page<TPromoCode> page = new Page<>(current, size, true);
  561. IPage<TPromoCode> iPage = promoCodeService.page(page, query);
  562. return R.ok(iPage);
  563. }
  564. @ApiOperation(value = "导出优惠码列表")
  565. @GetMapping("/export")
  566. public Object export(HttpServletResponse response, String adminId, String code, String isUse, Date useStratDate, Date useEndDate, Date createStratDate, Date createEndDate, long current, long size) {
  567. LambdaQueryWrapper<TPromoCode> query = Wrappers.lambdaQuery();
  568. String ifForeign = "0";
  569. String head = "优惠码数据-";
  570. if (StringUtils.isNotEmpty(adminId)) {
  571. TAdmin admin = adminService.getById(adminId);
  572. if (StringUtils.isNotEmpty(admin.getIfForeign()) && admin.getIfForeign().equals("1")) {
  573. ifForeign = admin.getIfForeign();
  574. head = "PromoCode-";
  575. }
  576. if (admin.getType() != 0 && admin.getType() != 1) {
  577. query.eq(TPromoCode::getAdminId, adminId);
  578. }
  579. // 申泽平台管理员
  580. if (admin.getId() == 2738) {
  581. LambdaQueryWrapper<TAdmin> adminQuery = Wrappers.lambdaQuery();
  582. adminQuery.isNull(TAdmin::getCompanyType).or()
  583. .eq(TAdmin::getCompanyType, "0");
  584. List<Long> adminIds = adminService.list(adminQuery).stream()
  585. .map(TAdmin::getId).collect(Collectors.toList());
  586. query.in(TPromoCode::getAdminId, adminIds);
  587. }
  588. // 七云平台管理员
  589. if (admin.getId() == 2739) {
  590. LambdaQueryWrapper<TAdmin> adminQuery = Wrappers.lambdaQuery();
  591. adminQuery.eq(TAdmin::getCompanyType, "1");
  592. List<Long> adminIds = adminService.list(adminQuery).stream()
  593. .map(TAdmin::getId).collect(Collectors.toList());
  594. query.in(TPromoCode::getAdminId, adminIds);
  595. }
  596. } else {
  597. return R.fail(A0001);
  598. }
  599. if (StringUtils.isNotEmpty(code)) {
  600. query.eq(TPromoCode::getCode, code);
  601. }
  602. if (StringUtils.isNotEmpty(isUse)) {
  603. query.like(TPromoCode::getIsUse, isUse);
  604. }
  605. if (useStratDate != null && useEndDate != null) {
  606. query.gt(TPromoCode::getUseDate, useStratDate);
  607. query.lt(TPromoCode::getUseDate, useEndDate);
  608. }
  609. if (createStratDate != null && createEndDate != null) {
  610. query.gt(TPromoCode::getCreateDate, createStratDate);
  611. query.lt(TPromoCode::getCreateDate, createEndDate);
  612. }
  613. Page<TPromoCode> page = new Page<>(current, size, true);
  614. IPage<TPromoCode> iPage = promoCodeService.page(page, query);
  615. List<TPromoCode> list = iPage.getRecords();
  616. Workbook workbook = null;
  617. if (ifForeign.equals("0")) {
  618. List<PromoCodeTarget> promoCodeTargets = promoCodeService.findById(list);
  619. ExportParams exportParams = new ExportParams("优惠码记录", "sheet1");
  620. workbook = ExcelExportUtil.exportExcel(exportParams, PromoCodeTarget.class, promoCodeTargets);
  621. } else {
  622. List<PromoCodeTargetEn> promoCodeTargetEns = promoCodeService.exportPromoCode(list);
  623. ExportParams exportParams = new ExportParams("Promo code record", "sheet1");
  624. workbook = ExcelExportUtil.exportExcel(exportParams, PromoCodeTargetEn.class, promoCodeTargetEns);
  625. }
  626. if (workbook != null) {
  627. OutputStream os = null;
  628. try {
  629. os = response.getOutputStream();
  630. SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmmss");
  631. response.setContentType("application/vnd.ms-excel;charset=utf-8");
  632. response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(head + format.format(new Date()) + ".xls", "UTF-8"));
  633. workbook.write(os);
  634. return JsonMessage.success("导出成功");
  635. } catch (Exception e) {
  636. e.printStackTrace();
  637. return JsonMessage.success("导出错误");
  638. } finally {
  639. try {
  640. os.close();
  641. workbook.close();
  642. } catch (IOException e) {
  643. e.printStackTrace();
  644. }
  645. }
  646. }
  647. return JsonMessage.success("导出成功");
  648. }
  649. @ApiOperation(value = "查询优惠码几折")
  650. @RequestMapping(value = "/getCodeDiscount", method = RequestMethod.GET)
  651. @ResponseBody
  652. public Object getCodeDiscount(String code) {
  653. if (!StringUtils.isEmpty(code)) {
  654. LambdaQueryWrapper<TPromoCode> wrapper = new LambdaQueryWrapper<>();
  655. wrapper.in(TPromoCode::getCode, code);
  656. wrapper.in(TPromoCode::getIsUse, "0");
  657. wrapper.orderByDesc(TPromoCode::getCreateDate);
  658. List<TPromoCode> list = promoCodeService.list(wrapper);
  659. if (list == null) {
  660. return null;
  661. }
  662. TPromoCode promoCode = list.get(0);
  663. Float discount = promoCode.getDiscount();
  664. String type = promoCode.getType();
  665. JSONObject kindData = new JSONObject();
  666. kindData.put("type", type.equals("1") ? type : "0");
  667. kindData.put("discount", discount);
  668. return JsonMessage.success(kindData.toString());
  669. } else {
  670. return null;
  671. }
  672. }
  673. @ApiOperation(value = "批量删除优惠码")
  674. @PostMapping(value = "/deletes")
  675. @ResponseBody
  676. public ResponseModel<?> deletes(@RequestBody Long[] ids) {
  677. if (ids == null) {
  678. return R.fail(ResponseCodesEnum.A0001);
  679. }
  680. if (ids.length > 0) {
  681. ArrayList<Long> idList = new ArrayList<>();
  682. for (int i = 0; i < ids.length; i++) {
  683. idList.add(ids[i]);
  684. }
  685. promoCodeService.removeByIds(idList);
  686. }
  687. return R.ok();
  688. }
  689. }