check that we dont try to change websocket mode on a connected cluste…

…r (this would be Bad Times™)
brainboxdotcc · Aug 17, 2023 · 0cd5c0e · 0cd5c0e
1 parent 6cebf6f
commit 0cd5c0e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/include/dpp/cluster.h b/include/dpp/cluster.h
@@ -273,6 +273,7 @@ class DPP_EXPORT cluster {
 	 * 
 	 * @param mode websocket protocol to use, either ws_json or ws_etf.
 	 * @return cluster& Reference to self for chaining.
+	 * @throw dpp::logic_exception If called after the cluster is started (this is not supported)
 	 */
 	cluster& set_websocket_protocol(websocket_protocol_t mode);
 

diff --git a/src/dpp/cluster.cpp b/src/dpp/cluster.cpp
@@ -119,6 +119,9 @@ request_queue* cluster::get_raw_rest() {
 }
 
 cluster& cluster::set_websocket_protocol(websocket_protocol_t mode) {
+	if (start_time > 0) {
+		throw dpp::logic_exception("Cannot change websocket protocol on a started cluster!");
+	}
 	ws_mode = mode;
 	return *this;
 }