From 366a50b4d028acaaf77c38da50005515d99da8b2 Mon Sep 17 00:00:00 2001
From: Rudi Grinberg <me@rgrinberg.com>
Date: Sun, 30 Jun 2024 15:41:29 +0100
Subject: [PATCH] refactor: simplify flush writer

Signed-off-by: Rudi Grinberg <me@rgrinberg.com>

<!-- ps-id: 391bec85-4b37-4893-a1f4-7961b403d153 -->
---
 cohttp/src/transfer_io.ml | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/cohttp/src/transfer_io.ml b/cohttp/src/transfer_io.ml
index f02675b29..7baa279f3 100644
--- a/cohttp/src/transfer_io.ml
+++ b/cohttp/src/transfer_io.ml
@@ -124,18 +124,15 @@ module Make (IO : S.IO) = struct
     if String.length s = 0 then return () else writer io s
 
   let make_writer ~flush mode =
+    let write =
+      match mode with
+      | Chunked -> Chunked.write
+      | Fixed _ -> Fixed.write
+      | Unknown -> Unknown.write
+    in
     match flush with
-    | false -> (
-        match mode with
-        | Chunked -> Chunked.write
-        | Fixed _ -> Fixed.write
-        | Unknown -> Unknown.write)
-    | true ->
-        (match mode with
-        | Chunked -> write_and_flush Chunked.write
-        | Fixed _ -> write_and_flush Fixed.write
-        | Unknown -> write_and_flush Unknown.write)
-        |> write_ignore_blank
+    | false -> write
+    | true -> write_and_flush write |> write_ignore_blank
 
   let read reader = reader ()
   let write writer buf = writer buf