CommonUtil.h 1.7 KB
Newer Older
1
// Copyright (C) 2019-2020 Zilliz. All rights reserved.
J
jinhai 已提交
2
//
3 4
// 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
J
jinhai 已提交
5
//
6 7 8 9 10
// 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.
J
jinhai 已提交
11

G
groot 已提交
12 13
#pragma once

S
starlord 已提交
14 15
#include "utils/Status.h"

G
groot 已提交
16
#include <time.h>
S
starlord 已提交
17
#include <string>
G
groot 已提交
18

J
jinhai 已提交
19
namespace milvus {
G
groot 已提交
20 21 22 23
namespace server {

class CommonUtil {
 public:
S
starlord 已提交
24 25 26
    static bool
    GetSystemMemInfo(uint64_t& total_mem, uint64_t& free_mem);
    static bool
Y
yudong.cai 已提交
27
    GetSystemAvailableThreads(int64_t& thread_count);
G
groot 已提交
28

S
starlord 已提交
29 30 31 32 33 34 35 36 37 38
    static bool
    IsFileExist(const std::string& path);
    static uint64_t
    GetFileSize(const std::string& path);
    static bool
    IsDirectoryExist(const std::string& path);
    static Status
    CreateDirectory(const std::string& path);
    static Status
    DeleteDirectory(const std::string& path);
G
groot 已提交
39

S
starlord 已提交
40 41 42 43
    static std::string
    GetFileName(std::string filename);
    static std::string
    GetExePath();
G
groot 已提交
44

S
starlord 已提交
45 46 47
    static bool
    TimeStrToTime(const std::string& time_str, time_t& time_integer, tm& time_struct,
                  const std::string& format = "%d-%d-%d %d:%d:%d");
G
groot 已提交
48

S
starlord 已提交
49 50 51 52
    static void
    ConvertTime(time_t time_integer, tm& time_struct);
    static void
    ConvertTime(tm time_struct, time_t& time_integer);
53 54 55

    static void
    EraseFromCache(const std::string& item_key);
G
groot 已提交
56 57
};

S
starlord 已提交
58 59
}  // namespace server
}  // namespace milvus