slight refactor

holepunchto · Dec 9, 2024 · 1bc05c5 · 1bc05c5
1 parent 14601af
commit 1bc05c5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/udx.c b/src/udx.c
@@ -2462,11 +2462,12 @@ static void
 set_stream_socket (udx_stream_t *stream, udx_socket_t *socket) {
   if (stream->socket == socket) return; // just in case
 
-  if (stream->socket == NULL) {
+  udx_socket_t *prev = stream->socket;
+
+  if (prev == NULL) {
     udx_t *udx = stream->udx;
     udx__link_remove(udx->streams, stream);
   } else {
-    udx_socket_t *prev = stream->socket;
     udx__link_remove(prev->streams, stream);
   }