io.github.ramerf.wind.WindApplication Maven / Gradle / Ivy

Go to download
package io.github.ramerf.wind;

import io.github.ramerf.wind.core.WindVersion;
import io.github.ramerf.wind.core.annotation.TableInfo;
import io.github.ramerf.wind.core.ansi.*;
import io.github.ramerf.wind.core.autoconfig.AutoConfigConfiguration;
import io.github.ramerf.wind.core.config.Configuration;
import io.github.ramerf.wind.core.config.JdbcEnvironment;
import io.github.ramerf.wind.core.exception.ClassInstantiationException;
import io.github.ramerf.wind.core.exception.WindException;
import io.github.ramerf.wind.core.executor.DaoFactory;
import io.github.ramerf.wind.core.executor.DefaultDaoFactory;
import io.github.ramerf.wind.core.handler.typehandler.ITypeHandler;
import io.github.ramerf.wind.core.handler.typehandler.TypeHandlerRegistryFactory;
import io.github.ramerf.wind.core.helper.EntityHelper;
import io.github.ramerf.wind.core.jdbc.transaction.TransactionFactory;
import io.github.ramerf.wind.core.jdbc.transaction.jdbc.JdbcTransactionFactory;
import io.github.ramerf.wind.core.jdbc.metadata.DbMetaData;
import io.github.ramerf.wind.core.plugin.DaoInterceptor;
import io.github.ramerf.wind.core.plugin.ServiceInterceptor;
import io.github.ramerf.wind.core.util.*;
import java.io.IOException;
import java.io.InputStream;
import java.util.Set;
import javax.annotation.Nonnull;
import javax.sql.DataSource;
import lombok.extern.slf4j.Slf4j;

/**
 * The type Wind application.
 *
 * @since 2022.03.19
 * @author ramer
 */
@Slf4j
public class WindApplication {
  private WindContext windContext = new WindContext();

  private WindApplication() {}

  /** 通过指定配置文件启动. */
  public static WindApplication run(final InputStream inputStream) {
    final AutoConfigConfiguration autoConfigConfiguration =
        YmlUtil.process(AutoConfigConfiguration.class, inputStream);
    return run(autoConfigConfiguration.getConfiguration());
  }

  /** 通过指定配置文件启动. */
  public static WindApplication run(final String configPath) {
    final AutoConfigConfiguration autoConfigConfiguration =
        YmlUtil.process(AutoConfigConfiguration.class, configPath);
    return run(autoConfigConfiguration.getConfiguration());
  }

  public static WindApplication run(@Nonnull final DataSource dataSource) {
    return run(new JdbcTransactionFactory(), dataSource);
  }

  public static WindApplication run(
      @Nonnull final TransactionFactory transactionFactory, @Nonnull final DataSource dataSource) {
    Configuration configuration = new Configuration();
    configuration.setJdbcEnvironment(new JdbcEnvironment(transactionFactory, dataSource));
    return run(configuration);
  }

  public static WindApplication run(@Nonnull Configuration configuration) {
    final JdbcEnvironment jdbcEnvironment = configuration.getJdbcEnvironment();
    Asserts.notNull(jdbcEnvironment, "DataSource not found");
    final DataSource dataSource = jdbcEnvironment.getDataSource();
    WindContext windContext = new WindContext();
    windContext.setDbMetaData(DbMetaData.getInstance(dataSource, configuration.getDialect()));
    windContext.setConfiguration(configuration);
    // 设置拦截器
    populateInterceptors(configuration);
    // 类型处理器
    populateTypeHandler(configuration);
    // 打印banner
    printBanner();
    // 初始化实体解析类
    EntityUtils.initial(windContext);
    EntityHelper.initial(windContext);
    // 解析实体元数据
    initEntityInfo(configuration);
    // 注册默认类型处理器
    BeanUtils.getClazz(TypeHandlerRegistryFactory.class.getName());
    WindApplication windApplication = new WindApplication();
    windApplication.windContext = windContext;
    return windApplication;
  }

