PAS/src/test/java/cn/cslg/pas/service/ClaimTextTests.java

package cn.cslg.pas.service;


import cn.cslg.pas.common.dto.PatentColumnDTO;
import cn.cslg.pas.common.dto.PatentDTO;
import cn.cslg.pas.common.model.request.StringRequest;
import cn.cslg.pas.common.utils.ClaimUtils.ClaimSplitUtils;
import cn.cslg.pas.common.vo.PatentRightParams;
import cn.cslg.pas.common.vo.RePatentClaim;
import cn.cslg.pas.domain.es.Text;
import cn.cslg.pas.service.business.CommonService;
import cn.cslg.pas.service.business.es.EsService;
import cn.cslg.pas.service.common.TranslateService;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.thymeleaf.templateparser.text.TextTemplateParser;

import java.util.ArrayList;
import java.util.List;

@SpringBootTest
public class ClaimTextTests {
    @Autowired
    private EsService esService;
    @Autowired
    private TranslateService translateService;

    @Test
    public void test() throws Exception {
        String json = CommonService.readFile("C:\\Users\\admin\\Desktop\\patentClaim\\SKE23254T3.json");
        JSONObject sqlObjects = JSON.parseObject(json, JSONObject.class);

        String pn = sqlObjects.get("PN").toString();
        String content = sqlObjects.get("content").toString();

        PatentRightParams rightParams = new PatentRightParams();
        rightParams.setPatentNo(pn);
        rightParams.setContent(content);
        List<RePatentClaim> rePatentClaims = ClaimSplitUtils.formatPatentRight(rightParams);
        StringBuilder stringBuilder = new StringBuilder();
        rePatentClaims.forEach(item -> stringBuilder.append(item.getContent()));

        if (stringBuilder.toString().equals(content)) {
            System.out.println(true);
        }

        System.out.println(pn);
    }


    @Test
    public void testTotal() throws Exception {
        StringRequest stringRequest = new StringRequest();
        stringRequest.setProjectId(323);
        stringRequest.setCurrent(1L);
        stringRequest.setSize(10L);
        PatentDTO patentDTO = esService.esSearch(stringRequest);
        List<PatentColumnDTO> patentColumnDTOS = patentDTO.getPatents();

        for (PatentColumnDTO patentColumnDTO : patentColumnDTOS) {
            String pn = patentColumnDTO.getPatentNo();
            List<Text> claims = patentColumnDTO.getClaim();
            if (claims == null || claims.size() == 0) {
                continue;
            }
            Text text = claims.stream().filter(item -> item.getIfOrigin().equals(true)).findFirst().orElse(null);
            if (text == null) {
                continue;
            }
            PatentRightParams rightParams = new PatentRightParams();
            String content = text.getTextContent();
            rightParams.setPatentNo(pn);
            rightParams.setContent(content);
            List<RePatentClaim> rePatentClaims = ClaimSplitUtils.formatPatentRight(rightParams);
            StringBuilder stringBuilder = new StringBuilder();
            rePatentClaims.forEach(item -> stringBuilder.append(item.getContent()));

            if (stringBuilder.toString().equals(content)) {
                System.out.println(true);
            }

            System.out.println(pn);
        }


    }

    @Test
    public void testTranslate() {
        try {
            String a = translateService.getTranslateContent("where are you from");
            System.out.println(a);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}