AI 에이전트의 변경을 "되돌리기": JetBrains Junie의 롤백 (Rollback) 기능을 사용해 보았다

서론

AI 코딩 에이전트 (AI Coding Agent)는 여러 파일에 걸친 변경을 일괄적으로 수행할 수 있습니다. 코드 리팩터링 (Refactoring), 문서 업데이트, 그리고 추가 지시에 대한 대응 등을 한 번에 처리해 줍니다.

그 편리함의 이면에는 실용적인 우려 사항이 있습니다. 에이전트가 하나의 태스크 (Task)로 여러 파일을 변경했을 경우, "어떤 변경을 남기고, 어떤 것을 조정하며, 어떤 것을 폐기해야 하는가"를 판단해야 합니다.

예를 들어, "리팩터링 변경은 좋았지만, 문서 업데이트 내용이 의도한 것이 아니다"라는 케이스나, "개선된 부분은 있지만 예기치 않은 다른 변경이 포함되어 버렸다"라는 상황이 발생할 수 있습니다. 여기에서는 AI 에이전트가 코드를 생성할 수 있는지 여부뿐만 아니라, 그 작업 내용을 안전하게 확인하고 필요에 따라 되돌릴 수 있는지 여부를 확인합니다.

JetBrains의 AI 코딩 에이전트 Junie에는 변경을 적용한 후에 그것들을 복구할 수 있는 롤백 (Rollback) 기능이 갖춰져 있으므로, 이번에는 Java 검증 환경을 준비하여 Junie의 롤백 조작을 확인합니다. 여러 파일이 변경되었을 때 특정 파일만 복구하는 경우, 추가 지시 후의 복구 등 각각의 동작이 어떻게 되는지 확인했습니다.

검증 내용

이번에 확인하는 것은 Junie의 코드 생성 능력 그 자체가 아니라, Junie가 수행한 변경을 얼마나 유연하게 다룰 수 있는가 하는 점입니다.

• 여러 파일 변경 후, 변경된 파일 목록을 확인할 수 있는가
• 변경을 적용하고 싶지 않은 경우, 모든 변경을 일괄적으로 취소하여 되돌릴 수 있는가
• 특정 파일만 변경을 되돌릴 수 있는가

다음으로 테스트에 사용한 검증 환경을 살펴보겠습니다.

검증 환경

검증용으로 Junie에게 다음과 같은 Java 프로젝트를 준비하도록 했습니다.

junie_rollback_verification/
README.md
run.ps1
...

InvoiceCalculator.java에는 검증용으로 중복된 계산 로직이 포함되어 있습니다.

public int calculateRetailTotal(int itemCount, int unitPrice) {
if (itemCount < 0) {
throw new IllegalArgumentException("itemCount must be 0 or greater");
...

calculateWholesaleTotal에도 유사한 체크와 계산이 존재하기 때문에, Junie에게 "중복된 로직을 삭제해 줘"라고 의뢰하기 쉬운 구성으로 되어 있습니다.

Junie에게 준 태스크

Junie에게 다음과 같은 지시를 내렸습니다.

src/main/java/demo/InvoiceCalculator.java 를 리팩터링하여 중복된 로직을 삭제하고, 공개 메서드(Public Method)의 이름이나 계산 결과는 변경하지 말 것. 변경 사항에 대한 짧은 개요를 docs/notes.md 에 추가해 줘.

여러 파일을 동시에 수정하도록 의뢰했습니다.

다음 섹션에서 테스트 중에 확인해야 할 동작을 살펴보겠습니다.

동작 확인

• 변경된 파일 목록 확인
• InvoiceCalculator.java의 차이(diff) 확인
• docs/notes.md의 차이(diff) 확인
• 모든 변경의 롤백 실행
• 특정 파일만의 롤백 실행

추가 지시 시도

첫 번째 지시 이후, 추가 지시를 보냅니다.

리팩터링 방침은 그대로 유지하면서, 헬퍼 메서드(Helper Method)의 이름을 더 알기 쉬운 것으로 변경해 줘.
헬퍼 메서드는 private 상태로 유지해 줘.
공개 메서드의 이름은 변경하지 마.

추가 작업이 완료된 후, 롤백을 수행했을 때 어떤 일이 일어나는지 검증했습니다.

검증 결과

이번 검증을 통해 Junie가 수행한 변경에 대해 다음과 같은 동작을 확인할 수 있었습니다.

• Junie의 태스크 완료 후, 차이(diff) 뷰에서 변경된 파일을 확인할 수 있음
• InvoiceCalculator.java와 docs/notes.md처럼 여러 파일에 걸친 변경을 횡단적으로 확인할 수 있음
• 수행된 모든 변경을 일괄적으로 롤백할 수 있음
• 특정 파일만 선택하여 롤백할 수 있음
• 추가 지시 후에 롤백을 사용하면, 프로젝트는 직전 변경 전의 상태로 돌아감
• 롤백을 반복함으로써 그 이전의 변경 상태까지 되돌릴 수 있음

특히 편리하다고 느낀 점은, Junie가 변경한 파일 중 특정 파일만 롤백 (Rollback)할 수 있다는 점입니다.

예를 들어, "Java 코드 리팩터링 (Refactoring)은 채택하고 싶지만, docs/notes.md의 수정만은 되돌리고 싶다"라는 경우, 파일 단위의 롤백으로 대응이 가능합니다.

또한, 추가 지시를 내린 후에도 "최근 변경 단위"로 되돌릴 수 있기 때문에, 의도하지 않은 수정이 포함되었을 경우에도 안심하고 다시 시도할 수 있다는 것을 알 수 있었습니다.

요약

AI 코딩 에이전트 (AI Coding Agent)는 여러 파일에 걸친 변경을 맡길 수 있을 때 진가를 발휘하지만, 그렇기에 작업 내용을 안전하게 리뷰하고 취소할 수 있는 메커니즘이 필수적입니다.

이번 검증에서는 검증용 프로젝트를 통해 Junie의 롤백 기능을 확인했습니다.

• 중복 로직의 리팩터링 (Refactoring) 요청
• 여러 파일의 차이 (Diff) 리뷰
• 추가 지시 시도
• 필요에 따른 롤백 실행

AI의 성과 중 일부만 활용하고 싶다거나, 특정 파일만 되돌리고 싶다, 혹은 추가 지시 단계만 취소하고 싶다와 같이, 실제 개발 현장에서 일어날 수 있는 상황에 대응할 수 있는 기능임을 알 수 있었습니다.

"에이전트에게 변경을 맡기고, 차이를 확인하며, 좋은 부분은 남기고 불필요한 부분은 되돌린다". 이 워크플로 (Workflow)가 확립됨으로써 AI 에이전트를 효율적으로 이용할 수 있게 됩니다.

낫토 시스템(Nattou System)에서 알려드립니다

당사는 JetBrains 제품에 관한 질문 및 상담 등을 접수하고 있습니다. 당사의 X 또는 이메일로 연락해 주시기 바랍니다.