From c49c73559a7aa3bec8b0d4c42dd23d987ac71862 Mon Sep 17 00:00:00 2001 From: yuanbopeng Date: Wed, 21 Oct 2020 12:37:35 +0800 Subject: [PATCH] Fix when hdfsBuilderConnect returns NULL, HadoopFileSystem::FileExists returns errors::NotFound exception --- .../experimental/filesystem/plugins/hadoop/hadoop_filesystem.cc | 2 +- tensorflow/core/platform/hadoop/hadoop_file_system.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/c/experimental/filesystem/plugins/hadoop/hadoop_filesystem.cc b/tensorflow/c/experimental/filesystem/plugins/hadoop/hadoop_filesystem.cc index 5ff28e4229a..0981bdd7f91 100644 --- a/tensorflow/c/experimental/filesystem/plugins/hadoop/hadoop_filesystem.cc +++ b/tensorflow/c/experimental/filesystem/plugins/hadoop/hadoop_filesystem.cc @@ -217,7 +217,7 @@ hdfsFS Connect(tf_hadoop_filesystem::HadoopFile* hadoop_file, hadoop_file->connection_cache.end()) { auto cacheFs = libhdfs->hdfsBuilderConnect(builder); if (cacheFs == nullptr) { - TF_SetStatusFromIOError(status, TF_NOT_FOUND, strerror(errno)); + TF_SetStatusFromIOError(status, TF_ABORTED, strerror(errno)); return cacheFs; } hadoop_file->connection_cache[cacheKey] = cacheFs; diff --git a/tensorflow/core/platform/hadoop/hadoop_file_system.cc b/tensorflow/core/platform/hadoop/hadoop_file_system.cc index 74195db7730..a5db01087b3 100644 --- a/tensorflow/core/platform/hadoop/hadoop_file_system.cc +++ b/tensorflow/core/platform/hadoop/hadoop_file_system.cc @@ -196,7 +196,7 @@ Status HadoopFileSystem::Connect(StringPiece fname, hdfsFS* fs) { if (connectionCache_.find(cacheKey) == connectionCache_.end()) { hdfsFS cacheFs = libhdfs()->hdfsBuilderConnect(builder); if (cacheFs == nullptr) { - return errors::NotFound(strerror(errno)); + return errors::Aborted(strerror(errno)); } connectionCache_[cacheKey] = cacheFs; }