CommonUtil.h 1.9 KB
Newer Older
J
jinhai 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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.

G
groot 已提交
18 19
#pragma once

S
starlord 已提交
20 21
#include "utils/Status.h"

G
groot 已提交
22
#include <time.h>
S
starlord 已提交
23
#include <string>
G
groot 已提交
24

J
jinhai 已提交
25
namespace milvus {
G
groot 已提交
26 27 28 29
namespace server {

class CommonUtil {
 public:
S
starlord 已提交
30 31 32
    static bool
    GetSystemMemInfo(uint64_t& total_mem, uint64_t& free_mem);
    static bool
Y
yudong.cai 已提交
33
    GetSystemAvailableThreads(int64_t& thread_count);
G
groot 已提交
34

S
starlord 已提交
35 36 37 38 39 40 41 42 43 44
    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 已提交
45

S
starlord 已提交
46 47 48 49
    static std::string
    GetFileName(std::string filename);
    static std::string
    GetExePath();
G
groot 已提交
50

S
starlord 已提交
51 52 53
    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 已提交
54

S
starlord 已提交
55 56 57 58
    static void
    ConvertTime(time_t time_integer, tm& time_struct);
    static void
    ConvertTime(tm time_struct, time_t& time_integer);
59 60 61

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

S
starlord 已提交
64 65
}  // namespace server
}  // namespace milvus