springboot在yaml文件裏面注入自定義常量的兩種方式

一 : 直接注入方式

1 、pom.xml

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.3.0.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.example</groupId>
    <artifactId>demo</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>demo</name>
    <description>Demo project for Spring Boot</description>

    <properties>
        <java.version>1.8</java.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-configuration-processor</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>

</project>

2、application.yaml

#JWT 密鑰
jwt:
  secretKey: 123456
  accessTokenExpireTime: PT2H
  refreshTokenExpireTime: PT8H
  refreshTokenExpireAppTime: P30D
  issuer: baidu.com

#JWT 密鑰
jwt2:
  secretKey: 123456
  accessTokenExpireTime: PT2H
  refreshTokenExpireTime: PT8H
  refreshTokenExpireAppTime: P30D
  issuer: baidu.com

3、util1包下面新建tookensetting

package com.example.demo.util1;

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

import java.time.Duration;

/**
 * @author 劉陽洋
 * @date 2020/6/12 12:23
 */
@Component
@ConfigurationProperties(prefix = "jwt")
public class TokenSetting {


    public static String secretKey;
    public static Duration accessTokenExpireTime;
    public static Duration refreshTokenExpireTime;
    public static Duration refreshTokenExpireAppTime;
    public static String issuer;

    public String getSecretKey() {
        return secretKey;
    }

    public void setSecretKey(String secretKey) {
        TokenSetting.secretKey = secretKey;
    }

    public Duration getAccessTokenExpireTime() {
        return accessTokenExpireTime;
    }

    public void setAccessTokenExpireTime(Duration accessTokenExpireTime) {
        TokenSetting.accessTokenExpireTime = accessTokenExpireTime;
    }

    public Duration getRefreshTokenExpireTime() {
        return refreshTokenExpireTime;
    }

    public void setRefreshTokenExpireTime(Duration refreshTokenExpireTime) {
        TokenSetting.refreshTokenExpireTime = refreshTokenExpireTime;
    }

    public Duration getRefreshTokenExpireAppTime() {
        return refreshTokenExpireAppTime;
    }

    public void setRefreshTokenExpireAppTime(Duration refreshTokenExpireAppTime) {
        TokenSetting.refreshTokenExpireAppTime = refreshTokenExpireAppTime;
    }

    public String getIssuer() {
        return issuer;
    }

    public void setIssuer(String issuer) {
        TokenSetting.issuer = issuer;
    }


}

4、DemoApplicationTests

package com.example.demo;

import com.example.demo.util1.TokenSetting;
import com.example.demo.util2.TokenSettings;
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;

import java.time.Duration;

@SpringBootTest
@RunWith(SpringRunner.class)
public class DemoApplicationTests {

    //第一種直接注入的方式
    @Test
    public void testToken1() {
        System.out.println("1、得到的值爲：" + TokenSetting.secretKey);
        System.out.println("2、得到的值爲：" + TokenSetting.issuer);
        System.out.println("3、得到的值爲：" + TokenSetting.accessTokenExpireTime);
        System.out.println("4、得到的值爲：" + TokenSetting.refreshTokenExpireAppTime);
        System.out.println("5、得到的值爲：" + TokenSetting.refreshTokenExpireTime);

    }




}

二 : 使用靜態方法初始化注入

1、util2包下面新建InitializerUtil.java

package com.example.demo.util2;

import com.example.demo.DemoApplication;
import org.springframework.stereotype.Component;

/**
 * @author 劉陽洋
 * @date 2020/6/12 12:36
 */

@Component
public class InitializerUtil {
    private TokenSettings tokenSettings;
    public InitializerUtil(TokenSettings tokenSettings){
        DemoApplication.setTokenSettings(tokenSettings);
    }
}

2、新建TokenSettings.java類

package com.example.demo.util2;

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.time.Duration;

/**
 * @author 劉陽洋
 * @date 2020/6/12 12:33
 */
@Component
@ConfigurationProperties(prefix = "jwt2")
public class TokenSettings {
    private String secretKey;
    private Duration accessTokenExpireTime;
    private Duration refreshTokenExpireTime;
    private Duration refreshTokenExpireAppTime;
    private String issuer;

    public String getSecretKey() {
        return secretKey;
    }

    public void setSecretKey(String secretKey) {
        this.secretKey = secretKey;
    }

    public Duration getAccessTokenExpireTime() {
        return accessTokenExpireTime;
    }

    public void setAccessTokenExpireTime(Duration accessTokenExpireTime) {
        this.accessTokenExpireTime = accessTokenExpireTime;
    }

    public Duration getRefreshTokenExpireTime() {
        return refreshTokenExpireTime;
    }

    public void setRefreshTokenExpireTime(Duration refreshTokenExpireTime) {
        this.refreshTokenExpireTime = refreshTokenExpireTime;
    }

    public Duration getRefreshTokenExpireAppTime() {
        return refreshTokenExpireAppTime;
    }

    public void setRefreshTokenExpireAppTime(Duration refreshTokenExpireAppTime) {
        this.refreshTokenExpireAppTime = refreshTokenExpireAppTime;
    }

    public String getIssuer() {
        return issuer;
    }

    public void setIssuer(String issuer) {
        this.issuer = issuer;
    }
}

3、在DemoApplication.java類中

package com.example.demo;

import com.example.demo.util2.TokenSettings;
import org.junit.Test;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

import java.time.Duration;

@SpringBootApplication
public class DemoApplication {

    private static String secretKey;
    private static Duration accessTokenExpireTime;
    private static Duration refreshTokenExpireTime;
    private static Duration refreshTokenExpireAppTime;
    private static String issuer;

    public static void setTokenSettings(TokenSettings tokenSettings){
        secretKey=tokenSettings.getSecretKey();
        accessTokenExpireTime=tokenSettings.getAccessTokenExpireTime();
        refreshTokenExpireTime=tokenSettings.getRefreshTokenExpireTime();
        refreshTokenExpireAppTime=tokenSettings.getRefreshTokenExpireAppTime();
        issuer=tokenSettings.getIssuer();
    }
    public static void main(String[] args) {
        SpringApplication.run(DemoApplication.class, args);

            System.out.println("1、得到的值爲："+secretKey);
            System.out.println("2、得到的值爲："+issuer);
            System.out.println("3、得到的值爲："+accessTokenExpireTime);
            System.out.println("4、得到的值爲："+refreshTokenExpireAppTime);
            System.out.println("5、得到的值爲："+refreshTokenExpireTime);
    }

}