[Spring] 빈 생명 주기 관리 콜백

스프링 빈의 생명 주기

스프링 빈이 생성되거나, 소멸될 때 특정한 작업을 해야하는 경우가 있다.

예를 들어 빈이 생성되는 시점 전에 데이터베이스 커넥션 풀이나, 네트워크 소켓처럼 애플리케이션 시작 시점에 미리 필요한 연결을 해두고, 애플리케이션 종료 시점에 연결을 모두 종료하는 작업을 하려면 객체의 초기화와 종료 작업이 필요하다. 이러한 작업은 어떻게 하는걸까?

 

간단하게 외부 네트워크를 연결하는 객체를 생성한다고 가정해보자.

public class NetworkClient {

    private String url;
    
    public NetworkClient() {
        System.out.println("생성자 호출, url = " + url);
        connect();
        call("초기화 연결 메시지");
    }
    
    public void setUrl(String url) {
    	this.url = url;
    }
    
    //서비스 시작시 호출
    public void connect() {
    	System.out.println("connect: " + url);
    }
    
    public void call(String message) {
    	System.out.println("call: " + url + " message = " + message);
    }
    
    //서비스 종료시 호출
    public void disconnect() {
    	System.out.println("close: " + url);
    }
    ...
}

 

위 클래스를 스프링 빈으로 등록하여 실행해보자.

public class BeanLifeCycleTest {
    @Test
    public void lifeCycleTest() {
        ConfigurableApplicationContext ac = new AnnotationConfigApplicationContext(LifeCycleConfig.class);
        NetworkClient client = ac.getBean(NetworkClient.class);
        ac.close(); // 종료
    }
    
    @Configuration
    static class LifeCycleConfig {
    
        @Bean
        public NetworkClient networkClient() {
            NetworkClient networkClient = new NetworkClient();
            networkClient.setUrl("http://spring.dev"); // url 초기화
            return networkClient;
        }
    }
}

 

이를 실행한 출력결과는 다음과 같다.

생성자 호출, url = null 

connect: null 

call: null message = 초기화 연결 메시지

 

생성자 호출 부분을 보면 url 없이 connect가 호출되는 것을 확인할 수 있다. 당연한 결과이지만 객체를 생성하는 단계에는 url이 없고, 객체를 생성한 다음에 외부에서 수정자 주입을 통해 url을 초기화한다. 스프링 빈은 객체 생성 → 의존관계 주입으로 라이프사이클을 가진다. 

 

스프링 빈은 객체를 생성하고, 의존관계 주입이 다 끝난 후 사용할 수 있는 준비가 완료된다. 따라서 초기화 작업은 의존관계 주입이 끝난 다음에 할 수 있다. 그렇다면 의존관계 주입이 완료된 시점을 어떻게 알 수 있을까? 스프링은 이를 콜백 메서드를 통해 초기화 시점을 알려주는 다양한 기능을 제공한다. 또한 스프링은 스프링 컨테이너가 종료되기 직전에 소멸 콜백을 준다.

 

 

스프링 빈의 이벤트 라이프사이클 

컨테이너 생성 → 빈 생성 → 의존관계 주입 →  초기화 콜백 → 사용 → 소멸전 콜백 → 종료 

 

• 초기화 콜백: 빈이 생성되고, 빈의 의존관계 주입이 완료된 후 호출 
• 소멸전 콜백: 빈이 소멸되기 직전에 호출

 

객체의 생성과 초기화 분리 

생성자는 필수 정보(파라미터)를 받고, 메모리를 할당해서 객체를 생성하는 책임을 가진다. 반면에 초기화는 생성된 객체를 활용해서 외부 커넥션을 연결하는 등 무거운 동작을 수행한다. 이로인해 유지보수 관점에서 객체를 생성하는 부분과 초기화 하는 부분을 명확하게 나누는 것이 좋다. 당연히 내부 값들만 단순히 변경하는 경우에는 생성자에서 처리하는게 더 나을 수도 있다.

 

 

스프링은 크게 3가지 방법으로 빈 생명주기 콜백을 지원한다. 

1. 인터페이스(InitializingBean, DisposableBean) 
2. 설정 정보에 초기화 메서드, 종료 메서드 지정 
3. @PostConstruct, @PreDestroy 애노테이션 지원

 

이 중에 스프링에서 가장 권장하는 방법인 @PostConstruct, @PreDestroy 애너테이션을 알아보자.

 

@PostConstruct 

• 객체의 초기화 부분 시점 
• 객체가 생성된 후 별도의 초기화 작업을 위해 실행하는 메소드를 선언한다. 
• @PostConstruct 애너테이션을 설정해놓은 init 메소드는 WAS가 띄워질 때 실행된다. 

@PreDestroy 

• 마지막 소멸 단계 시점 
• 스프링 컨테이너에서 빈을 제거하기 전에 해야할 작업에 사용하는 애너테이션
• close() 하기 직전에 실행 -> ((AbstractApplicationContext) context).close()

public class NetworkClient {

    private String url;
    
    public NetworkClient() {
    	System.out.println("생성자 호출, url = " + url);
    }
    
    public void setUrl(String url) {
    	this.url = url;
    }
    
    //서비스 시작시 호출
    public void connect() {
    	System.out.println("connect: " + url);
    }
    
    public void call(String message) {
    	System.out.println("call: " + url + " message = " + message);
    }
    
    //서비스 종료시 호출
    public void disConnect() {
    	System.out.println("close + " + url);
    }
    
    @PostConstruct
    public void init() {
        System.out.println("NetworkClient.init");
        connect();
        call("초기화 연결 메시지");
    }
    
    @PreDestroy
    public void close() {
        System.out.println("NetworkClient.close");
        disConnect();
    }
}

 

@Configuration
static class LifeCycleConfig {

    @Bean
    public NetworkClient networkClient() {
        NetworkClient networkClient = new NetworkClient();
        networkClient.setUrl("http://spring.dev");
        return networkClient;
    }
}

 

이를 실행한 출력 결과는 다음과 같다.

 

생성자 호출 시점 → 생성자 호출, url = null

@PostConstruct → NetworkClient.init

@PostConstruct → connect: http://spring.dev
@PostConstruct → call: http://spring.dev message = 초기화 연결 메시지

19:40:50.269 [main] DEBUG

org.springframework.context.annotation.AnnotationConfigApplicationContext -

@PreDestory → Closing NetworkClient.close
@PreDestory → close + http://spring.dev

 

 

 

결론

단 한번만 실행하면 되는 메서드라면 @PostConstruct, @PreDestroy 애너테이션을 쓰면 아름답게 처리 가능하다.
코드를 코칠 수 없는 외부 라이브러리를 초기화, 종료해야 한다면 @Bean의 initMethod , destroyMethod를 사용하여 초기화 메서드, 종료 메서드를 지정하자.

 

 

Reference

스프링 핵심 원리 - 기본편 - 인프런 | 강의