Server.cpp 5.3 KB
Newer Older
G
groot 已提交
1 2 3 4 5 6
////////////////////////////////////////////////////////////////////////////////
// Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
// Unauthorized copying of this file, via any medium is strictly prohibited.
// Proprietary and confidential.
////////////////////////////////////////////////////////////////////////////////
#include "Server.h"
G
groot 已提交
7
#include "ServerConfig.h"
G
groot 已提交
8
#include "VecServiceWrapper.h"
G
groot 已提交
9
#include "utils/Log.h"
G
groot 已提交
10 11
#include "utils/SignalUtil.h"
#include "utils/TimeRecorder.h"
J
jinhai 已提交
12
#include "license/License.h"
G
groot 已提交
13

G
groot 已提交
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <csignal>
#include <numaif.h>
#include <unistd.h>
#include <string.h>

namespace zilliz {
namespace vecwise {
namespace server {

Server*
Server::Instance() {
    static Server server;
    return &server;
}

G
groot 已提交
32
Server::Server() {
G
groot 已提交
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51

}
Server::~Server() {

}

void
Server::Init(int64_t daemonized, const std::string& pid_filename, const std::string& config_filename) {
    daemonized_ = daemonized;
    pid_filename_ = pid_filename;
    config_filename_ = config_filename;
}

void
Server::Daemonize() {
    if (daemonized_ == 0) {
        return;
    }

G
groot 已提交
52
    SERVER_LOG_INFO << "Vecwise server run in daemonize mode";
G
groot 已提交
53 54 55

//    std::string log_path(GetLogDirFullPath());
//    log_path += "zdb_server.(INFO/WARNNING/ERROR/CRITICAL)";
G
groot 已提交
56
//    SERVER_LOG_INFO << "Log will be exported to: " + log_path);
G
groot 已提交
57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107

    pid_t pid = 0;

    // Fork off the parent process
    pid = fork();

    // An error occurred
    if (pid < 0) {
        exit(EXIT_FAILURE);
    }

    // Success: terminate parent
    if (pid > 0) {
        exit(EXIT_SUCCESS);
    }

    // On success: The child process becomes session leader
    if (setsid() < 0) {
        exit(EXIT_FAILURE);
    }

    // Ignore signal sent from child to parent process
    signal(SIGCHLD, SIG_IGN);

    // Fork off for the second time
    pid = fork();

    // An error occurred
    if (pid < 0) {
        exit(EXIT_FAILURE);
    }

    // Terminate the parent
    if (pid > 0) {
        exit(EXIT_SUCCESS);
    }

    // Set new file permissions
    umask(0);

    // Change the working directory to root
    int ret = chdir("/");
    if(ret != 0){
        return;
    }

    // Close all open fd
    for (long fd = sysconf(_SC_OPEN_MAX); fd > 0; fd--) {
        close(fd);
    }

G
groot 已提交
108
    SERVER_LOG_INFO << "Redirect stdin/stdout/stderr to /dev/null";
G
groot 已提交
109 110 111 112 113 114 115 116 117

    // Redirect stdin/stdout/stderr to /dev/null
    stdin = fopen("/dev/null", "r");
    stdout = fopen("/dev/null", "w+");
    stderr = fopen("/dev/null", "w+");
    // Try to write PID of daemon to lockfile
    if (!pid_filename_.empty()) {
        pid_fd = open(pid_filename_.c_str(), O_RDWR | O_CREAT, 0640);
        if (pid_fd < 0) {
G
groot 已提交
118
            SERVER_LOG_INFO << "Can't open filename: " + pid_filename_ + ", Error: " + strerror(errno);
G
groot 已提交
119 120 121
            exit(EXIT_FAILURE);
        }
        if (lockf(pid_fd, F_TLOCK, 0) < 0) {
G
groot 已提交
122
            SERVER_LOG_INFO << "Can't lock filename: " + pid_filename_ + ", Error: " + strerror(errno);
G
groot 已提交
123 124 125 126 127 128 129 130 131 132 133 134 135
            exit(EXIT_FAILURE);
        }

        std::string pid_file_context = std::to_string(getpid());
        ssize_t res = write(pid_fd, pid_file_context.c_str(), pid_file_context.size());
        if(res != 0){
            return;
        }
    }
}

int
Server::Start() {
J
jinhai 已提交
136 137 138 139 140 141
    std::string license_file_path = "/tmp/vecwise.license";
    if(LicenseValidate(license_file_path) != SERVER_SUCCESS) {
        SERVER_LOG_ERROR << "License check failed";
        return 1;
    }

G
groot 已提交
142 143 144 145 146 147 148 149 150 151 152 153
    if (daemonized_) {
        Daemonize();
    }

    do {
        try {
            // Read config file
            if(LoadConfig() != SERVER_SUCCESS) {
                return 1;
            }

            //log path is defined by LoadConfig, so InitLog must be called after LoadConfig
G
groot 已提交
154 155
            ServerConfig &config = ServerConfig::GetInstance();
            ConfigNode server_config = config.GetConfig(CONFIG_SERVER);
G
groot 已提交
156 157

            //print config into console and log
G
groot 已提交
158
            config.PrintAll();
G
groot 已提交
159 160 161 162 163 164

            // Handle Signal
            signal(SIGINT, SignalUtil::HandleSignal);
            signal(SIGHUP, SignalUtil::HandleSignal);
            signal(SIGTERM, SignalUtil::HandleSignal);

G
groot 已提交
165
            SERVER_LOG_INFO << "Vecwise server is running...";
G
groot 已提交
166
            StartService();
G
groot 已提交
167 168

        } catch(std::exception& ex){
G
groot 已提交
169 170
            SERVER_LOG_ERROR << "Vecwise server encounter exception: " << std::string(ex.what())
                             << "Is another server instance running?";
G
groot 已提交
171 172 173 174 175 176 177 178 179 180
            break;
        }
    } while(false);

    Stop();
    return 0;
}

void
Server::Stop() {
G
groot 已提交
181
    SERVER_LOG_INFO << "Vecwise server will be closed";
G
groot 已提交
182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207

    // Unlock and close lockfile
    if (pid_fd != -1) {
        int ret = lockf(pid_fd, F_ULOCK, 0);
        if(ret != 0){

        }
        ret = close(pid_fd);
        if(ret != 0){

        }
    }

    // Try to delete lockfile
    if (!pid_filename_.empty()) {
        int ret = unlink(pid_filename_.c_str());
        if(ret != 0){

        }
    }

    running_ = 0;

    StopService();


G
groot 已提交
208
    SERVER_LOG_INFO << "Vecwise server closed";
G
groot 已提交
209 210 211 212 213
}


ServerError
Server::LoadConfig() {
G
groot 已提交
214
    ServerConfig::GetInstance().LoadConfigFile(config_filename_);
G
groot 已提交
215 216 217 218 219 220

    return SERVER_SUCCESS;
}

void
Server::StartService() {
G
groot 已提交
221
    VecServiceWrapper::StartService();
G
groot 已提交
222 223 224 225
}

void
Server::StopService() {
G
groot 已提交
226
    VecServiceWrapper::StopService();
G
groot 已提交
227 228 229 230 231
}

}
}
}