mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
32 lines
1 KiB
C++
32 lines
1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
#include "test_server_environment.h"
|
|
#include <spdlog/spdlog.h>
|
|
#include <spdlog/sinks/sink.h>
|
|
#include <spdlog/sinks/stdout_sinks.h>
|
|
#include <assert.h>
|
|
|
|
namespace onnxruntime {
|
|
namespace server {
|
|
namespace test {
|
|
static std::unique_ptr<onnxruntime::server::ServerEnvironment> s_env;
|
|
|
|
ServerEnvironment* ServerEnv() {
|
|
assert(s_env != nullptr);
|
|
return s_env.get();
|
|
}
|
|
|
|
TestServerEnvironment::TestServerEnvironment() {
|
|
auto console = spdlog::stdout_logger_mt("console");
|
|
spdlog::set_default_logger(console);
|
|
spdlog::sink_ptr ptr = std::make_shared<spdlog::sinks::stdout_sink_st>();
|
|
s_env = std::make_unique<onnxruntime::server::ServerEnvironment>(ORT_LOGGING_LEVEL_WARNING, spdlog::sinks_init_list{ptr});
|
|
}
|
|
TestServerEnvironment::~TestServerEnvironment() {
|
|
//destruct env to make sure the default logger is destoryed before the logger mutex.
|
|
s_env = nullptr;
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace server
|
|
} // namespace onnxruntime
|