diff options
Diffstat (limited to 'zenserver/testing/httptest.h')
| -rw-r--r-- | zenserver/testing/httptest.h | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/zenserver/testing/httptest.h b/zenserver/testing/httptest.h index 2f3482abc..5809d4e2e 100644 --- a/zenserver/testing/httptest.h +++ b/zenserver/testing/httptest.h @@ -24,17 +24,24 @@ public: class PackageHandler : public IHttpPackageHandler { public: - PackageHandler(const HttpServerRequest& Request); + PackageHandler(HttpTestingService& Svc, uint32_t RequestId); ~PackageHandler(); - void FilterOffer(std::vector<IoHash>& OfferCids) override; - void OnBeginChunks() override; - IoBuffer CreateTarget(const IoHash& Cid, uint64_t StorageSize) override; - void OnEndChunks() override; + virtual void FilterOffer(std::vector<IoHash>& OfferCids) override; + virtual void OnRequestBegin() override; + virtual IoBuffer CreateTarget(const IoHash& Cid, uint64_t StorageSize) override; + virtual void OnRequestComplete() override; + + private: + HttpTestingService& m_Svc; + uint32_t m_RequestId; }; private: HttpRequestRouter m_Router; + + RwLock m_RwLock; + std::unordered_map<uint32_t, Ref<PackageHandler>> m_HandlerMap; }; } // namespace zen |