xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow cache server to run inside a docker env

This commit is contained in:
Lixia Chen 2021-05-25 14:42:55 -04:00
parent 05bbd1d8fa
commit 029e0b52d7
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
mindspore/ccsrc/minddata/dataset/engine/cache/cache_common.h vendored
View File

@ -69,15 +69,27 @@ inline void Status2CacheReply(const Status &rc, CacheReply *reply) {
reply->set_rc(static_cast<int32_t>(rc.StatusCode())); reply->set_rc(static_cast<int32_t>(rc.StatusCode()));
reply->set_msg(rc.ToString()); reply->set_msg(rc.ToString());
} }
/// Return the default cache path for a user
inline std::string DefaultUserDir() {
#if !defined(_WIN32) && !defined(_WIN64) && !defined(__ANDROID__) && !defined(ANDROID) && !defined(__APPLE__)
char user[LOGIN_NAME_MAX];
int rc = getlogin_r(user, sizeof(user));
if (rc == 0) {
return kDefaultCommonPath + std::string(user);
} else {
return kDefaultCommonPath;
}
#else
return kDefaultCommonPath;
#endif
}
/// \brief Generate the unix socket file we use on both client/server side given a tcp/ip port number /// \brief Generate the unix socket file we use on both client/server side given a tcp/ip port number
/// \param port /// \param port
/// \return unix socket url /// \return unix socket url
inline std::string PortToUnixSocketPath(int port) { inline std::string PortToUnixSocketPath(int port) {
#if !defined(_WIN32) && !defined(_WIN64) && !defined(__ANDROID__) && !defined(ANDROID) && !defined(__APPLE__) return DefaultUserDir() + std::string("/cache_server_p") + std::to_string(port);
return kDefaultCommonPath + Services::GetUserName() + std::string("/cache_server_p") + std::to_string(port);
#else
return kDefaultCommonPath;
#endif
} }
/// \brief Round up to the next 4k /// \brief Round up to the next 4k
@ -96,13 +108,7 @@ using numa_id_t = int32_t;
using cpu_id_t = int32_t; using cpu_id_t = int32_t;
/// Return the default log dir for cache /// Return the default log dir for cache
inline std::string DefaultLogDir() { inline std::string DefaultLogDir() { return DefaultUserDir() + std::string("/cache/log"); }
#if !defined(_WIN32) && !defined(_WIN64) && !defined(__ANDROID__) && !defined(ANDROID) && !defined(__APPLE__)
return kDefaultCommonPath + Services::GetUserName() + std::string("/cache/log");
#else
return kDefaultCommonPath;
#endif
}
} // namespace dataset } // namespace dataset
} // namespace mindspore } // namespace mindspore
#endif // MINDSPORE_CCSRC_MINDDATA_DATASET_ENGINE_CACHE_COMMON_H_ #endif // MINDSPORE_CCSRC_MINDDATA_DATASET_ENGINE_CACHE_COMMON_H_