#pragma once

#include "Status.h"
#include "Meta.h"
#include "IDGenerator.h"

namespace zilliz {
namespace milvus {
namespace engine {

class VectorSource {

public:

    using Ptr = std::shared_ptr<VectorSource>;

    VectorSource(const size_t& n, const float* vectors);

    Status Add(const meta::TableFileSchema& table_file_schema, const size_t& num_vectors_to_add, size_t& num_vectors_added);

    size_t GetNumVectorsAdded();

    bool AllAdded();

    IDNumbers GetVectorIds();

private:

    const size_t n_;
    const float* vectors_;
    IDNumbers vector_ids_;

    size_t current_num_vectors_added;

    IDGenerator* id_generator_;

}; //VectorSource

} // namespace engine
} // namespace milvus
} // namespace zilliz