  public WindContext getWindContext() {
    return windContext;
  }

  public Configuration getConfiguration() {
    return windContext.getConfiguration();
  }

  public DaoFactory getDaoFactory() {
    return DefaultDaoFactory.of(getConfiguration());
  }

  private static void populateInterceptors(final Configuration configuration) {
    final String interceptorPackage = configuration.getInterceptorPackage();
    if (StringUtils.nonEmpty(interceptorPackage)) {
      // dao 拦截器
      try {
        final Set> classes =
            BeanUtils.scanClasses(interceptorPackage, DaoInterceptor.class);
        for (Class clazz : classes) {
          try {
            configuration.addInterceptor(BeanUtils.initial(clazz));
          } catch (ClassInstantiationException e) {
            throw new WindException(
                "Fail to initial dao interceptor:" + clazz + ",require no arg constructor", e);
          }
        }
      } catch (IOException e) {
        log.warn("Fail to populate dao interceptors:" + interceptorPackage, e);
      }
      // service 拦截器
      try {
        final Set> classes =
            BeanUtils.scanClasses(interceptorPackage, ServiceInterceptor.class);
        for (Class clazz : classes) {
          try {
            configuration.addInterceptor(BeanUtils.initial(clazz));
          } catch (ClassInstantiationException e) {
            throw new WindException(
                "Fail to initial service interceptor:" + clazz + ",require no arg constructor", e);
          }
        }
      } catch (IOException e) {
        log.warn("Fail to populate service interceptors:" + interceptorPackage, e);
      }
    }
  }

  private static void populateTypeHandler(final Configuration configuration) {
    final String typeHandlerPackage = configuration.getTypeHandlerPackage();
    if (StringUtils.nonEmpty(typeHandlerPackage)) {
      try {
        @SuppressWarnings("rawtypes")
        final Set> classes =
            BeanUtils.scanClasses(typeHandlerPackage, ITypeHandler.class);
        //noinspection rawtypes
        for (Class clazz : classes) {
          try {
            configuration.addTypeHandler(BeanUtils.initial(clazz));
          } catch (ClassInstantiationException e) {
            throw new WindException(
                "Fail to initial dao type handler:" + clazz + ",require no arg constructor", e);
          }
        }
      } catch (IOException e) {
        log.warn("Fail to populate type handler:" + typeHandlerPackage, e);
      }
    }
  }

  private static void printBanner() {
    System.out.println(
        "\n"
            + " _       __    ____    _   __    ____ \n"
            + "| |     / /   /  _/   / | / /   / __ \\\n"
            + "| | /| / /    / /    /  |/ /   / / / /\n"
            + "| |/ |/ /   _/ /_   / /|  /   / /_/ /\n"
            + "|__/|__/   /___/   /_/ |_/   /_____/");
    System.out.println(
        AnsiOutput.toString(
            AnsiColor.GREEN,
            " :: wind ::",
            AnsiColor.DEFAULT,
            " (v" + WindVersion.getVersion() + ")\n",
            AnsiStyle.FAINT));
  }

  private static void initEntityInfo(final Configuration configuration) {
    String entityPackage;
    if (StringUtils.nonEmpty(configuration.getEntityPackage())) {
      entityPackage = configuration.getEntityPackage();
    } else {
      entityPackage = WindVersion.class.getPackage().getName();
    }
    log.info("initEntityInfo:package[{}]", entityPackage);
    Set> entities;
    try {
      entities = BeanUtils.scanClassesWithAnnotation(entityPackage, TableInfo.class);
      if (entities.size() < 1) {
        log.info(
            "no entity with @TableInfo annotation found in path: [{}], correct your configuration:wind.entity-package",
            entityPackage);
      }
    } catch (IOException e) {
      log.warn("Fail to init entity info:" + entityPackage, e);
      return;
    }
    if (!entities.isEmpty()) {
      entities.forEach(EntityHelper::initEntity);
      EntityHelper.initEntityMapping();
    }
  }
}