fix registry._setup_database.lock remove issue and rollback the session after UNIQUE constraint failed exception

docker_registry/lib/index/db.py

@@ -113,7 +113,11 @@ def _generate_index(self, session):
         session.add(Version(id=self.version))
         for repository in self._walk_storage(store=store):
             session.add(Repository(**repository))
-        session.commit()
+        try:
+            session.commit()
+        except Exception as e:
+            logger.exception(str(e))
+            session.rollback()
 
     @retry
     def _handle_repository_created(

docker_registry/toolkit.py

@@ -327,7 +327,8 @@ def wrapper(*args, **kwargs):
         try:
             result = f(*args, **kwargs)
         finally:
-            os.remove(lock_path)
+            if os.path.exists(lock_path):
+                os.remove(lock_path)
         return result
     return wrapper