diff --git a/cpp/CHANGELOG.md b/cpp/CHANGELOG.md index 09929ed8d0..5fb91c72d2 100644 --- a/cpp/CHANGELOG.md +++ b/cpp/CHANGELOG.md @@ -19,6 +19,7 @@ Please mark all change in change log and use the ticket from JIRA. - MS-566 - Refactor cmake - MS-555 - Remove old scheduler - MS-578 - Makesure milvus5.0 don't crack 0.3.1 data +- MS-585 - Update namespace in scheduler ## New Feature diff --git a/cpp/src/db/DBImpl.cpp b/cpp/src/db/DBImpl.cpp index 8323564024..a763c9199c 100644 --- a/cpp/src/db/DBImpl.cpp +++ b/cpp/src/db/DBImpl.cpp @@ -133,9 +133,9 @@ Status DBImpl::DeleteTable(const std::string& table_id, const meta::DatesT& date meta_ptr_->DeleteTable(table_id); //soft delete table //scheduler will determine when to delete table files - auto nres = ResMgrInst::GetInstance()->GetNumOfComputeResource(); + auto nres = scheduler::ResMgrInst::GetInstance()->GetNumOfComputeResource(); scheduler::DeleteJobPtr job = std::make_shared(0, table_id, meta_ptr_, nres); - JobMgrInst::GetInstance()->Put(job); + scheduler::JobMgrInst::GetInstance()->Put(job); job->WaitAndDelete(); } else { meta_ptr_->DropPartitionsByDates(table_id, dates); diff --git a/cpp/src/scheduler/Algorithm.cpp b/cpp/src/scheduler/Algorithm.cpp index 0a9aefbce0..b388c17bdd 100644 --- a/cpp/src/scheduler/Algorithm.cpp +++ b/cpp/src/scheduler/Algorithm.cpp @@ -20,7 +20,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { constexpr uint64_t MAXINT = std::numeric_limits::max(); diff --git a/cpp/src/scheduler/Algorithm.h b/cpp/src/scheduler/Algorithm.h index 763ef8ad5d..be77cf351f 100644 --- a/cpp/src/scheduler/Algorithm.h +++ b/cpp/src/scheduler/Algorithm.h @@ -24,7 +24,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { uint64_t ShortestPath(const ResourcePtr &src, diff --git a/cpp/src/scheduler/Definition.h b/cpp/src/scheduler/Definition.h new file mode 100644 index 0000000000..9ac647de88 --- /dev/null +++ b/cpp/src/scheduler/Definition.h @@ -0,0 +1,48 @@ +// 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. + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "db/meta/MetaTypes.h" +#include "db/engine/EngineFactory.h" +#include "db/engine/ExecutionEngine.h" + + +namespace zilliz { +namespace milvus { +namespace scheduler { + +using TableFileSchemaPtr = engine::meta::TableFileSchemaPtr; +using TableFileSchema = engine::meta::TableFileSchema; + +using ExecutionEnginePtr = engine::ExecutionEnginePtr; +using EngineFactory = engine::EngineFactory; +using EngineType = engine::EngineType; +using MetricType = engine::MetricType; + +} +} +} diff --git a/cpp/src/scheduler/JobMgr.h b/cpp/src/scheduler/JobMgr.h index 1d8720c4b4..8d22b5d4f5 100644 --- a/cpp/src/scheduler/JobMgr.h +++ b/cpp/src/scheduler/JobMgr.h @@ -36,9 +36,6 @@ namespace zilliz { namespace milvus { namespace scheduler { -using engine::TaskPtr; -using engine::ResourceMgrPtr; - class JobMgr { public: explicit diff --git a/cpp/src/scheduler/ResourceFactory.cpp b/cpp/src/scheduler/ResourceFactory.cpp index 40cd00f206..066b47aad7 100644 --- a/cpp/src/scheduler/ResourceFactory.cpp +++ b/cpp/src/scheduler/ResourceFactory.cpp @@ -21,7 +21,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::shared_ptr ResourceFactory::Create(const std::string &name, diff --git a/cpp/src/scheduler/ResourceFactory.h b/cpp/src/scheduler/ResourceFactory.h index 58a29497b0..39fb79154b 100644 --- a/cpp/src/scheduler/ResourceFactory.h +++ b/cpp/src/scheduler/ResourceFactory.h @@ -28,7 +28,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class ResourceFactory { public: diff --git a/cpp/src/scheduler/ResourceMgr.cpp b/cpp/src/scheduler/ResourceMgr.cpp index dafc15713a..0f9fab8e8d 100644 --- a/cpp/src/scheduler/ResourceMgr.cpp +++ b/cpp/src/scheduler/ResourceMgr.cpp @@ -22,7 +22,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { void diff --git a/cpp/src/scheduler/ResourceMgr.h b/cpp/src/scheduler/ResourceMgr.h index ebe0894770..8bd045ab7d 100644 --- a/cpp/src/scheduler/ResourceMgr.h +++ b/cpp/src/scheduler/ResourceMgr.h @@ -30,7 +30,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class ResourceMgr { public: diff --git a/cpp/src/scheduler/SchedInst.cpp b/cpp/src/scheduler/SchedInst.cpp index 3b14158e58..a13640073d 100644 --- a/cpp/src/scheduler/SchedInst.cpp +++ b/cpp/src/scheduler/SchedInst.cpp @@ -25,7 +25,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { ResourceMgrPtr ResMgrInst::instance = nullptr; std::mutex ResMgrInst::mutex_; diff --git a/cpp/src/scheduler/SchedInst.h b/cpp/src/scheduler/SchedInst.h index 010a7e84f5..d23c200b05 100644 --- a/cpp/src/scheduler/SchedInst.h +++ b/cpp/src/scheduler/SchedInst.h @@ -27,7 +27,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class ResMgrInst { public: diff --git a/cpp/src/scheduler/Scheduler.cpp b/cpp/src/scheduler/Scheduler.cpp index ff66e6ac21..abaccd4fcb 100644 --- a/cpp/src/scheduler/Scheduler.cpp +++ b/cpp/src/scheduler/Scheduler.cpp @@ -25,7 +25,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { Scheduler::Scheduler(ResourceMgrWPtr res_mgr) : running_(false), diff --git a/cpp/src/scheduler/Scheduler.h b/cpp/src/scheduler/Scheduler.h index feb1374a8b..b591fdd256 100644 --- a/cpp/src/scheduler/Scheduler.h +++ b/cpp/src/scheduler/Scheduler.h @@ -30,7 +30,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { // TODO: refactor, not friendly to unittest, logical in framework code diff --git a/cpp/src/scheduler/TaskCreator.cpp b/cpp/src/scheduler/TaskCreator.cpp index 1d471135d6..0d3b2a8a96 100644 --- a/cpp/src/scheduler/TaskCreator.cpp +++ b/cpp/src/scheduler/TaskCreator.cpp @@ -45,7 +45,7 @@ TaskCreator::Create(const SearchJobPtr &job) { std::vector tasks; for (auto &index_file : job->index_files()) { auto task = std::make_shared(index_file.second); - task->label() = std::make_shared(); + task->label() = std::make_shared(); task->job_ = job; tasks.emplace_back(task); } @@ -57,7 +57,7 @@ std::vector TaskCreator::Create(const DeleteJobPtr &job) { std::vector tasks; auto task = std::make_shared(job); - task->label() = std::make_shared(); + task->label() = std::make_shared(); task->job_ = job; tasks.emplace_back(task); diff --git a/cpp/src/scheduler/TaskCreator.h b/cpp/src/scheduler/TaskCreator.h index b0e947600d..d4e87bd0a2 100644 --- a/cpp/src/scheduler/TaskCreator.h +++ b/cpp/src/scheduler/TaskCreator.h @@ -39,10 +39,6 @@ namespace zilliz { namespace milvus { namespace scheduler { -using engine::TaskPtr; -using engine::XSearchTask; -using engine::XDeleteTask; - class TaskCreator { public: static std::vector diff --git a/cpp/src/scheduler/TaskTable.cpp b/cpp/src/scheduler/TaskTable.cpp index 1678665593..badcb46c2d 100644 --- a/cpp/src/scheduler/TaskTable.cpp +++ b/cpp/src/scheduler/TaskTable.cpp @@ -27,7 +27,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::string ToString(TaskTableItemState state) { diff --git a/cpp/src/scheduler/TaskTable.h b/cpp/src/scheduler/TaskTable.h index c4be442d11..c9557c8f9e 100644 --- a/cpp/src/scheduler/TaskTable.h +++ b/cpp/src/scheduler/TaskTable.h @@ -27,7 +27,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { enum class TaskTableItemState { INVALID, diff --git a/cpp/src/scheduler/Utils.cpp b/cpp/src/scheduler/Utils.cpp index b011751105..f207845c20 100644 --- a/cpp/src/scheduler/Utils.cpp +++ b/cpp/src/scheduler/Utils.cpp @@ -24,7 +24,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { uint64_t get_current_timestamp() { diff --git a/cpp/src/scheduler/Utils.h b/cpp/src/scheduler/Utils.h index 066695b35b..f1056f104a 100644 --- a/cpp/src/scheduler/Utils.h +++ b/cpp/src/scheduler/Utils.h @@ -21,7 +21,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { uint64_t get_current_timestamp(); diff --git a/cpp/src/scheduler/action/Action.h b/cpp/src/scheduler/action/Action.h index 57fc8fe686..9f2fa9f6c5 100644 --- a/cpp/src/scheduler/action/Action.h +++ b/cpp/src/scheduler/action/Action.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class Action { public: diff --git a/cpp/src/scheduler/action/PushTaskToNeighbour.cpp b/cpp/src/scheduler/action/PushTaskToNeighbour.cpp index 76beaa3e07..d53903ba2d 100644 --- a/cpp/src/scheduler/action/PushTaskToNeighbour.cpp +++ b/cpp/src/scheduler/action/PushTaskToNeighbour.cpp @@ -25,7 +25,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::vector get_neighbours(const ResourcePtr &self) { diff --git a/cpp/src/scheduler/event/Event.h b/cpp/src/scheduler/event/Event.h index af8be5ecf0..b00bd7f902 100644 --- a/cpp/src/scheduler/event/Event.h +++ b/cpp/src/scheduler/event/Event.h @@ -21,7 +21,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { enum class EventType { START_UP, diff --git a/cpp/src/scheduler/event/EventDump.cpp b/cpp/src/scheduler/event/EventDump.cpp index 384fd98e8d..3d06dfd5f5 100644 --- a/cpp/src/scheduler/event/EventDump.cpp +++ b/cpp/src/scheduler/event/EventDump.cpp @@ -25,7 +25,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::ostream &operator<<(std::ostream &out, const Event &event) { out << event.Dump(); diff --git a/cpp/src/scheduler/event/FinishTaskEvent.h b/cpp/src/scheduler/event/FinishTaskEvent.h index 3c88a62693..3ffa855f30 100644 --- a/cpp/src/scheduler/event/FinishTaskEvent.h +++ b/cpp/src/scheduler/event/FinishTaskEvent.h @@ -22,7 +22,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class FinishTaskEvent : public Event { public: diff --git a/cpp/src/scheduler/event/LoadCompletedEvent.h b/cpp/src/scheduler/event/LoadCompletedEvent.h index 7e7abaa697..286f26723f 100644 --- a/cpp/src/scheduler/event/LoadCompletedEvent.h +++ b/cpp/src/scheduler/event/LoadCompletedEvent.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class LoadCompletedEvent : public Event { public: diff --git a/cpp/src/scheduler/event/StartUpEvent.h b/cpp/src/scheduler/event/StartUpEvent.h index 0ae4db5ce0..089eb1377b 100644 --- a/cpp/src/scheduler/event/StartUpEvent.h +++ b/cpp/src/scheduler/event/StartUpEvent.h @@ -22,7 +22,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class StartUpEvent : public Event { public: diff --git a/cpp/src/scheduler/event/TaskTableUpdatedEvent.h b/cpp/src/scheduler/event/TaskTableUpdatedEvent.h index 766df9254d..70ee28fdab 100644 --- a/cpp/src/scheduler/event/TaskTableUpdatedEvent.h +++ b/cpp/src/scheduler/event/TaskTableUpdatedEvent.h @@ -22,7 +22,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class TaskTableUpdatedEvent : public Event { public: diff --git a/cpp/src/scheduler/resource/Connection.h b/cpp/src/scheduler/resource/Connection.h index ae7bf88e86..38fbc65bb5 100644 --- a/cpp/src/scheduler/resource/Connection.h +++ b/cpp/src/scheduler/resource/Connection.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class Connection { public: diff --git a/cpp/src/scheduler/resource/CpuResource.cpp b/cpp/src/scheduler/resource/CpuResource.cpp index 863a427096..3933bbb5a0 100644 --- a/cpp/src/scheduler/resource/CpuResource.cpp +++ b/cpp/src/scheduler/resource/CpuResource.cpp @@ -21,7 +21,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::ostream &operator<<(std::ostream &out, const CpuResource &resource) { out << resource.Dump(); diff --git a/cpp/src/scheduler/resource/CpuResource.h b/cpp/src/scheduler/resource/CpuResource.h index 19bac6090d..1f6f235621 100644 --- a/cpp/src/scheduler/resource/CpuResource.h +++ b/cpp/src/scheduler/resource/CpuResource.h @@ -24,7 +24,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class CpuResource : public Resource { public: diff --git a/cpp/src/scheduler/resource/DiskResource.cpp b/cpp/src/scheduler/resource/DiskResource.cpp index bb07f96b96..6495396fdb 100644 --- a/cpp/src/scheduler/resource/DiskResource.cpp +++ b/cpp/src/scheduler/resource/DiskResource.cpp @@ -20,7 +20,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::ostream &operator<<(std::ostream &out, const DiskResource &resource) { out << resource.Dump(); diff --git a/cpp/src/scheduler/resource/DiskResource.h b/cpp/src/scheduler/resource/DiskResource.h index 8c8a6ae46a..b9e7a4035c 100644 --- a/cpp/src/scheduler/resource/DiskResource.h +++ b/cpp/src/scheduler/resource/DiskResource.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class DiskResource : public Resource { public: diff --git a/cpp/src/scheduler/resource/GpuResource.cpp b/cpp/src/scheduler/resource/GpuResource.cpp index c8959cf393..c3981e65cb 100644 --- a/cpp/src/scheduler/resource/GpuResource.cpp +++ b/cpp/src/scheduler/resource/GpuResource.cpp @@ -21,7 +21,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::ostream &operator<<(std::ostream &out, const GpuResource &resource) { out << resource.Dump(); diff --git a/cpp/src/scheduler/resource/GpuResource.h b/cpp/src/scheduler/resource/GpuResource.h index 12947a35fa..f08e767c51 100644 --- a/cpp/src/scheduler/resource/GpuResource.h +++ b/cpp/src/scheduler/resource/GpuResource.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class GpuResource : public Resource { public: diff --git a/cpp/src/scheduler/resource/Node.cpp b/cpp/src/scheduler/resource/Node.cpp index d4c42a5a97..719bfb1427 100644 --- a/cpp/src/scheduler/resource/Node.cpp +++ b/cpp/src/scheduler/resource/Node.cpp @@ -22,7 +22,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { Node::Node() { static std::atomic_uint_fast8_t counter(0); diff --git a/cpp/src/scheduler/resource/Node.h b/cpp/src/scheduler/resource/Node.h index 311ec8156a..46506da10d 100644 --- a/cpp/src/scheduler/resource/Node.h +++ b/cpp/src/scheduler/resource/Node.h @@ -27,7 +27,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class Node; diff --git a/cpp/src/scheduler/resource/Resource.cpp b/cpp/src/scheduler/resource/Resource.cpp index b094bb1c06..22e7eb8502 100644 --- a/cpp/src/scheduler/resource/Resource.cpp +++ b/cpp/src/scheduler/resource/Resource.cpp @@ -22,7 +22,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::ostream & operator<<(std::ostream &out, const Resource &resource) { diff --git a/cpp/src/scheduler/resource/Resource.h b/cpp/src/scheduler/resource/Resource.h index f49ea71b5c..0b0eaae304 100644 --- a/cpp/src/scheduler/resource/Resource.h +++ b/cpp/src/scheduler/resource/Resource.h @@ -37,7 +37,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { // TODO(wxyu): Storage, Route, Executor enum class ResourceType { diff --git a/cpp/src/scheduler/resource/TestResource.cpp b/cpp/src/scheduler/resource/TestResource.cpp index ebdcb403bd..b49ea12d77 100644 --- a/cpp/src/scheduler/resource/TestResource.cpp +++ b/cpp/src/scheduler/resource/TestResource.cpp @@ -20,7 +20,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { std::ostream &operator<<(std::ostream &out, const TestResource &resource) { out << resource.Dump(); diff --git a/cpp/src/scheduler/resource/TestResource.h b/cpp/src/scheduler/resource/TestResource.h index b8f749ba58..861efbd825 100644 --- a/cpp/src/scheduler/resource/TestResource.h +++ b/cpp/src/scheduler/resource/TestResource.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class TestResource : public Resource { public: diff --git a/cpp/src/scheduler/task/DeleteTask.cpp b/cpp/src/scheduler/task/DeleteTask.cpp index b1b1697f0e..65a7a537fd 100644 --- a/cpp/src/scheduler/task/DeleteTask.cpp +++ b/cpp/src/scheduler/task/DeleteTask.cpp @@ -21,7 +21,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { XDeleteTask::XDeleteTask(const scheduler::DeleteJobPtr &delete_job) : Task(TaskType::DeleteTask), delete_job_(delete_job) {} diff --git a/cpp/src/scheduler/task/DeleteTask.h b/cpp/src/scheduler/task/DeleteTask.h index f21a1a96e4..b64f9a977f 100644 --- a/cpp/src/scheduler/task/DeleteTask.h +++ b/cpp/src/scheduler/task/DeleteTask.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class XDeleteTask : public Task { public: diff --git a/cpp/src/scheduler/task/Path.h b/cpp/src/scheduler/task/Path.h index 64df58b46c..a463494b77 100644 --- a/cpp/src/scheduler/task/Path.h +++ b/cpp/src/scheduler/task/Path.h @@ -23,7 +23,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class Path { public: diff --git a/cpp/src/scheduler/task/SearchTask.cpp b/cpp/src/scheduler/task/SearchTask.cpp index e39cd00d2e..284152f14e 100644 --- a/cpp/src/scheduler/task/SearchTask.cpp +++ b/cpp/src/scheduler/task/SearchTask.cpp @@ -27,7 +27,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { static constexpr size_t PARALLEL_REDUCE_THRESHOLD = 10000; static constexpr size_t PARALLEL_REDUCE_BATCH = 1000; @@ -79,8 +79,8 @@ std::mutex XSearchTask::merge_mutex_; void CollectFileMetrics(int file_type, size_t file_size) { switch (file_type) { - case meta::TableFileSchema::RAW: - case meta::TableFileSchema::TO_INDEX: { + case TableFileSchema::RAW: + case TableFileSchema::TO_INDEX: { server::Metrics::GetInstance().RawFileSizeHistogramObserve(file_size); server::Metrics::GetInstance().RawFileSizeTotalIncrement(file_size); server::Metrics::GetInstance().RawFileSizeGaugeSet(file_size); @@ -95,7 +95,7 @@ CollectFileMetrics(int file_type, size_t file_size) { } } -XSearchTask::XSearchTask(meta::TableFileSchemaPtr file) +XSearchTask::XSearchTask(TableFileSchemaPtr file) : Task(TaskType::SearchTask), file_(file) { if (file_) { index_engine_ = EngineFactory::Build(file_->dimension_, diff --git a/cpp/src/scheduler/task/SearchTask.h b/cpp/src/scheduler/task/SearchTask.h index e54e742e24..6f0ec6bc53 100644 --- a/cpp/src/scheduler/task/SearchTask.h +++ b/cpp/src/scheduler/task/SearchTask.h @@ -18,19 +18,19 @@ #pragma once #include "Task.h" -#include "db/meta/MetaTypes.h" #include "scheduler/job/SearchJob.h" +#include "scheduler/Definition.h" namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { // TODO: rewrite class XSearchTask : public Task { public: explicit - XSearchTask(meta::TableFileSchemaPtr file); + XSearchTask(TableFileSchemaPtr file); void Load(LoadType type, uint8_t device_id) override; @@ -56,7 +56,7 @@ public: scheduler::ResultSet &result_target); public: - meta::TableFileSchemaPtr file_; + TableFileSchemaPtr file_; size_t index_id_ = 0; int index_type_ = 0; diff --git a/cpp/src/scheduler/task/Task.h b/cpp/src/scheduler/task/Task.h index c4a2c1cb4c..785ef9cfb7 100644 --- a/cpp/src/scheduler/task/Task.h +++ b/cpp/src/scheduler/task/Task.h @@ -28,7 +28,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { enum class LoadType { DISK2CPU, diff --git a/cpp/src/scheduler/task/TestTask.cpp b/cpp/src/scheduler/task/TestTask.cpp index 5d5ac976b5..32e0a8587e 100644 --- a/cpp/src/scheduler/task/TestTask.cpp +++ b/cpp/src/scheduler/task/TestTask.cpp @@ -22,10 +22,9 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { - -TestTask::TestTask(meta::TableFileSchemaPtr &file) : XSearchTask(file) {} +TestTask::TestTask(TableFileSchemaPtr &file) : XSearchTask(file) {} void TestTask::Load(LoadType type, uint8_t device_id) { diff --git a/cpp/src/scheduler/task/TestTask.h b/cpp/src/scheduler/task/TestTask.h index e489fc537f..3367506d88 100644 --- a/cpp/src/scheduler/task/TestTask.h +++ b/cpp/src/scheduler/task/TestTask.h @@ -22,12 +22,12 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class TestTask : public XSearchTask { public: explicit - TestTask(meta::TableFileSchemaPtr& file); + TestTask(TableFileSchemaPtr& file); public: void diff --git a/cpp/src/scheduler/tasklabel/BroadcastLabel.h b/cpp/src/scheduler/tasklabel/BroadcastLabel.h index 2faa49acd1..1c89460df0 100644 --- a/cpp/src/scheduler/tasklabel/BroadcastLabel.h +++ b/cpp/src/scheduler/tasklabel/BroadcastLabel.h @@ -24,7 +24,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class BroadcastLabel : public TaskLabel { diff --git a/cpp/src/scheduler/tasklabel/DefaultLabel.h b/cpp/src/scheduler/tasklabel/DefaultLabel.h index d1619684b9..d61028117d 100644 --- a/cpp/src/scheduler/tasklabel/DefaultLabel.h +++ b/cpp/src/scheduler/tasklabel/DefaultLabel.h @@ -24,7 +24,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class DefaultLabel : public TaskLabel { public: diff --git a/cpp/src/scheduler/tasklabel/SpecResLabel.h b/cpp/src/scheduler/tasklabel/SpecResLabel.h index 3955625412..8e94a7677b 100644 --- a/cpp/src/scheduler/tasklabel/SpecResLabel.h +++ b/cpp/src/scheduler/tasklabel/SpecResLabel.h @@ -29,7 +29,7 @@ using ResourceWPtr = std::weak_ptr; namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class SpecResLabel : public TaskLabel { public: diff --git a/cpp/src/scheduler/tasklabel/TaskLabel.h b/cpp/src/scheduler/tasklabel/TaskLabel.h index da8049ebf9..5d43adb7db 100644 --- a/cpp/src/scheduler/tasklabel/TaskLabel.h +++ b/cpp/src/scheduler/tasklabel/TaskLabel.h @@ -21,7 +21,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { enum class TaskLabelType { DEFAULT, // means can be executed in any resource diff --git a/cpp/src/server/Server.cpp b/cpp/src/server/Server.cpp index da6d96e217..8dacf04d41 100644 --- a/cpp/src/server/Server.cpp +++ b/cpp/src/server/Server.cpp @@ -252,7 +252,7 @@ Server::LoadConfig() { void Server::StartService() { engine::KnowhereResource::Initialize(); - engine::StartSchedulerService(); + scheduler::StartSchedulerService(); DBWrapper::GetInstance().StartService(); grpc::GrpcServer::GetInstance().Start(); } @@ -261,7 +261,7 @@ void Server::StopService() { grpc::GrpcServer::GetInstance().Stop(); DBWrapper::GetInstance().StopService(); - engine::StopSchedulerService(); + scheduler::StopSchedulerService(); engine::KnowhereResource::Finalize(); } diff --git a/cpp/src/server/grpc_impl/GrpcRequestTask.cpp b/cpp/src/server/grpc_impl/GrpcRequestTask.cpp index ad0abd713d..0a8a84855d 100644 --- a/cpp/src/server/grpc_impl/GrpcRequestTask.cpp +++ b/cpp/src/server/grpc_impl/GrpcRequestTask.cpp @@ -772,7 +772,7 @@ CmdTask::OnExecute() { if (cmd_ == "version") { result_ = MILVUS_VERSION; } else if (cmd_ == "tasktable") { - result_ = engine::ResMgrInst::GetInstance()->DumpTaskTables(); + result_ = scheduler::ResMgrInst::GetInstance()->DumpTaskTables(); } else { result_ = "OK"; diff --git a/cpp/unittest/db/search_test.cpp b/cpp/unittest/db/search_test.cpp index 295650f2d2..b1c83c6db7 100644 --- a/cpp/unittest/db/search_test.cpp +++ b/cpp/unittest/db/search_test.cpp @@ -127,23 +127,23 @@ TEST(DBSearchTest, TOPK_TEST) { std::vector target_ids; std::vector target_distence; scheduler::ResultSet src_result; - auto status = engine::XSearchTask::ClusterResult(target_ids, target_distence, NQ, TOP_K, src_result); + auto status = scheduler::XSearchTask::ClusterResult(target_ids, target_distence, NQ, TOP_K, src_result); ASSERT_FALSE(status.ok()); ASSERT_TRUE(src_result.empty()); BuildResult(NQ, TOP_K, ascending, target_ids, target_distence); - status = engine::XSearchTask::ClusterResult(target_ids, target_distence, NQ, TOP_K, src_result); + status = scheduler::XSearchTask::ClusterResult(target_ids, target_distence, NQ, TOP_K, src_result); ASSERT_TRUE(status.ok()); ASSERT_EQ(src_result.size(), NQ); scheduler::ResultSet target_result; - status = engine::XSearchTask::TopkResult(target_result, TOP_K, ascending, target_result); + status = scheduler::XSearchTask::TopkResult(target_result, TOP_K, ascending, target_result); ASSERT_TRUE(status.ok()); - status = engine::XSearchTask::TopkResult(target_result, TOP_K, ascending, src_result); + status = scheduler::XSearchTask::TopkResult(target_result, TOP_K, ascending, src_result); ASSERT_FALSE(status.ok()); - status = engine::XSearchTask::TopkResult(src_result, TOP_K, ascending, target_result); + status = scheduler::XSearchTask::TopkResult(src_result, TOP_K, ascending, target_result); ASSERT_TRUE(status.ok()); ASSERT_TRUE(src_result.empty()); ASSERT_EQ(target_result.size(), NQ); @@ -153,10 +153,10 @@ TEST(DBSearchTest, TOPK_TEST) { uint64_t wrong_topk = TOP_K - 10; BuildResult(NQ, wrong_topk, ascending, src_ids, src_distence); - status = engine::XSearchTask::ClusterResult(src_ids, src_distence, NQ, wrong_topk, src_result); + status = scheduler::XSearchTask::ClusterResult(src_ids, src_distence, NQ, wrong_topk, src_result); ASSERT_TRUE(status.ok()); - status = engine::XSearchTask::TopkResult(src_result, TOP_K, ascending, target_result); + status = scheduler::XSearchTask::TopkResult(src_result, TOP_K, ascending, target_result); ASSERT_TRUE(status.ok()); for(uint64_t i = 0; i < NQ; i++) { ASSERT_EQ(target_result[i].size(), TOP_K); @@ -165,7 +165,7 @@ TEST(DBSearchTest, TOPK_TEST) { wrong_topk = TOP_K + 10; BuildResult(NQ, wrong_topk, ascending, src_ids, src_distence); - status = engine::XSearchTask::TopkResult(src_result, TOP_K, ascending, target_result); + status = scheduler::XSearchTask::TopkResult(src_result, TOP_K, ascending, target_result); ASSERT_TRUE(status.ok()); for(uint64_t i = 0; i < NQ; i++) { ASSERT_EQ(target_result[i].size(), TOP_K); @@ -183,15 +183,15 @@ TEST(DBSearchTest, MERGE_TEST) { uint64_t src_count = 5, target_count = 8; BuildResult(1, src_count, ascending, src_ids, src_distence); BuildResult(1, target_count, ascending, target_ids, target_distence); - auto status = engine::XSearchTask::ClusterResult(src_ids, src_distence, 1, src_count, src_result); + auto status = scheduler::XSearchTask::ClusterResult(src_ids, src_distence, 1, src_count, src_result); ASSERT_TRUE(status.ok()); - status = engine::XSearchTask::ClusterResult(target_ids, target_distence, 1, target_count, target_result); + status = scheduler::XSearchTask::ClusterResult(target_ids, target_distence, 1, target_count, target_result); ASSERT_TRUE(status.ok()); { scheduler::Id2DistanceMap src = src_result[0]; scheduler::Id2DistanceMap target = target_result[0]; - status = engine::XSearchTask::MergeResult(src, target, 10, ascending); + status = scheduler::XSearchTask::MergeResult(src, target, 10, ascending); ASSERT_TRUE(status.ok()); ASSERT_EQ(target.size(), 10); CheckResult(src_result[0], target_result[0], target, ascending); @@ -200,7 +200,7 @@ TEST(DBSearchTest, MERGE_TEST) { { scheduler::Id2DistanceMap src = src_result[0]; scheduler::Id2DistanceMap target; - status = engine::XSearchTask::MergeResult(src, target, 10, ascending); + status = scheduler::XSearchTask::MergeResult(src, target, 10, ascending); ASSERT_TRUE(status.ok()); ASSERT_EQ(target.size(), src_count); ASSERT_TRUE(src.empty()); @@ -210,7 +210,7 @@ TEST(DBSearchTest, MERGE_TEST) { { scheduler::Id2DistanceMap src = src_result[0]; scheduler::Id2DistanceMap target = target_result[0]; - status = engine::XSearchTask::MergeResult(src, target, 30, ascending); + status = scheduler::XSearchTask::MergeResult(src, target, 30, ascending); ASSERT_TRUE(status.ok()); ASSERT_EQ(target.size(), src_count + target_count); CheckResult(src_result[0], target_result[0], target, ascending); @@ -219,7 +219,7 @@ TEST(DBSearchTest, MERGE_TEST) { { scheduler::Id2DistanceMap target = src_result[0]; scheduler::Id2DistanceMap src = target_result[0]; - status = engine::XSearchTask::MergeResult(src, target, 30, ascending); + status = scheduler::XSearchTask::MergeResult(src, target, 30, ascending); ASSERT_TRUE(status.ok()); ASSERT_EQ(target.size(), src_count + target_count); CheckResult(src_result[0], target_result[0], target, ascending); @@ -242,7 +242,7 @@ TEST(DBSearchTest, PARALLEL_CLUSTER_TEST) { BuildResult(nq, topk, ascending, target_ids, target_distence); rc.RecordSection("build id/dietance map"); - auto status = engine::XSearchTask::ClusterResult(target_ids, target_distence, nq, topk, src_result); + auto status = scheduler::XSearchTask::ClusterResult(target_ids, target_distence, nq, topk, src_result); ASSERT_TRUE(status.ok()); ASSERT_EQ(src_result.size(), nq); @@ -282,14 +282,14 @@ TEST(DBSearchTest, PARALLEL_TOPK_TEST) { TimeRecorder rc("DoCluster"); BuildResult(nq, topk, ascending, target_ids, target_distence); - auto status = engine::XSearchTask::ClusterResult(target_ids, target_distence, nq, topk, src_result); + auto status = scheduler::XSearchTask::ClusterResult(target_ids, target_distence, nq, topk, src_result); rc.RecordSection("cluster result"); BuildResult(nq, insufficient_topk, ascending, insufficient_ids, insufficient_distence); - status = engine::XSearchTask::ClusterResult(target_ids, target_distence, nq, insufficient_topk, insufficient_result); + status = scheduler::XSearchTask::ClusterResult(target_ids, target_distence, nq, insufficient_topk, insufficient_result); rc.RecordSection("cluster result"); - engine::XSearchTask::TopkResult(insufficient_result, topk, ascending, src_result); + scheduler::XSearchTask::TopkResult(insufficient_result, topk, ascending, src_result); ASSERT_TRUE(status.ok()); rc.RecordSection("topk"); diff --git a/cpp/unittest/db/utils.cpp b/cpp/unittest/db/utils.cpp index 0fb47df13c..08e3c8f090 100644 --- a/cpp/unittest/db/utils.cpp +++ b/cpp/unittest/db/utils.cpp @@ -80,20 +80,20 @@ engine::DBOptions BaseTest::GetOptions() { void DBTest::SetUp() { BaseTest::SetUp(); - auto res_mgr = engine::ResMgrInst::GetInstance(); + auto res_mgr = scheduler::ResMgrInst::GetInstance(); res_mgr->Clear(); - res_mgr->Add(engine::ResourceFactory::Create("disk", "DISK", 0, true, false)); - res_mgr->Add(engine::ResourceFactory::Create("cpu", "CPU", 0, true, false)); - res_mgr->Add(engine::ResourceFactory::Create("gtx1660", "GPU", 0, true, true)); + res_mgr->Add(scheduler::ResourceFactory::Create("disk", "DISK", 0, true, false)); + res_mgr->Add(scheduler::ResourceFactory::Create("cpu", "CPU", 0, true, false)); + res_mgr->Add(scheduler::ResourceFactory::Create("gtx1660", "GPU", 0, true, true)); - auto default_conn = engine::Connection("IO", 500.0); - auto PCIE = engine::Connection("IO", 11000.0); + auto default_conn = scheduler::Connection("IO", 500.0); + auto PCIE = scheduler::Connection("IO", 11000.0); res_mgr->Connect("disk", "cpu", default_conn); res_mgr->Connect("cpu", "gtx1660", PCIE); res_mgr->Start(); - engine::SchedInst::GetInstance()->Start(); + scheduler::SchedInst::GetInstance()->Start(); - engine::JobMgrInst::GetInstance()->Start(); + scheduler::JobMgrInst::GetInstance()->Start(); auto options = GetOptions(); db_ = engine::DBFactory::Build(options); @@ -103,9 +103,9 @@ void DBTest::TearDown() { db_->Stop(); db_->DropAll(); - engine::JobMgrInst::GetInstance()->Stop(); - engine::SchedInst::GetInstance()->Stop(); - engine::ResMgrInst::GetInstance()->Stop(); + scheduler::JobMgrInst::GetInstance()->Stop(); + scheduler::SchedInst::GetInstance()->Stop(); + scheduler::ResMgrInst::GetInstance()->Stop(); BaseTest::TearDown(); diff --git a/cpp/unittest/scheduler/algorithm_test.cpp b/cpp/unittest/scheduler/algorithm_test.cpp index 5272edf0d7..fe619c4981 100644 --- a/cpp/unittest/scheduler/algorithm_test.cpp +++ b/cpp/unittest/scheduler/algorithm_test.cpp @@ -26,7 +26,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class AlgorithmTest : public testing::Test { protected: diff --git a/cpp/unittest/scheduler/event_test.cpp b/cpp/unittest/scheduler/event_test.cpp index 895aabe19d..34d4c2ce23 100644 --- a/cpp/unittest/scheduler/event_test.cpp +++ b/cpp/unittest/scheduler/event_test.cpp @@ -25,7 +25,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { TEST(EventTest, START_UP_EVENT) { ResourceWPtr res(ResourcePtr(nullptr)); diff --git a/cpp/unittest/scheduler/node_test.cpp b/cpp/unittest/scheduler/node_test.cpp index 36dd10b73d..b87611d3ca 100644 --- a/cpp/unittest/scheduler/node_test.cpp +++ b/cpp/unittest/scheduler/node_test.cpp @@ -20,7 +20,7 @@ #include -using namespace zilliz::milvus::engine; +using namespace zilliz::milvus::scheduler; class NodeTest : public ::testing::Test { protected: diff --git a/cpp/unittest/scheduler/normal_test.cpp b/cpp/unittest/scheduler/normal_test.cpp index 02ed3dae69..535fa96731 100644 --- a/cpp/unittest/scheduler/normal_test.cpp +++ b/cpp/unittest/scheduler/normal_test.cpp @@ -26,7 +26,7 @@ #include -using namespace zilliz::milvus::engine; +using namespace zilliz::milvus::scheduler; TEST(NormalTest, INST_TEST) { @@ -46,7 +46,7 @@ TEST(NormalTest, INST_TEST) { const uint64_t NUM_TASK = 1000; std::vector> tasks; - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; auto disks = res_mgr->GetDiskResources(); ASSERT_FALSE(disks.empty()); diff --git a/cpp/unittest/scheduler/resource_factory_test.cpp b/cpp/unittest/scheduler/resource_factory_test.cpp index 1bf39f669f..e2a84257eb 100644 --- a/cpp/unittest/scheduler/resource_factory_test.cpp +++ b/cpp/unittest/scheduler/resource_factory_test.cpp @@ -20,7 +20,7 @@ #include -using namespace zilliz::milvus::engine; +using namespace zilliz::milvus::scheduler; TEST(ResourceFactoryTest, CREATE) { auto disk = ResourceFactory::Create("ssd", "DISK", 0); diff --git a/cpp/unittest/scheduler/resource_mgr_test.cpp b/cpp/unittest/scheduler/resource_mgr_test.cpp index 8aaef5bdf5..c2be785a00 100644 --- a/cpp/unittest/scheduler/resource_mgr_test.cpp +++ b/cpp/unittest/scheduler/resource_mgr_test.cpp @@ -27,7 +27,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { /************ ResourceMgrBaseTest ************/ @@ -187,7 +187,7 @@ TEST_F(ResourceMgrAdvanceTest, REGISTER_SUBSCRIBER) { flag = true; }; mgr1_->RegisterSubscriber(callback); - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; disk_res->task_table().Put(std::make_shared(dummy)); sleep(1); ASSERT_TRUE(flag); diff --git a/cpp/unittest/scheduler/resource_test.cpp b/cpp/unittest/scheduler/resource_test.cpp index 1fdeb13062..b335a601db 100644 --- a/cpp/unittest/scheduler/resource_test.cpp +++ b/cpp/unittest/scheduler/resource_test.cpp @@ -29,7 +29,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { /************ ResourceBaseTest ************/ class ResourceBaseTest : public testing::Test { @@ -180,7 +180,7 @@ protected: TEST_F(ResourceAdvanceTest, DISK_RESOURCE_TEST) { const uint64_t NUM = 100; std::vector> tasks; - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; for (uint64_t i = 0; i < NUM; ++i) { auto task = std::make_shared(dummy); tasks.push_back(task); @@ -205,7 +205,7 @@ TEST_F(ResourceAdvanceTest, DISK_RESOURCE_TEST) { TEST_F(ResourceAdvanceTest, CPU_RESOURCE_TEST) { const uint64_t NUM = 100; std::vector> tasks; - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; for (uint64_t i = 0; i < NUM; ++i) { auto task = std::make_shared(dummy); tasks.push_back(task); @@ -230,7 +230,7 @@ TEST_F(ResourceAdvanceTest, CPU_RESOURCE_TEST) { TEST_F(ResourceAdvanceTest, GPU_RESOURCE_TEST) { const uint64_t NUM = 100; std::vector> tasks; - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; for (uint64_t i = 0; i < NUM; ++i) { auto task = std::make_shared(dummy); tasks.push_back(task); @@ -255,7 +255,7 @@ TEST_F(ResourceAdvanceTest, GPU_RESOURCE_TEST) { TEST_F(ResourceAdvanceTest, TEST_RESOURCE_TEST) { const uint64_t NUM = 100; std::vector> tasks; - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; for (uint64_t i = 0; i < NUM; ++i) { auto task = std::make_shared(dummy); tasks.push_back(task); diff --git a/cpp/unittest/scheduler/schedinst_test.cpp b/cpp/unittest/scheduler/schedinst_test.cpp index 27b6a17bbb..26ecce9f64 100644 --- a/cpp/unittest/scheduler/schedinst_test.cpp +++ b/cpp/unittest/scheduler/schedinst_test.cpp @@ -24,7 +24,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class SchedInstTest : public testing::Test { diff --git a/cpp/unittest/scheduler/scheduler_test.cpp b/cpp/unittest/scheduler/scheduler_test.cpp index ebe3a1f20d..dd2c8e8cb9 100644 --- a/cpp/unittest/scheduler/scheduler_test.cpp +++ b/cpp/unittest/scheduler/scheduler_test.cpp @@ -15,22 +15,23 @@ // specific language governing permissions and limitations // under the License. -#include "scheduler/Scheduler.h" #include -#include "src/scheduler/tasklabel/DefaultLabel.h" -#include "cache/DataObj.h" -#include "cache/GpuCacheMgr.h" + +#include "scheduler/Scheduler.h" +#include "scheduler/tasklabel/DefaultLabel.h" +#include "scheduler/tasklabel/SpecResLabel.h" #include "scheduler/task/TestTask.h" #include "scheduler/ResourceFactory.h" #include "scheduler/resource/Resource.h" +#include "cache/DataObj.h" +#include "cache/GpuCacheMgr.h" #include "utils/Error.h" -#include "src/wrapper/vec_index.h" -#include "scheduler/tasklabel/SpecResLabel.h" +#include "wrapper/vec_index.h" namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { class MockVecIndex : public engine::VecIndex { public: @@ -155,7 +156,7 @@ insert_dummy_index_into_gpu_cache(uint64_t device_id) { TEST_F(SchedulerTest, ON_LOAD_COMPLETED) { const uint64_t NUM = 10; std::vector> tasks; - meta::TableFileSchemaPtr dummy = std::make_shared(); + TableFileSchemaPtr dummy = std::make_shared(); dummy->location_ = "location"; insert_dummy_index_into_gpu_cache(1); @@ -175,7 +176,7 @@ TEST_F(SchedulerTest, ON_LOAD_COMPLETED) { TEST_F(SchedulerTest, PUSH_TASK_TO_NEIGHBOUR_RANDOMLY_TEST) { const uint64_t NUM = 10; std::vector> tasks; - meta::TableFileSchemaPtr dummy1 = std::make_shared(); + TableFileSchemaPtr dummy1 = std::make_shared(); dummy1->location_ = "location"; tasks.clear(); @@ -246,7 +247,7 @@ protected: TEST_F(SchedulerTest2, SPECIFIED_RESOURCE_TEST) { const uint64_t NUM = 10; std::vector> tasks; - meta::TableFileSchemaPtr dummy = std::make_shared(); + TableFileSchemaPtr dummy = std::make_shared(); dummy->location_ = "location"; for (uint64_t i = 0; i < NUM; ++i) { diff --git a/cpp/unittest/scheduler/task_test.cpp b/cpp/unittest/scheduler/task_test.cpp index 8a8b455ae5..17d1e28cd6 100644 --- a/cpp/unittest/scheduler/task_test.cpp +++ b/cpp/unittest/scheduler/task_test.cpp @@ -22,7 +22,7 @@ namespace zilliz { namespace milvus { -namespace engine { +namespace scheduler { TEST(TaskTest, INVALID_INDEX) { diff --git a/cpp/unittest/scheduler/tasktable_test.cpp b/cpp/unittest/scheduler/tasktable_test.cpp index 0c3f65e146..5c64a72351 100644 --- a/cpp/unittest/scheduler/tasktable_test.cpp +++ b/cpp/unittest/scheduler/tasktable_test.cpp @@ -21,7 +21,7 @@ #include -using namespace zilliz::milvus::engine; +using namespace zilliz::milvus::scheduler; /************ TaskTableBaseTest ************/ @@ -169,7 +169,7 @@ class TaskTableBaseTest : public ::testing::Test { protected: void SetUp() override { - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; invalid_task_ = nullptr; task1_ = std::make_shared(dummy); task2_ = std::make_shared(dummy); @@ -339,7 +339,7 @@ class TaskTableAdvanceTest : public ::testing::Test { protected: void SetUp() override { - meta::TableFileSchemaPtr dummy = nullptr; + TableFileSchemaPtr dummy = nullptr; for (uint64_t i = 0; i < 8; ++i) { auto task = std::make_shared(dummy); table1_.Put(task);