ttc_5.0/ttc-app/src/test/cn/ubitech/ttc/Test1.java

package cn.ubitech.ttc;

import cn.ubitech.ttc.common.SnowflakeIdWorker;
import cn.ubitech.ttc.dao.LabelIntelligentCommonInfoMapper;
import cn.ubitech.ttc.dao.ResourcelibraryMapper;
import cn.ubitech.ttc.entity.CompanyProvincialScienceSmall;
import cn.ubitech.ttc.entity.LabelIntelligentCommonInfo;
import cn.ubitech.ttc.entity.OrganizationInfo;
import cn.ubitech.ttc.entity.Resourcelibrary;
import cn.ubitech.ttc.model.common.Constant;
import cn.ubitech.ttc.model.common.ElasticSearchTTCModel;
import cn.ubitech.ttc.repository.ResourceRepository;
import cn.ubitech.ttc.service.mybaits.ICompanyProvincialScienceSmallService;
import cn.ubitech.ttc.service.mybaits.IOrganizationInfoService;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.plugins.Page;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;
import org.springframework.data.elasticsearch.core.query.NativeSearchQuery;
import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
import org.springframework.test.context.junit4.SpringRunner;

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.stream.Collectors;

/**
 * @author lys
 * @create 2021-10-11 11:38
 */
@RunWith(SpringRunner.class)
@SpringBootTest
@Slf4j
public class Test1 {

    @Autowired
    private IOrganizationInfoService infoService;
    @Autowired
    private ICompanyProvincialScienceSmallService companyProvincialScienceSmallService;
    @Autowired
    private ElasticsearchTemplate elasticsearchTemplate;
    @Autowired
    private ResourcelibraryMapper resourcelibraryMapper;
    @Autowired
    private LabelIntelligentCommonInfoMapper labelMapper;
    @Autowired
    private ResourceRepository resourceRepository;

    @Test
    public void test1() throws ParseException {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
        Calendar instance = Calendar.getInstance();
        instance.setTime(new Date());
        instance.add(Calendar.YEAR, -3);
        System.out.println(sdf.format(instance.getTime()));
    }

    @Test
    public void test2() {
        List<OrganizationInfo> list = new ArrayList<>();
        OrganizationInfo info = new OrganizationInfo();
        info.setName("aaaa");
        info.setRegistrationdate(new Date());
        list.add(info);
        infoService.insertOrUpdateBatch(list);
    }

    @Test
    public void test3() {
        List<CompanyProvincialScienceSmall> list = new ArrayList<>();
        CompanyProvincialScienceSmall companyProvincialScienceSmall = new CompanyProvincialScienceSmall();
        companyProvincialScienceSmall.setId(54L);

        list.add(companyProvincialScienceSmall);
        companyProvincialScienceSmallService.insertOrUpdateAllColumnBatch(list);
    }

    @Test
    public void test4() {
        List<CompanyProvincialScienceSmall> list = new ArrayList<>();
        CompanyProvincialScienceSmall companyProvincialScienceSmall = new CompanyProvincialScienceSmall();
        companyProvincialScienceSmall.setId(55L);
        companyProvincialScienceSmall.setRecognitiondate(new Date());
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        String format = sdf.format(new Date());
        System.out.println("------------>" + format);
        list.add(companyProvincialScienceSmall);
        companyProvincialScienceSmallService.insertOrUpdateAllColumnBatch(list);
    }

    @Test
    public void test5() {

        List<LabelIntelligentCommonInfo> labels = labelMapper.selectList(new EntityWrapper<LabelIntelligentCommonInfo>().eq("by1", "positionaltitles").eq("node", 320117400));

        List<String> names = labels.stream().map(LabelIntelligentCommonInfo::getName).collect(Collectors.toList());

        NativeSearchQueryBuilder nativeSearchQueryBuilder = new NativeSearchQueryBuilder();
        BoolQueryBuilder bqb = QueryBuilders.boolQuery();
        bqb.must(QueryBuilders.termQuery("type", 8));
        bqb.must(QueryBuilders.matchQuery("groupid", 1994));
        nativeSearchQueryBuilder.withQuery(bqb).withPageable(new PageRequest(0,10000));
        NativeSearchQuery build = nativeSearchQueryBuilder.build();
        List<ElasticSearchTTCModel> elasticSearchTTCModels = elasticsearchTemplate.queryForList(build, ElasticSearchTTCModel.class);

        for (ElasticSearchTTCModel model : elasticSearchTTCModels) {
            Resourcelibrary resourcelibrary = resourcelibraryMapper.selectById(model.getUnique());
            if (StringUtils.isNotBlank(resourcelibrary.getPositionaltitles())) {
                StringBuilder codes = new StringBuilder();
                for (LabelIntelligentCommonInfo label : labels) {
                    if (resourcelibrary.getPositionaltitles().contains(label.getName()) || resourcelibrary.getPositionaltitles().contains(label.getCode())) {
                        codes.append(label.getCode()).append(",");
                    }
                }
                if (StringUtils.isNotEmpty(codes.toString())) {
                    List<String> list1 = Arrays.asList(codes.toString().split(",")).stream().distinct().collect(Collectors.toList());
                    codes = new StringBuilder();
                    for (String s : list1) {
                        codes.append(s).append(",");
                    }
                    codes = new StringBuilder(codes.substring(0, codes.length() - 1));
                }
                model.setBy6(codes.toString());
                resourceRepository.save(model);
                System.out.println("修改的es id为：" + model.getId() + " , 修改的资源库id为：" + model.getUnique() + ", 修改的职称数据为：" + model.getBy6());
            }
        }
    }

}