🔎 Self Invocation?

기본 개념

public class Sample {
    public void innerMethod1() {
        innerMethod2();
    }
    
    public void innerMethod2() {
        System.out.println("innerMethod2");
    }
}

Sample 클래스 내에 innerMethod1이라는 내부 메서드가 innerMethod2를 호출하고 있습니다.
이렇게 한 클래스 내에서 내부 메서드가 또 다른 내부 메서드를 호출하는 것을 Self Invocation, 즉 자가 호출이라고 합니다.

이런 Self Invocation은 보통의 경우라면 아무 문제 없는 코드이지만 만약 @Transactional, @Async와 같이 프록시를 통해 동작하는 어노테이션을 만나면 의도대로 동작하지 않을 수도 있습니다.

@Service
@RequiredArgsConstructor
public class SampleService {
    private final SampleRepository sampleRepository;

    @Transactional(readOnly = true)
    public void makeNewSample(long id) {
        sample sample = sampleRepository.findById(id);
        
        if(sample != null)
            throw new DuplicateException();
        
        Sample sample = new Sample();
        saveSample(sample);
    }
    
    @Transactional(propagation = Propagation.REQUIRES_NEW)
    public void saveSample(Sample sample) {
        sampleRepository.save(sample);
    }
}

위의 코드를 보면 makeNewSample() 메서드 내부에서 saveSample() 메서드를 호출하는데
이때 @Transactional(propagation = Propagation.REQUIRES_NEW) 설정이 무시되고 makeNewSample() 메서드의 readOnly = true 트랜잭션 컨텍스트가 그대로 유지되어 sampleRepository.save(sample);이 동작하지 않게 됩니다.

왜 이런 상황이 발생할까?

@Transactional, @Async 와 같은 AOP 어노테이션들은 프록시 방식을 사용합니다.
그러므로 @Transactional 어노테이션이 선언된 메서드를 호출하면 프록시를 통해 트랜잭션이 시작되고 이후에 실제 객체의 makeNewSample() 메서드가 호출됩니다.
이때 makeNewSample() 메서드에서 saveSample() 메서드를 호출하면 이는 프록시가 아닌 this 참조에 대해 호출됩니다.

이렇게 Self Inovcation이 일어나면 해당 메서드에 적용된 어드바이스(Advice)가 실행되지 않습니다. 위의 예시에서는 Transactional(propagation = Propagation.REQUIRES_NEW) 설정이 무시되어서 데이터가 저장되지 않는 상황이 발생하게 됩니다.

Self Invocation 해결법

Avoid Self Invocation

보통 별도의 클래스로 분리해 Self Invocation이 일어나지 않도록 합니다.
이로 인해 클래스가 과도하게 많아질 수 있기 때문에 적절한 책임 분리가 필요합니다.

Inject a self reference

Self Injection으로 this 대신에 주입된 Proxy 참조를 통해 메서드를 호출합니다.
순환 참조가 발생할 수 있기 때문에 @Lazy로 사용 시 초기화할 수 있도록 해야 합니다.

@Service
@RequiredArgsConstructor
public class SampleService {
    private final SampleRepository sampleRepository;
    
    @Lazy
    @Autowired
    private SampleService sampleService;

    @Transactional(readOnly = true)
    public void makeNewSample(long id) {
        sample sample = sampleRepository.findById(id);
        
        if(sample != null)
            throw new DuplicateException();
        
        Sample sample = new Sample();
        sampleService.saveSample(sample);
    }
    
    @Transactional(propagation = Propagation.REQUIRES_NEW)
    public void saveSample(Sample sample) {
        sampleRepository.save(sample);
    }
}    

Use AopContext.currentProxy()

AopContext의 currentProxy() 메서드를 이용해 호출합니다.
해당 방법은 Spring Docs에 권장되지 않고, 최후의 수단으로 가급적 피해야 한다고 작성되어 있습니다.

public class SampleServiceImpl implements SampleService {

	@Transactional(readOnly = true)
    public void makeNewSample() {
		// This works, but it should be avoided if possible.
		((SampleService) AopContext.currentProxy()).saveSample(sample);
	}

	@Transactional(propagation = Propagation.REQUIRES_NEW)
    public void saveSample(Sample sample) {
		// some logic...
	}
}