mirror of
https://github.com/ocaml-multicore/eio.git
synced 2025-07-30 00:02:24 -04:00
Compare commits
No commits in common. "b942dde3b3ae396c56d58751cfddc5c472be7a14" and "ca1d239507c18f496845d80893300e7701beefb9" have entirely different histories.
b942dde3b3
...
ca1d239507
@ -1,44 +0,0 @@
|
|||||||
(* A client opens a connection to an echo service and sends a load of data via it. *)
|
|
||||||
|
|
||||||
open Eio.Std
|
|
||||||
|
|
||||||
let chunk_size = 1 lsl 16
|
|
||||||
let n_chunks = 10000
|
|
||||||
let n_bytes = n_chunks * chunk_size
|
|
||||||
|
|
||||||
let run_client sock =
|
|
||||||
Fiber.both
|
|
||||||
(fun () ->
|
|
||||||
let chunk = Cstruct.create chunk_size in
|
|
||||||
for _ = 1 to n_chunks do
|
|
||||||
Eio.Flow.write sock [chunk]
|
|
||||||
done;
|
|
||||||
Eio.Flow.shutdown sock `Send
|
|
||||||
)
|
|
||||||
(fun () ->
|
|
||||||
let chunk = Cstruct.create chunk_size in
|
|
||||||
for _ = 1 to n_chunks do
|
|
||||||
Eio.Flow.read_exact sock chunk
|
|
||||||
done
|
|
||||||
)
|
|
||||||
|
|
||||||
let time name service =
|
|
||||||
Switch.run @@ fun sw ->
|
|
||||||
let client_sock, server_sock = Eio_unix.Net.socketpair_stream ~sw () in
|
|
||||||
let t0 = Unix.gettimeofday () in
|
|
||||||
Fiber.both
|
|
||||||
(fun () -> service server_sock)
|
|
||||||
(fun () -> run_client client_sock);
|
|
||||||
let t1 = Unix.gettimeofday () in
|
|
||||||
let time = t1 -. t0 in
|
|
||||||
let bytes_per_second = float n_bytes /. time in
|
|
||||||
traceln "%s: %.2f MB/s" name (bytes_per_second /. 1024. /. 1024.);
|
|
||||||
Metric.create name (`Float bytes_per_second) "bytes/s" (name ^ " Flow.copy")
|
|
||||||
|
|
||||||
let run _env =
|
|
||||||
[
|
|
||||||
time "default" (fun sock -> Eio.Flow.copy sock sock);
|
|
||||||
time "buf_read" (fun sock ->
|
|
||||||
let r = Eio.Buf_read.of_flow sock ~initial_size:(64 * 1024) ~max_size:(64 * 1024) |> Eio.Buf_read.as_flow in
|
|
||||||
Eio.Flow.copy r sock);
|
|
||||||
]
|
|
@ -11,7 +11,6 @@ let benchmarks = [
|
|||||||
"Eio_unix.Fd", Bench_fd.run;
|
"Eio_unix.Fd", Bench_fd.run;
|
||||||
"File.stat", Bench_fstat.run;
|
"File.stat", Bench_fstat.run;
|
||||||
"Path.stat", Bench_stat.run;
|
"Path.stat", Bench_stat.run;
|
||||||
"Flow.copy", Bench_copy.run;
|
|
||||||
]
|
]
|
||||||
|
|
||||||
let usage_error () =
|
let usage_error () =
|
||||||
|
@ -140,13 +140,7 @@ module F = struct
|
|||||||
consume t len;
|
consume t len;
|
||||||
len
|
len
|
||||||
|
|
||||||
let rsb t fn =
|
let read_methods = []
|
||||||
ensure t 1;
|
|
||||||
let data = peek t in
|
|
||||||
let sent = fn [data] in
|
|
||||||
consume t sent
|
|
||||||
|
|
||||||
let read_methods = [Flow.Read_source_buffer rsb]
|
|
||||||
end
|
end
|
||||||
|
|
||||||
let as_flow =
|
let as_flow =
|
||||||
|
@ -64,22 +64,7 @@ module Impl = struct
|
|||||||
with Unix.Unix_error (code, name, arg) ->
|
with Unix.Unix_error (code, name, arg) ->
|
||||||
raise (Err.wrap code name arg)
|
raise (Err.wrap code name arg)
|
||||||
|
|
||||||
(* Copy using the [Read_source_buffer] optimisation.
|
let copy t ~src = Eio.Flow.Pi.simple_copy ~single_write t ~src
|
||||||
Avoids a copy if the source already has the data. *)
|
|
||||||
let copy_with_rsb rsb dst =
|
|
||||||
try
|
|
||||||
while true do rsb (single_write dst) done
|
|
||||||
with End_of_file -> ()
|
|
||||||
|
|
||||||
let copy t ~src =
|
|
||||||
let Eio.Resource.T (src_t, ops) = src in
|
|
||||||
let module Src = (val (Eio.Resource.get ops Eio.Flow.Pi.Source)) in
|
|
||||||
let rec aux = function
|
|
||||||
| Eio.Flow.Read_source_buffer rsb :: _ -> copy_with_rsb (rsb src_t) t
|
|
||||||
| _ :: xs -> aux xs
|
|
||||||
| [] -> Eio.Flow.Pi.simple_copy ~single_write t ~src
|
|
||||||
in
|
|
||||||
aux Src.read_methods
|
|
||||||
|
|
||||||
let single_read t buf =
|
let single_read t buf =
|
||||||
match Low_level.readv t [| buf |] with
|
match Low_level.readv t [| buf |] with
|
||||||
|
Loading…
x
Reference in New Issue
Block a user