tiny_httpd/src/Tiny_httpd_buf.ml

type t = {
  mutable bytes: bytes;
  mutable i: int;
}

let create ?(size=4_096) () : t =
  { bytes=Bytes.make size ' '; i=0 }

let size self = self.i
let bytes_slice self = self.bytes
let clear self : unit =
  if Bytes.length self.bytes > 4_096 * 1_024 then (
    self.bytes <- Bytes.make 4096 ' '; (* free big buffer *)
  );
  self.i <- 0

let resize self new_size : unit =
  let new_buf = Bytes.make new_size ' ' in
  Bytes.blit self.bytes 0 new_buf 0 self.i;
  self.bytes <- new_buf

let add_bytes (self:t) s i len : unit =
  if self.i + len >= Bytes.length self.bytes then (
    resize self (self.i + self.i / 2 + len + 10);
  );
  Bytes.blit s i self.bytes self.i len;
  self.i <- self.i + len

let contents (self:t) : string = Bytes.sub_string self.bytes 0 self.i

let contents_and_clear (self:t) : string =
  let x = contents self in
  clear self;
  x