KuaishouLiveBroadcastServiceImpl.java 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317
  1. package com.ruixuan.live.service.impl;
  2. import com.alibaba.fastjson.JSON;
  3. import com.alibaba.fastjson.JSONArray;
  4. import com.alibaba.fastjson.JSONObject;
  5. import com.ruixuan.common.core.domain.ResultResponse;
  6. import com.ruixuan.common.utils.Check;
  7. import com.ruixuan.data.service.IKuaishouLiveDataAnalysisService;
  8. import com.ruixuan.data.service.IKuaishouLiveDataFunnelService;
  9. import com.ruixuan.data.service.IKuaishouLiveDataKanbanService;
  10. import com.ruixuan.data.service.IKuaishouLiveDataOverviewService;
  11. import com.ruixuan.data.utils.LiveDataExcelUtils;
  12. import com.ruixuan.live.entity.KuaishouLiveBroadcast;
  13. import com.ruixuan.live.entity.KuaishouLiveBroadcastFiles;
  14. import com.ruixuan.live.mapper.KuaishouLiveBroadcastFilesMapper;
  15. import com.ruixuan.live.mapper.KuaishouLiveBroadcastMapper;
  16. import com.ruixuan.live.service.IKuaishouLiveBroadcastService;
  17. import com.ruixuan.system.service.ISysUserService;
  18. import org.springframework.beans.BeanUtils;
  19. import org.springframework.beans.factory.annotation.Autowired;
  20. import org.springframework.stereotype.Service;
  21. import org.springframework.web.multipart.MultipartFile;
  22. import java.util.Arrays;
  23. import java.util.HashSet;
  24. import java.util.List;
  25. import java.util.UUID;
  26. /**
  27. * 快手—直播Service业务层处理
  28. *
  29. * @author ruoyi
  30. * @date 2022-05-31
  31. */
  32. @Service
  33. public class KuaishouLiveBroadcastServiceImpl implements IKuaishouLiveBroadcastService {
  34. @Autowired
  35. private KuaishouLiveBroadcastMapper kuaishouLiveBroadcastMapper;
  36. @Autowired
  37. private KuaishouLiveBroadcastFilesMapper filesMapper;
  38. @Autowired
  39. private ISysUserService sysUserService;
  40. @Autowired
  41. private IKuaishouLiveDataAnalysisService analysisService;
  42. @Autowired
  43. private IKuaishouLiveDataFunnelService funnelService;
  44. @Autowired
  45. private IKuaishouLiveDataKanbanService kanbanService;
  46. @Autowired
  47. private IKuaishouLiveDataOverviewService overviewService;
  48. /**
  49. * 查询快手—直播
  50. *
  51. * @param id 快手—直播主键
  52. * @return 快手—直播
  53. */
  54. @Override
  55. public ResultResponse selectKuaishouLiveBroadcastById(String id) {
  56. KuaishouLiveBroadcast liveBroadcast = kuaishouLiveBroadcastMapper.selectKuaishouLiveBroadcastById(id);
  57. return ResultResponse.success(liveBroadcast);
  58. }
  59. @Override
  60. public ResultResponse getForEdit(String id) {
  61. KuaishouLiveBroadcast liveBroadcast = kuaishouLiveBroadcastMapper.selectKuaishouLiveBroadcastById(id);
  62. JSONObject object = JSONObject.parseObject(JSON.toJSONString(liveBroadcast));
  63. object.put("traderArray", getUsers(liveBroadcast.getTraderId()));
  64. object.put("centralControlArray", getUsers(liveBroadcast.getCentralControlId()));
  65. object.put("fieldControlArray", getUsers(liveBroadcast.getFieldControlId()));
  66. object.put("contentControlArray", getUsers(liveBroadcast.getContentControlId()));
  67. object.put("inputControlArray", getUsers(liveBroadcast.getInputControlId()));
  68. object.put("anchorArray", getUsers(liveBroadcast.getAnchorId()));
  69. return ResultResponse.success(object);
  70. }
  71. private List<JSONObject> getUsers(String ids) {
  72. if (ids != null && ids != "-") {
  73. ids = ids.replace(",", ",");
  74. List<String> userIds = Arrays.asList(ids.split(","));
  75. return sysUserService.queryUserByIds(userIds);
  76. }
  77. return null;
  78. }
  79. /**
  80. * 查询快手—直播列表
  81. *
  82. * @param kuaishouLiveBroadcast 快手—直播
  83. * @return 快手—直播
  84. */
  85. @Override
  86. public List<KuaishouLiveBroadcast> selectKuaishouLiveBroadcastList(KuaishouLiveBroadcast kuaishouLiveBroadcast) {
  87. return kuaishouLiveBroadcastMapper.selectKuaishouLiveBroadcastList(kuaishouLiveBroadcast);
  88. }
  89. /**
  90. * 新增快手—直播
  91. *
  92. * @param kuaishouLiveBroadcast 快手—直播
  93. * @return 结果
  94. */
  95. @Override
  96. public ResultResponse insertKuaishouLiveBroadcast(KuaishouLiveBroadcast broadcast) {
  97. KuaishouLiveBroadcast one = kuaishouLiveBroadcastMapper.getOne(broadcast);
  98. if (one != null) {
  99. return ResultResponse.error("当前时间段已存在直播!");
  100. }
  101. broadcast.setId(UUID.randomUUID().toString().replace("-", ""));
  102. int num = kuaishouLiveBroadcastMapper.insertKuaishouLiveBroadcast(broadcast);
  103. if (num > 0) {
  104. return ResultResponse.success();
  105. }
  106. return ResultResponse.error("fail");
  107. }
  108. /**
  109. * 修改快手—直播
  110. *
  111. * @param kuaishouLiveBroadcast 快手—直播
  112. * @return 结果
  113. */
  114. @Override
  115. public ResultResponse updateKuaishouLiveBroadcast(KuaishouLiveBroadcast broadcast) {
  116. KuaishouLiveBroadcast one = kuaishouLiveBroadcastMapper.selectKuaishouLiveBroadcastById(broadcast.getId());
  117. if (one == null) {
  118. return ResultResponse.error("未查询到直播!");
  119. }
  120. KuaishouLiveBroadcast checkOne = kuaishouLiveBroadcastMapper.getOne(broadcast);
  121. if (checkOne == null) {
  122. KuaishouLiveBroadcast copy = new KuaishouLiveBroadcast();
  123. BeanUtils.copyProperties(copy, broadcast);
  124. copy.setId(null);
  125. if (kuaishouLiveBroadcastMapper.getOne(copy) != null) {
  126. return ResultResponse.error("当前时间段已存在直播!");
  127. }
  128. }
  129. int i = kuaishouLiveBroadcastMapper.updateKuaishouLiveBroadcast(broadcast);
  130. if (i > 0) {
  131. return ResultResponse.success();
  132. }
  133. return ResultResponse.error("fail");
  134. }
  135. /**
  136. * 批量删除快手—直播
  137. *
  138. * @param ids 需要删除的快手—直播主键
  139. * @return 结果
  140. */
  141. @Override
  142. public int deleteKuaishouLiveBroadcastByIds(String[] ids) {
  143. return kuaishouLiveBroadcastMapper.deleteKuaishouLiveBroadcastByIds(ids);
  144. }
  145. /**
  146. * 删除快手—直播信息
  147. *
  148. * @param id 快手—直播主键
  149. * @return 结果
  150. */
  151. @Override
  152. public int deleteKuaishouLiveBroadcastById(String id) {
  153. return kuaishouLiveBroadcastMapper.deleteKuaishouLiveBroadcastById(id);
  154. }
  155. @Override
  156. public ResultResponse dataUpload(String id, String actualGmv, String dapanUrls, String portraitUrls, MultipartFile[] files) throws Exception {
  157. KuaishouLiveBroadcast broadcast = kuaishouLiveBroadcastMapper.selectKuaishouLiveBroadcastById(id);
  158. if (broadcast == null) {
  159. return ResultResponse.error("未查询到直播!");
  160. }
  161. // 目前支持 直播数据看板、直播数据概览、直播漏斗数据、商品分析数据 四种数据导入
  162. String msg = "";
  163. //记录上传文件
  164. int count = 0;
  165. if (!(Check.isNull(files) || "".equals(files[0].getOriginalFilename()))) {
  166. for (MultipartFile file : files) {
  167. String fileName = file.getOriginalFilename();
  168. if (!(fileName.endsWith(".xls") || fileName.endsWith(".xlsx") || fileName.endsWith(".csv"))) {
  169. msg += fileName + "、";
  170. continue;
  171. }
  172. List<JSONObject> list = LiveDataExcelUtils.analysisFile(file, fileName);
  173. if (Check.isNotNull(list)) {
  174. if (fileName.contains("看板")) {
  175. count += kanbanService.inserExcelInfo(broadcast, list);
  176. } else if (fileName.contains("概览")) {
  177. count += overviewService.inserExcelInfo(broadcast, list);
  178. } else if (fileName.contains("漏斗")) {
  179. count += funnelService.inserExcelInfo(broadcast, list);
  180. } else if (fileName.contains("分析")) {
  181. count += analysisService.inserExcelInfo(broadcast, list);
  182. }
  183. }
  184. }
  185. }
  186. //上传到cos的文件夹名称 (操盘手)
  187. String upFileName = "CaoPanShou";
  188. KuaishouLiveBroadcastFiles file = new KuaishouLiveBroadcastFiles();
  189. //上传数据大盘
  190. if (Check.isNotNull(dapanUrls) && !"[]".equals(dapanUrls)) {
  191. List<String> urls = JSONArray.parseArray(dapanUrls, String.class);
  192. HashSet<String> set = new HashSet<>(urls);
  193. String urlFix = "";
  194. String names = "";
  195. int dapanCount = set.size();
  196. for (String url : set) {
  197. // String pictureName = "dapan_000" + (dapanCount + 1);
  198. // String cosUrl = CosUpload.getJPGCosUrl(url, upFileName, pictureName);
  199. if (Check.isNotNull(url)) {
  200. int lastNo = url.lastIndexOf("/") + 1;
  201. urlFix = "https:" + url.substring(0, lastNo);
  202. String name = url.substring(lastNo);
  203. names += "," + name;
  204. }
  205. }
  206. file.setUpCommonUrl(urlFix);
  207. file.setDapanCount(dapanCount);
  208. file.setDapanNames(names.substring(1));
  209. file.setDapanMsg("success");
  210. }
  211. //画像分析上传
  212. if (Check.isNotNull(portraitUrls) && !"[]".equals(portraitUrls)) {
  213. List<String> urls = JSONArray.parseArray(portraitUrls, String.class);
  214. HashSet<String> set = new HashSet<>(urls);
  215. String urlFix = "";
  216. String names = "";
  217. int portraitCount = set.size();
  218. for (String url : set) {
  219. // String pictureName = "portrait_000" + (portraitCount + 1);
  220. // String cosUrl = CosUpload.getJPGCosUrl(url, upFileName, pictureName);
  221. if (Check.isNotNull(url)) {
  222. int lastNo = url.lastIndexOf("/") + 1;
  223. urlFix = "https:" + url.substring(0, lastNo);
  224. String name = url.substring(lastNo);
  225. names += "," + name;
  226. }
  227. }
  228. file.setUpCommonUrl(urlFix);
  229. file.setPortraitCount(portraitCount);
  230. file.setPortraitNames(names.substring(1));
  231. file.setPortraitMsg("success");
  232. }
  233. KuaishouLiveBroadcastFiles one = filesMapper.selectKuaishouLiveBroadcastFilesById(id);
  234. file.setId(id);
  235. if (Check.isNull(one)) {
  236. filesMapper.insertKuaishouLiveBroadcastFiles(file);
  237. } else {
  238. filesMapper.updateKuaishouLiveBroadcastFiles(file);
  239. }
  240. //更新直播表
  241. KuaishouLiveBroadcast updateEntity = new KuaishouLiveBroadcast();
  242. updateEntity.setId(id);
  243. updateEntity.setUploadFileCount(count);
  244. int dapanCount = Check.isNotNull(file.getDapanCount()) ? file.getDapanCount() : 0;
  245. int portraitCount = Check.isNotNull(file.getPortraitCount()) ? file.getPortraitCount() : 0;
  246. updateEntity.setUploadPictureCount(dapanCount + portraitCount);
  247. updateEntity.setActualGmv(actualGmv);
  248. kuaishouLiveBroadcastMapper.updateKuaishouLiveBroadcast(updateEntity);
  249. if (msg.length() > 0) {
  250. msg = msg.substring(0, msg.length() - 1) + "不支持文件格式";
  251. } else {
  252. msg = "上传完成";
  253. }
  254. return ResultResponse.successMsg(msg, null);
  255. }
  256. @Override
  257. public ResultResponse replay(String id, String replayPictureUrls, String replayFile) {
  258. //上传到cos的文件夹名称 (操盘手)
  259. String upFileName = "CaoPanShou";
  260. KuaishouLiveBroadcastFiles file = new KuaishouLiveBroadcastFiles();
  261. //上传复盘图片
  262. if (Check.isNotNull(replayPictureUrls) && !"[]".equals(replayPictureUrls)) {
  263. List<String> urls = JSONArray.parseArray(replayPictureUrls, String.class);
  264. HashSet<String> set = new HashSet<>(urls);
  265. String urlFix = "";
  266. String names = "";
  267. int replayPictureCount = set.size();
  268. for (String url : set) {
  269. // String pictureName = "replayPicture_000" + (replayPictureCount + 1);
  270. // String cosUrl = CosUpload.getJPGCosUrl(url, upFileName, pictureName);
  271. if (Check.isNotNull(url)) {
  272. int lastNo = url.lastIndexOf("/") + 1;
  273. urlFix = "https:" + url.substring(0, lastNo);
  274. String name = url.substring(lastNo);
  275. names += "," + name;
  276. }
  277. }
  278. file.setReplayCommonUrl(urlFix);
  279. file.setReplayPictureCount(replayPictureCount);
  280. file.setReplayPictureNames(names.substring(1));
  281. file.setReplayPictureMsg("success");
  282. }
  283. if (Check.isNotNull(replayFile)) {
  284. file.setReplayFileUrl(replayFile);
  285. }
  286. KuaishouLiveBroadcastFiles one = filesMapper.selectKuaishouLiveBroadcastFilesById(id);
  287. file.setId(id);
  288. if (Check.isNull(one)) {
  289. filesMapper.insertKuaishouLiveBroadcastFiles(file);
  290. } else {
  291. filesMapper.updateKuaishouLiveBroadcastFiles(file);
  292. }
  293. return ResultResponse.success();
  294. }
  295. }