fix ya.make

1 files changed, 35 insertions, 0 deletions

diff --git a/library/cpp/neh/multi.cpp b/library/cpp/neh/multi.cpp
new file mode 100644
index 0000000000..4929d4efd9
--- /dev/null
+++ b/library/cpp/neh/multi.cpp
@@ -0,0 +1,35 @@
+#include "rpc.h"
+#include "multi.h"
+#include "location.h"
+#include "factory.h"
+
+#include <util/string/cast.h>
+#include <util/generic/hash.h>
+
+using namespace NNeh;
+
+namespace {
+    namespace NMulti {
+        class TRequester: public IRequester {
+        public:
+            inline TRequester(const TListenAddrs& addrs, IOnRequest* cb) {
+                for (const auto& addr : addrs) {
+                    TParsedLocation loc(addr);
+                    IRequesterRef& req = R_[ToString(loc.Scheme) + ToString(loc.GetPort())];
+
+                    if (!req) {
+                        req = ProtocolFactory()->Protocol(loc.Scheme)->CreateRequester(cb, loc);
+                    }
+                }
+            }
+
+        private:
+            typedef THashMap<TString, IRequesterRef> TRequesters;
+            TRequesters R_;
+        };
+    }
+}
+
+IRequesterRef NNeh::MultiRequester(const TListenAddrs& addrs, IOnRequest* cb) {
+    return new NMulti::TRequester(addrs, cb);
+}