EventServiceTests.java 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200
  1. package cn.cslg.pas.service;
  2. import cn.cslg.pas.common.dto.PatentDTO;
  3. import cn.cslg.pas.common.dto.business.EsPatentFamilyDTO;
  4. import cn.cslg.pas.common.model.cronModel.Records;
  5. import cn.cslg.pas.common.model.request.*;
  6. import cn.cslg.pas.common.utils.Response;
  7. import cn.cslg.pas.controller.EventController;
  8. import cn.cslg.pas.controller.PatentController;
  9. import cn.cslg.pas.domain.es.FamilyPatent;
  10. import cn.cslg.pas.domain.es.Patent;
  11. import cn.cslg.pas.domain.es.PatentFamilyMessage;
  12. import cn.cslg.pas.service.business.ProductMarketDataService;
  13. import cn.cslg.pas.service.business.es.EsService;
  14. import co.elastic.clients.elasticsearch.core.DeleteRequest;
  15. import com.alibaba.fastjson.JSONObject;
  16. import org.apache.http.entity.ContentType;
  17. import org.junit.Before;
  18. import org.junit.jupiter.api.Test;
  19. import org.springframework.beans.factory.annotation.Autowired;
  20. import org.springframework.boot.test.context.SpringBootTest;
  21. import org.springframework.mock.web.MockMultipartFile;
  22. import org.springframework.web.multipart.MultipartFile;
  23. import java.io.File;
  24. import java.io.FileInputStream;
  25. import java.io.IOException;
  26. import java.util.*;
  27. /**
  28. * @author chenyu
  29. * @date 2023/9/6
  30. */
  31. @SpringBootTest
  32. public class EventServiceTests {
  33. @Autowired
  34. private EventController eventController;
  35. @Autowired
  36. PatentController patentController;
  37. @Autowired
  38. private EsService esService;
  39. @Autowired
  40. private ProductMarketDataService productMarketDataService;
  41. @Test
  42. void test() throws Exception {
  43. StringRequest queryRequest = new StringRequest();
  44. queryRequest.setSearchQuery("id=52");
  45. List<OrderDTO> dtos = new ArrayList<>();
  46. OrderDTO orderDTO1 = new OrderDTO();
  47. orderDTO1.setOrderBy("name");
  48. orderDTO1.setOrderType(0);
  49. OrderDTO orderDTO2 = new OrderDTO();
  50. orderDTO2.setOrderBy("createName");
  51. orderDTO2.setOrderType(1);
  52. dtos.add(orderDTO1);
  53. dtos.add(orderDTO2);
  54. queryRequest.setOrderDTOList(dtos);
  55. queryRequest.setCurrent(Long.parseLong("1"));
  56. queryRequest.setSize(Long.parseLong("10"));
  57. Response response = eventController.queryEvent(queryRequest);
  58. System.out.println(response); }
  59. @Test
  60. void add() throws Exception {
  61. JSONObject jsonObj = new JSONObject();
  62. jsonObj.put("name", "事件4");
  63. jsonObj.put("description", "描述1");
  64. jsonObj.put("clientId", 1);
  65. jsonObj.put("scenarioId", 2);
  66. jsonObj.put("eventDate","");
  67. String json_to_string = JSONObject.toJSONString(jsonObj);
  68. List<MultipartFile> list = new ArrayList<>();
  69. File file = new File("C:\\Users\\admin\\Desktop\\test.txt");
  70. MultipartFile mulFile = new MockMultipartFile(
  71. "th.jpg", //文件名
  72. "th.jpg", //originalName 相当于上传文件在客户机上的文件名
  73. ContentType.APPLICATION_OCTET_STREAM.toString(), //文件类型
  74. new FileInputStream(file) //文件流
  75. );
  76. list.add(mulFile);
  77. eventController.addEvent(json_to_string, list);
  78. }
  79. @Test
  80. void groupEvent() throws Exception {
  81. StringGroupRequest queryRequest = new StringGroupRequest();
  82. queryRequest.setSearchQuery("name~事件 and createName=朱");
  83. List<OrderDTO> dtos = new ArrayList<>();
  84. OrderDTO orderDTO1 = new OrderDTO();
  85. orderDTO1.setOrderBy("name");
  86. orderDTO1.setOrderType(0);
  87. OrderDTO orderDTO2 = new OrderDTO();
  88. orderDTO2.setOrderBy("createName");
  89. orderDTO2.setOrderType(1);
  90. dtos.add(orderDTO1);
  91. dtos.add(orderDTO2);
  92. queryRequest.setOrderDTOList(dtos);
  93. queryRequest.setCurrent(Long.parseLong("1"));
  94. queryRequest.setSize(Long.parseLong("10"));
  95. queryRequest.setGroupBy("name");
  96. eventController.groupEvent(queryRequest);
  97. }
  98. @Test
  99. void test1() throws Exception {
  100. StringGroupRequest groupRequest = new StringGroupRequest();
  101. groupRequest.setGroupBy("groupMonthTime");
  102. Records records=(Records) productMarketDataService.getGroup(groupRequest,"productMarketData");
  103. System.out.println(records);
  104. }
  105. @Test
  106. void test11() throws Exception {
  107. // List<Patent> list = esService.Search("publicDate=2009-03", 0, 50,null);
  108. // MapRequest mapRequest = new MapRequest();
  109. // Map<String, Object> map = new HashMap<>();
  110. // map.put("titleTextContent", "电子");
  111. // map.put("patentNo", "CN201199922Y");
  112. // mapRequest.setSearchQuery(map);
  113. // mapRequest.setCurrent(0L);
  114. // mapRequest.setSize(50L);
  115. // mapRequest.setProjectId(1);
  116. StringRequest stringRequest = new StringRequest();
  117. stringRequest.setSearchQuery("appCity=济南");
  118. // stringRequest.setSearchQuery("mipcLevel1=H");
  119. // stringRequest.setSearchQuery("agent=李桂存");
  120. // stringRequest.setSearchQuery("IN=郝旭东");
  121. // stringRequest.setSearchQuery("simpleStatus=3");
  122. // stringRequest.setSearchQuery("CL=一种考虑电流限制和出力水平的分布式光伏机电暂态建模方法");
  123. // stringRequest.setSearchQuery("AB=一种考虑电流限制和出力水平的分布式光伏机电暂态模型");
  124. // stringRequest.setSearchQuery("PN=CN114513005B");
  125. // stringRequest.setSearchQuery("AN=CN201810025278.2");
  126. // stringRequest.setSearchQuery("patentNo=CN");
  127. // stringRequest.setSearchQuery("PD<=200903");
  128. // stringRequest.setSearchQuery("patentNo=CN201199922Y and (simpleFamilyNum>1 or simpleFamilyNum=0)");
  129. stringRequest.setCurrent(1L);
  130. stringRequest.setSize(5L);
  131. // stringRequest.setProjectId(1);
  132. //
  133. PatentDTO patentDTO = esService.esSearch(stringRequest);
  134. System.out.println(patentDTO);
  135. }
  136. @Test
  137. void test12() throws IOException {
  138. Boolean bool = esService.searchPatent("YiQCGowBmB3pRkTj4NNG", 5);
  139. System.out.println(bool);
  140. }
  141. @Test
  142. void getpagetexst() throws IOException{
  143. //1.添加一批数据(10)
  144. //2.调用查询取第一页(5笔一页)
  145. //3.判断5笔数据是正确的;
  146. //4.删除测试数据
  147. List<Integer> list = Arrays.asList(1, 2, 3, 4);
  148. List<Integer> list1 = Arrays.asList(4, 5);
  149. // list1.removeAll(list);
  150. System.out.println(list1);
  151. }
  152. @Test
  153. void add1() throws Exception {
  154. List<FamilyPatent> list = new ArrayList<>();
  155. FamilyPatent patent =new FamilyPatent();
  156. patent.setAppNo("CN147258369");
  157. patent.setGrantNo("CN258369147");
  158. patent.setPublicNo("CN369258147");
  159. FamilyPatent patent1 =new FamilyPatent();
  160. patent1.setAppNo("US147258369");
  161. patent1.setGrantNo("US258369147");
  162. patent1.setPublicNo("US369258147");
  163. list.add(patent);
  164. list.add(patent1);
  165. PatentFamilyMessage patentFamilyMessage =new PatentFamilyMessage();
  166. patentFamilyMessage.setPatent(list);
  167. patentFamilyMessage.setFamilyType("test");
  168. String id = esService.addPatentFamily(patentFamilyMessage);
  169. System.out.println(id);
  170. }
  171. @Test
  172. void test3() throws IOException {
  173. List<String> stringList = Arrays.asList("US369258147", "US258369147", "14528");
  174. EsPatentFamilyDTO test = esService.addEsPatentFamily(stringList, "test");
  175. System.out.println(test);
  176. }
  177. @Test
  178. void test4() throws IOException {
  179. // List<String> stringList = Arrays.asList("US369258147", "US258369147", "14528");
  180. // EsPatentFamilyDTO test = esService.selectEsPatentFamily(stringList, "test");
  181. // System.out.println(test);
  182. }
  183. }