mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 15:05:31 +08:00
551be85d83
* Improve ut coverage (#2516) Signed-off-by: yhz <413554850@qq.com> * Improve ut coverage Signed-off-by: yhz <413554850@qq.com> * CI retry Signed-off-by: yhz <413554850@qq.com> * Update helm config Signed-off-by: JinHai-CN <hai.jin@zilliz.com> * CI retry Signed-off-by: Yhz <yinghao.zou@zilliz.com> * CI retry Signed-off-by: Yhz <yinghao.zou@zilliz.com> Co-authored-by: JinHai-CN <hai.jin@zilliz.com>
71 lines
2.2 KiB
C++
71 lines
2.2 KiB
C++
// Copyright (C) 2019-2020 Zilliz. All rights reserved.
|
|
//
|
|
// 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 "server/init/InstanceLockCheck.h"
|
|
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include <fiu-local.h>
|
|
|
|
#include "utils/Log.h"
|
|
|
|
namespace milvus {
|
|
namespace server {
|
|
|
|
Status
|
|
InstanceLockCheck::Check(const std::string& path) {
|
|
std::string lock_path = path + "/lock";
|
|
auto fd = open(lock_path.c_str(), O_RDWR | O_CREAT | O_NOFOLLOW, 0640);
|
|
fiu_do_on("InstanceLockCheck.Check.fd", fd = -1);
|
|
if (fd < 0) {
|
|
std::string msg;
|
|
if (errno == EROFS) {
|
|
// Not using locking for read-only lock file
|
|
msg += "Lock file is read-only.";
|
|
}
|
|
msg += "Could not open file: " + lock_path + ", " + strerror(errno);
|
|
return Status(SERVER_UNEXPECTED_ERROR, msg);
|
|
}
|
|
|
|
// Acquire a write lock
|
|
struct flock fl;
|
|
// exclusive lock
|
|
fl.l_type = F_WRLCK;
|
|
fl.l_whence = SEEK_SET;
|
|
fl.l_start = 0;
|
|
fl.l_len = 0;
|
|
auto fcl = fcntl(fd, F_SETLK, &fl);
|
|
fiu_do_on("InstanceLockCheck.Check.fcntl", fcl = -1);
|
|
if (fcl == -1) {
|
|
std::string msg = "Can't lock file: " + lock_path + ", due to ";
|
|
if (errno == EACCES || errno == EAGAIN) {
|
|
msg += "permission denied. ";
|
|
} else if (errno == ENOLCK) {
|
|
// Not using locking for nfs mounted lock file
|
|
msg += "using nfs. ";
|
|
} else {
|
|
msg += std::string(strerror(errno)) + ". ";
|
|
}
|
|
close(fd);
|
|
return Status(SERVER_UNEXPECTED_ERROR, msg);
|
|
}
|
|
|
|
LOG_SERVER_INFO_ << "InstanceLockCheck passed.";
|
|
|
|
return Status::OK();
|
|
}
|
|
|
|
} // namespace server
|
|
} // namespace milvus
|