mirror of https://github.com/grpc/grpc.git
ZhenLian
4 years ago
parent
486989f250
commit
d5d35650b0
23 changed files with 357 additions and 5 deletions
@ -0,0 +1,38 @@ |
|||||||
|
//
|
||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
//
|
||||||
|
|
||||||
|
#ifndef GRPC_CORE_LIB_GPRPP_STAT_H |
||||||
|
#define GRPC_CORE_LIB_GPRPP_STAT_H |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include <stdio.h> |
||||||
|
#include <time.h> |
||||||
|
|
||||||
|
#include "absl/status/status.h" |
||||||
|
#include "absl/strings/string_view.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
// Gets the last-modified timestamp of a file or a directory.
|
||||||
|
// On success, the correct timestamp will be filled with an StatusCode::kOk
|
||||||
|
// returned. Otherwise, timestamp will be untouched and an
|
||||||
|
// StatusCode::kInternal will be returned.
|
||||||
|
absl::Status GetFileModificationTime(const char* filename, time_t* timestamp); |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif // GRPC_CORE_LIB_GPRPP_STAT_H
|
||||||
@ -0,0 +1,49 @@ |
|||||||
|
//
|
||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
//
|
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#ifdef GPR_POSIX_STAT |
||||||
|
|
||||||
|
#include <errno.h> |
||||||
|
#include <sys/stat.h> |
||||||
|
#include <sys/types.h> |
||||||
|
#include <unistd.h> |
||||||
|
|
||||||
|
#include <grpc/support/log.h> |
||||||
|
|
||||||
|
#include "src/core/lib/gprpp/stat.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
absl::Status GetFileModificationTime(const char* filename, time_t* timestamp) { |
||||||
|
GPR_ASSERT(filename != nullptr); |
||||||
|
GPR_ASSERT(timestamp != nullptr); |
||||||
|
struct stat buf; |
||||||
|
if (stat(filename, &buf) != 0) { |
||||||
|
const char* error_msg = strerror(errno); |
||||||
|
gpr_log(GPR_ERROR, "stat failed for filename %s with error %s.", filename, |
||||||
|
error_msg); |
||||||
|
return absl::Status(absl::StatusCode::kInternal, error_msg); |
||||||
|
} |
||||||
|
// Last file/directory modification time.
|
||||||
|
*timestamp = buf.st_mtime; |
||||||
|
return absl::OkStatus(); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif // GPR_POSIX_STAT
|
||||||
@ -0,0 +1,48 @@ |
|||||||
|
//
|
||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
//
|
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#ifdef GPR_WINDOWS_STAT |
||||||
|
|
||||||
|
#include <errno.h> |
||||||
|
#include <sys/stat.h> |
||||||
|
#include <sys/types.h> |
||||||
|
|
||||||
|
#include <grpc/support/log.h> |
||||||
|
|
||||||
|
#include "src/core/lib/gprpp/stat.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
absl::Status GetFileModificationTime(const char* filename, time_t* timestamp) { |
||||||
|
GPR_ASSERT(filename != nullptr); |
||||||
|
GPR_ASSERT(timestamp != nullptr); |
||||||
|
struct _stat buf; |
||||||
|
if (_stat(filename, &buf) != 0) { |
||||||
|
const char* error_msg = strerror(errno); |
||||||
|
gpr_log(GPR_ERROR, "_stat failed for filename %s with error %s.", filename, |
||||||
|
error_msg); |
||||||
|
return absl::Status(absl::StatusCode::kInternal, error_msg); |
||||||
|
} |
||||||
|
// Last file/directory modification time.
|
||||||
|
*timestamp = buf.st_mtime; |
||||||
|
return absl::OkStatus(); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif // GPR_WINDOWS_STAT
|
||||||
@ -0,0 +1,74 @@ |
|||||||
|
//
|
||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
//
|
||||||
|
|
||||||
|
#include <stdio.h> |
||||||
|
#include <string.h> |
||||||
|
|
||||||
|
#include <gmock/gmock.h> |
||||||
|
#include <gtest/gtest.h> |
||||||
|
|
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <grpc/slice.h> |
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
|
||||||
|
#include "src/core/lib/gpr/string.h" |
||||||
|
#include "src/core/lib/gpr/tmpfile.h" |
||||||
|
#include "src/core/lib/gprpp/stat.h" |
||||||
|
#include "src/core/lib/iomgr/load_file.h" |
||||||
|
#include "test/core/util/test_config.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
namespace testing { |
||||||
|
namespace { |
||||||
|
|
||||||
|
TEST(STAT, GetTimestampOnTmpFile) { |
||||||
|
// Create a temporary empty file.
|
||||||
|
FILE* tmp = nullptr; |
||||||
|
char* tmp_name; |
||||||
|
tmp = gpr_tmpfile("prefix", &tmp_name); |
||||||
|
ASSERT_NE(tmp_name, nullptr); |
||||||
|
ASSERT_NE(tmp, nullptr); |
||||||
|
fclose(tmp); |
||||||
|
// Check the last modified date is correctly set.
|
||||||
|
time_t timestamp = 0; |
||||||
|
absl::Status status = |
||||||
|
grpc_core::GetFileModificationTime(tmp_name, ×tamp); |
||||||
|
EXPECT_EQ(status.code(), absl::StatusCode::kOk); |
||||||
|
EXPECT_GT(timestamp, 0); |
||||||
|
// Clean up.
|
||||||
|
remove(tmp_name); |
||||||
|
gpr_free(tmp_name); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(STAT, GetTimestampOnFailure) { |
||||||
|
time_t timestamp = 0; |
||||||
|
absl::Status status = |
||||||
|
grpc_core::GetFileModificationTime("/DOES_NOT_EXIST", ×tamp); |
||||||
|
EXPECT_EQ(status.code(), absl::StatusCode::kInternal); |
||||||
|
// Check the last modified date is not set.
|
||||||
|
EXPECT_EQ(timestamp, 0); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
grpc::testing::TestEnvironment env(argc, argv); |
||||||
|
::testing::InitGoogleTest(&argc, argv); |
||||||
|
return RUN_ALL_TESTS(); |
||||||
|
} |
||||||
Loading…
Reference in new issue