SpringBoot自定义加载yml实现方式,附源码解读

import org.springframework.boot.env.YamlPropertySourceLoader

import org.springframework.core.env.PropertySource

import org.springframework.core.io.support.{DefaultPropertySourceFactory, EncodedResource}

/**

   * yaml资源加载类

   */

class YamlPropertyLoaderFactory extends DefaultPropertySourceFactory{

   override def createPropertySource(name: String, resource: EncodedResource): PropertySource[_] = {

    if (resource == null ) {

      super .createPropertySource(name, resource)

    }

    return new YamlPropertySourceLoader().load(resource.getResource.getFilename, resource.getResource, null )

   }

}

import com.core.conf.YamlPropertyLoaderFactory

import javax.persistence.EntityManagerFactory

import javax.sql.DataSource

import org.springframework.beans.factory.annotation.{Autowired, Qualifier}

import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder

import org.springframework.boot.context.properties.ConfigurationProperties

import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder

import org.springframework.context.annotation.{Bean, Configuration, PropertySource}

import org.springframework.core.env.Environment

import org.springframework.data.jpa.repository.config.EnableJpaRepositories

import org.springframework.orm.jpa.{JpaTransactionManager, LocalContainerEntityManagerFactoryBean}

import org.springframework.transaction.PlatformTransactionManager

/**

   * JPA 数据源配置

   */

@Configuration

@PropertySource (value = Array( "classpath:/bootstrap-report.yml" ), factory = classOf[YamlPropertyLoaderFactory])

@EnableJpaRepositories (

   entityManagerFactoryRef = "reportEntityManager" ,

   transactionManagerRef = "reportTransactionManager" ,

   basePackages = Array( "com.report.dao" )

)

class ReportDBConfig {

   @Autowired

   private var env: Environment = _

   @Bean

   @ConfigurationProperties (prefix = "spring.datasource.report" )

   def reportDataSource(): DataSource = DataSourceBuilder.create.build

   @Bean (name = Array( "reportEntityManager" ))

   def reportEntityManagerFactory(builder: EntityManagerFactoryBuilder): LocalContainerEntityManagerFactoryBean = {

    val entityManager = builder

      .dataSource(reportDataSource())

      .packages( "com.report.model" ) //设置JPA实体包路径

      .persistenceUnit( "reportPU" )

      .build

    entityManager.setJpaProperties(additionalProperties())

    entityManager

   }

   @Bean (name = Array( "reportTransactionManager" ))

   def reportTransactionManager( @Qualifier ( "reportEntityManager" )

                                entityManagerFactory: EntityManagerFactory): PlatformTransactionManager = {

    new JpaTransactionManager(entityManagerFactory)

   }

   /**

    * 获取JPA配置

    *

    * @return

    */

   def additionalProperties(): Properties = {

    val properties = new Properties();

    properties.setProperty( "hibernate.hbm2ddl.auto" , env.getProperty( "spring.jpa.report.hibernate.ddl-auto" ))

    properties.setProperty( "hibernate.show_sql" , env.getProperty( "spring.jpa.report.show-sql" ))

    properties.setProperty( "hibernate.dialect" , env.getProperty( "spring.jpa.report.database-platform" ))

    properties

   }

}

package org.springframework.context.annotation;

import java.lang.annotation.Documented;

import java.lang.annotation.ElementType;

import java.lang.annotation.Repeatable;

import java.lang.annotation.Retention;

import java.lang.annotation.RetentionPolicy;

import java.lang.annotation.Target;

import org.springframework.core.io.support.PropertySourceFactory;

@Target (ElementType.TYPE)

@Retention (RetentionPolicy.RUNTIME)

@Documented

@Repeatable (PropertySources. class )

public @interface PropertySource {

     // 用于声明属性源名称

     String name() default "" ;

     // 声明属性文件位置

     String[] value();

     // 是否忽略未找到的资源

     boolean ignoreResourceNotFound() default false ;

     // 声明配置文件的编码

     String encoding() default "" ;

     // 声明解析配置文件的类

     Class<? extends PropertySourceFactory> factory() default PropertySourceFactory. class ;

}

spring: 

  profiles:

    active: resources

user:

  filepath: 12346

  uname: "13"

admin:

  aname: 26

@Component

//@ConfigurationProperties(prefix = "user")

public class User {

    @Value ( "${user.filepath}" )

    private String filepath;

    @Value ( "${user.uname}" )

    private String uname;

    public String getFilepath() {

        return filepath;

    }

    public void setFilepath(String filepath) {

        this .filepath = filepath;

    }

    public String getUname() {

        return uname;

    }

    public void setUname(String uname) {

        this .uname = uname;

    }

    @Override

    public String toString() {

        return "User{" +

                "filepath='" + filepath + '\ '' +

                ", uname='" + uname + '\ '' +

                '}' ;

    }

}

@Component

@ConfigurationProperties (prefix = "user" )

public class User {

    //@Value("${user.filepath}")

    private String filepath;

    //@Value("${user.uname}")

    private String uname;

    public String getFilepath() {

        return filepath;

    }

    public void setFilepath(String filepath) {

        this .filepath = filepath;

    }

    public String getUname() {

        return uname;

    }

    public void setUname(String uname) {

        this .uname = uname;

    }

    @Override

    public String toString() {

        return "User{" +

                "filepath='" + filepath + '\ '' +

                ", uname='" + uname + '\ '' +

                '}' ;

    }

}

package com.sun123.springboot;

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.test.context.junit4.SpringRunner;

@RunWith (SpringRunner. class )

@SpringBootTest

public class UTest {

    @Autowired

    User user;

    @Test

    public void test01(){

        System.out.println(user);

    }

}