public AtlasServer getServer(String endpoint, HiveConf conf) throws SemanticException {
    Retryable retryable = Retryable.builder()
      .withHiveConf(conf)
      .withRetryOnException(AtlasServiceException.class).build();
    try {
      return retryable.executeCallable((Callable<AtlasServer>) () -> {
        try {
          return clientV2.getServer(endpoint);
        } catch (AtlasServiceException e) {
          int statusCode = e.getStatus() != null ? e.getStatus().getStatusCode() : -1;
          if (NOT_FOUND.getStatusCode() == statusCode) {
            // Atlas server entity is initialized on first import/export o/p.
            LOG.info("Atlas server entity is not found");
            return null;
          }
          throw e;
        }
      });
    } catch (Exception e) {
      throw new SemanticException(ErrorMsg.REPL_RETRY_EXHAUSTED.format(e.getMessage()), e);
    }
  }
Implementation of RESTClient, encapsulates Atlas' REST APIs.
ql/src/java/org/apache/hadoop/hive/ql/exec/repl/atlas/AtlasRestClientImpl.java