admin
2022-04-16 04f09e52ffd4681bdfd85e51acd3da0d1280c3d3
src/test/java/com/hxh/spring/test/GeneralTest.java
@@ -1,80 +1,37 @@
package com.hxh.spring.test;
import com.yeshi.buwan.dao.base.MongodbBaseDao;
import com.yeshi.buwan.dao.live.TVLiveProgramResourceDao;
import com.yeshi.buwan.dao.video.VideoResourceVersionMapDao;
import com.yeshi.buwan.domain.live.TVLiveProgramResource;
import com.yeshi.buwan.domain.user.UserDPContentWatchStatistic;
import com.yeshi.buwan.domain.video.StarInfo;
import com.yeshi.buwan.domain.video.VideoResourceVersionMap;
import com.yeshi.buwan.service.inter.video.VideoResourceVersionMapService;
import com.yeshi.buwan.dao.video.VideoResourceInfoMapDao;
import com.yeshi.buwan.domain.VideoResource;
import com.yeshi.buwan.domain.recommend.HomeTypeV2;
import com.yeshi.buwan.domain.recommend.HomeVideoV2;
import com.yeshi.buwan.domain.video.VideoInfoV2;
import com.yeshi.buwan.domain.video.VideoResourceInfoMap;
import com.yeshi.buwan.service.inter.video.VideoResourceInfoMapService;
import com.yeshi.buwan.vo.CommonQuery;
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;
import org.yeshi.utils.generater.GeneraterManager;
import org.yeshi.utils.generater.entity.ClassInfo;
public class GeneralTest {
    private String packageBaseName = "com.yeshi.buwan";
    private Class mongoBaseDao = MongodbBaseDao.class;
    private String PROJECT_PATH = "D:\\workspace\\BuWan";
    @Test
    public void createVideoInfoV2() {
        GeneraterManager.getInstance().init("D:\\workspace\\BuWan\\src\\main\\java", VideoInfoV2.class);
        try {
            GeneraterManager.getInstance().createDao("com.yeshi.buwan.dao.video");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    @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);
    public void createHomeTypeV2() {
        GeneraterManager.getInstance().init("D:\\workspace\\BuWan\\src\\main\\java", HomeTypeV2.class);
        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);
            GeneraterManager.getInstance().createDao("com.yeshi.buwan.dao.recommend");
        } catch (Exception e) {
            e.printStackTrace();
        }
@@ -82,51 +39,30 @@
    }
    @org.junit.Test
    public void createDao() {
        MongoDBDaoData.Builder builder = new MongoDBDaoData.Builder();
        builder.setBaseDaoClass(mongoBaseDao).setDaoPackageName(packageBaseName + ".dao.user");
        builder.setEntityClass(UserDPContentWatchStatistic.class);
    @Test
    public void createHomeVideoV2() {
        GeneraterManager.getInstance().init("D:\\workspace\\BuWan\\src\\main\\java", HomeVideoV2.class);
        try {
            SpringComponentGenerater.createMongoDao(builder.create(), PROJECT_PATH + "\\src\\main\\java\\com\\yeshi\\buwan\\dao\\user\\");
            GeneraterManager.getInstance().createDao("com.yeshi.buwan.dao.recommend");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    //
    @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(builder.build(), PROJECT_PATH + "\\src\\main\\java\\com\\yeshi\\buwan\\service\\inter\\goldcorn\\");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    //
//
    @org.junit.Test
    public void createServiceImpl() {
        try {
            SpringComponentGenerater.createServiceImpl(new ServiceData.Builder().setPackageName("com.yeshi.buwan.service.impl.video").setServiceClass(VideoResourceVersionMapService.class).setDaoClass(VideoResourceVersionMapDao.class).setEntityClass(VideoResourceVersionMap.class).build(), "D:\\workspace\\BuWan\\src\\main\\java\\com\\yeshi\\buwan\\service\\imp\\video\\");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
//
//    @org.junit.Test
//    public void createException() {
    @Test
    public void createViideoResourceInfoMap() throws Exception {
        GeneraterManager.getInstance().init("D:\\workspace\\BuWan\\src\\main\\java", VideoResourceInfoMap.class);
//        try {
//            SpringComponentGenerater.createException(new ExceptionData.Builder().setPackageName("com.yeshi.buwan.exception.video").setBaseClass(BaseException.class).setEntityClass(LuckySponsorMoneyRecord.class).build(), "D:\\workspace\\DayLucky\\facade-lucky\\src\\main\\java\\com\\ks\\lucky\\exception\\");
//            GeneraterManager.getInstance().createDao("com.yeshi.buwan.dao.video");
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//    }
//        GeneraterManager.getInstance().createService("com.yeshi.buwan.service.inter.video", "com.yeshi.buwan.service.imp.video", ClassInfo.create(CommonQuery.class), ClassInfo.create(VideoResourceInfoMapDao.class));
        GeneraterManager.getInstance().createAdmin("com.yeshi.buwan.controller.admin.api","D:\\workspace\\BuWan\\src\\main\\webapp\\admin\\new",ClassInfo.create(CommonQuery.class),ClassInfo.create(VideoResourceInfoMapService.class));
    }
}