es-server/src/test/java/es/UnitInit.java

package es;

import com.miniframe.Application;
import com.miniframe.core.ext.UtilTools;
import com.miniframe.model.es.EsUnit;
import com.miniframe.model.es.EsUnitGroup;
import com.miniframe.model.es.EsUnitSQLBuilder;
import com.miniframe.model.es.EsUnitSysDel;
import com.miniframe.model.es.dao.EsUnitGroupMapper;
import com.miniframe.model.es.dao.EsUnitMapper;
import com.miniframe.model.es.dao.EsUnitSysDelMapper;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

import java.math.BigDecimal;
import java.util.List;

@SpringBootTest(classes = Application.class)
public class UnitInit {
    @Test
    void initSysUnitGroup(){
        EsUnitGroupMapper esUnitGroupMapper = UtilTools.getBean(EsUnitGroupMapper.class);
        List<EsUnitGroup> esUnitGroupList = esUnitGroupMapper.selectAll();
        EsUnitSysDelMapper esUnitSysDelMapper = UtilTools.getBean(EsUnitSysDelMapper.class);
        EsUnitMapper dao = UtilTools.getBean(EsUnitMapper.class);
        for(EsUnitGroup esUnitGroup:esUnitGroupList){
            EsUnitSysDel esUnitSysDel = new EsUnitSysDel();
            esUnitSysDel.setGsutId(UtilTools.getUUid());
            esUnitSysDel.setSutId("1");
            esUnitSysDel.setGutId(esUnitGroup.getGutId());
            EsUnitSQLBuilder sb = new EsUnitSQLBuilder();
            EsUnitSQLBuilder.Criteria sc = sb.createCriteria();
            sc.andGutIdEqualTo(esUnitGroup.getGutId());
            sc.andFactorEqualTo(new BigDecimal(1));
            sb.setOrderByClause("ser asc");
            List<EsUnit> us = dao.selectByExample(sb);
            if(us==null||us.isEmpty()){
                esUnitSysDel.setUtId("");
                esUnitSysDel.setValue("");
            }else{
                EsUnit u= us.get(0);
                esUnitSysDel.setUtId(u.getUtId());
                esUnitSysDel.setValue(u.getValue());
            }
            esUnitSysDelMapper.insert(esUnitSysDel);
        }
    }
}