#pragma once

#include <a8/singleton.h>

class UpStream;
class UpStreamMgr : public a8::Singleton<UpStreamMgr>
{
 private:
    UpStreamMgr() {};
    friend class a8::Singleton<UpStreamMgr>;

 public:

    void Init();
    void UnInit();

    std::weak_ptr<UpStream> GetConnByKey(const std::string& key);
    std::weak_ptr<UpStream> GetConnById(int instance_id);
    std::weak_ptr<UpStream> RecreateUpStream(const std::string& host, int port);

 private:
    unsigned short curr_id_ = 1000;
    std::map<std::string, std::shared_ptr<UpStream>> key_hash_;
    std::map<int, std::shared_ptr<UpStream>> id_hash_;
};