ATLAS-2982: import fails to create classification-def - #2

repository/src/main/java/org/apache/atlas/repository/impexp/ImportService.java

@@ -80,8 +80,6 @@ public class ImportService {
 
     public AtlasImportResult run(ZipSource source, AtlasImportRequest request, String userName,
                                  String hostName, String requestingIP) throws AtlasBaseException {
-        RequestContext.get().setImportInProgress(true);
-
         if (request == null) {
             request = new AtlasImportRequest();
         }
@@ -91,6 +89,8 @@ public class ImportService {
         try {
             LOG.info("==> import(user={}, from={}, request={})", userName, requestingIP, request);
 
+            RequestContext.get().setImportInProgress(true);
+
             String transforms = MapUtils.isNotEmpty(request.getOptions()) ? request.getOptions().get(TRANSFORMS_KEY) : null;
             setImportTransform(source, transforms);
 
@@ -110,6 +110,8 @@ public class ImportService {
 
             throw new AtlasBaseException(excp);
         } finally {
+            RequestContext.get().setImportInProgress(false);
+
             source.close();
             LOG.info("<== import(user={}, from={}): status={}", userName, requestingIP, result.getOperationStatus());
         }