From 354a21f90bdaad069ef58e6e53d0f2999135f11e Mon Sep 17 00:00:00 2001
From: Yangwook Jeong <yangwookee@gmail.com>
Date: Thu, 9 May 2024 13:06:26 +0900
Subject: [PATCH] fix: add conditional statement into __exit__ method

---
 service_layer/uow.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/service_layer/uow.py b/service_layer/uow.py
index b19916d..32f875d 100644
--- a/service_layer/uow.py
+++ b/service_layer/uow.py
@@ -22,7 +22,8 @@ def __enter__(self):
         return self
 
     def __exit__(self, exc_type, exc_value, traceback):
-        self.rollback()
+        if exc_type:
+            self.rollback()
 
     @abstractmethod
     def commit(self):