X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=serverpool.cpp;h=7b790c29b4e87de793a58de34d9a077861da54cb;hp=c3fc69d04a3414fd00fee354966bc02a886c8d11;hb=edf39a7cec7860905689cbec7dd41d7fb3fc8df7;hpb=ed218ece51e5be77fd8bd8f014b5f05708e4be5b

diff --git a/serverpool.cpp b/serverpool.cpp
index c3fc69d..7b790c2 100644
--- a/serverpool.cpp
+++ b/serverpool.cpp
@@ -1,4 +1,5 @@
 #include <assert.h>
+#include <fcntl.h>
 #include <stdlib.h>
 #include <sys/types.h>
 
@@ -178,6 +179,16 @@ void ServerPool::add_data(int stream_index, const char *data, size_t bytes, uint
 	}
 }
 
+void ServerPool::set_unavailable(int stream_index)
+{
+	assert(stream_index >= 0 && stream_index < ssize_t(num_http_streams + udp_streams.size()));
+	if (stream_index < ssize_t(num_http_streams)) {
+		for (int i = 0; i < num_servers; ++i) {
+			servers[i].set_unavailable(stream_index);
+		}
+	}
+}
+
 void ServerPool::add_gen204(const std::string &url, const std::string &allow_origin)
 {
 	for (int i = 0; i < num_servers; ++i) {