admin
2021-09-24 f788607ff771a47bc60d6a86e00b3433c40f3d2c
src/test/java/com/hxh/spring/test/GeneralTest.java
@@ -1,35 +1,94 @@
package com.hxh.spring.test;
import com.yeshi.buwan.dao.base.MongodbBaseDao;
import com.yeshi.buwan.dao.goldcorn.GoldCornTaskActionDetailDao;
import com.yeshi.buwan.dao.live.TVLiveProgramResourceDao;
import com.yeshi.buwan.dao.video.VideoResourceVersionMapDao;
import com.yeshi.buwan.domain.goldcorn.GoldCornTaskActionDetail;
import com.yeshi.buwan.domain.live.*;
import com.yeshi.buwan.domain.pptv.PPTVPlayProcessRecord;
import com.yeshi.buwan.domain.special.SearchSpecial;
import com.yeshi.buwan.domain.special.SearchSpecialPosition;
import com.yeshi.buwan.domain.special.SearchSpecialPositionMap;
import com.yeshi.buwan.domain.tb.YouZhiHaoHuoGoods;
import com.yeshi.buwan.domain.live.TVLiveProgramResource;
import com.yeshi.buwan.domain.video.StarInfo;
import com.yeshi.buwan.domain.video.VideoResourceVersionMap;
import com.yeshi.buwan.domain.vip.VideoBuyRecord;
import com.yeshi.buwan.live.migu.MiGuLiveListInfo;
import com.yeshi.buwan.service.inter.video.VideoResourceVersionMapService;
import com.yeshi.buwan.videos.bilibili.entity.BilibiliEpInfo;
import com.yeshi.buwan.videos.bilibili.entity.BilibiliMediaInfo;
import com.yeshi.buwan.videos.bilibili.entity.BilibiliSearchVideoMap;
import com.yeshi.buwan.videos.bilibili.entity.BilibiliVideo;
import org.junit.Test;
import org.yeshi.utils.generater.SpringComponentGenerater;
import org.yeshi.utils.generater.entity.MongoDBDaoData;
import org.yeshi.utils.generater.entity.ServiceData;
import java.io.File;
public class GeneralTest {
    private String packageBaseName = "com.yeshi.buwan";
    private Class mongoBaseDao = MongodbBaseDao.class;
    private String PROJECT_PATH = "D:\\workspace\\BuWan";
    @Test
    public void generalMongoDaoAndService() throws ClassNotFoundException {
        /************可更改区域开始*************/
        Class entityClass = StarInfo.class;
        //dao的相对包名
        String daoRelativePackageName = "dao.video";
        String serviceRelativePackageName = "service.inter.video";
        String serviceImplRelativePackageName = "service.imp.video";
        /************可更改区结束*************/
        String daoPackageName = packageBaseName + "." + daoRelativePackageName;
        String daoClassName = daoPackageName + "." + entityClass.getSimpleName() + "Dao";
        //生成dao
        MongoDBDaoData.Builder daoBuilder = new MongoDBDaoData.Builder();
        daoBuilder.setBaseDaoClass(mongoBaseDao).setDaoPackageName(daoPackageName);
        daoBuilder.setEntityClass(entityClass);
        try {
            String targetDir = PROJECT_PATH + "\\src\\main\\java\\" + daoPackageName.replace(".", "\\") + "\\";
            if (!new File(targetDir).exists())
                new File(targetDir).mkdirs();
            SpringComponentGenerater.createMongoDao(daoBuilder.create(), targetDir);
        } catch (Exception e) {
            e.printStackTrace();
        }
        //生成服务接口
        String servicePackageName = packageBaseName + "." + serviceRelativePackageName;
        String serviceClassName = servicePackageName + "." + entityClass.getSimpleName() + "Service";
        ServiceData.Builder builder = new ServiceData.Builder();
        builder.setPackageName(servicePackageName);
        builder.setDaoClass(Class.forName(daoClassName));
        builder.setEntityClass(entityClass);
        try {
            String targetDir = PROJECT_PATH + "\\src\\main\\java\\" + servicePackageName.replace(".", "\\") + "\\";
            if (!new File(targetDir).exists())
                new File(targetDir).mkdirs();
            SpringComponentGenerater.createService(builder.build(), targetDir);
        } catch (Exception e) {
            e.printStackTrace();
        }
        //生成服务实现接口
        String serviceImplPackageName = packageBaseName + "." + serviceImplRelativePackageName;
        builder = new ServiceData.Builder();
        builder.setPackageName(serviceImplPackageName).setServiceClass(Class.forName(serviceClassName)).setDaoClass(Class.forName(daoClassName)).setEntityClass(entityClass);
        try {
            String targetDir = PROJECT_PATH + "\\src\\main\\java\\" + serviceImplPackageName.replace(".", "\\") + "\\";
            if (!new File(targetDir).exists())
                new File(targetDir).mkdirs();
            SpringComponentGenerater.createServiceImpl(builder.build(), targetDir);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    @org.junit.Test
    public void createDao() {
        MongoDBDaoData.Builder builder = new MongoDBDaoData.Builder();
        builder.setBaseDaoClass(mongoBaseDao).setDaoPackageName(packageBaseName + ".dao.live");
        builder.setEntityClass(TVLiveProgramResource.class);
        try {
            SpringComponentGenerater.createMongoDao(new MongoDBDaoData.Builder().setBaseDaoClass(MongodbBaseDao.class).setDaoPackageName("com.yeshi.buwan.dao.search").setEntityClass(SearchSpecialPosition.class).create(), "D:\\workspace\\BuWan\\src\\main\\java\\com\\yeshi\\buwan\\dao\\search\\");
            SpringComponentGenerater.createMongoDao(builder.create(), PROJECT_PATH + "\\src\\main\\java\\com\\yeshi\\buwan\\dao\\live\\");
        } catch (Exception e) {
            e.printStackTrace();
        }
@@ -38,8 +97,12 @@
    //
    @org.junit.Test
    public void createService() {
        ServiceData.Builder builder = new ServiceData.Builder();
        builder.setPackageName(packageBaseName + ".service.inter.live");
        builder.setDaoClass(TVLiveProgramResourceDao.class);
        builder.setEntityClass(TVLiveProgramResource.class);
        try {
            SpringComponentGenerater.createService(new ServiceData.Builder().setPackageName("com.yeshi.buwan.service.inter.goldcorn").setDaoClass(GoldCornTaskActionDetailDao.class).setEntityClass(GoldCornTaskActionDetail.class).build(), "D:\\workspace\\BuWan\\src\\main\\java\\com\\yeshi\\buwan\\service\\inter\\goldcorn\\");
            SpringComponentGenerater.createService(builder.build(), PROJECT_PATH + "\\src\\main\\java\\com\\yeshi\\buwan\\service\\inter\\goldcorn\\");
        } catch (Exception e) {
            e.printStackTrace();
        }