Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1227)

Unified Diff: mojo/python/mojo_utils/data_pipe_utils.py

Issue 952223002: Add Python util to read from a data pipe into a bytearray (Closed) Base URL: https://github.com/domokit/mojo.git@master
Patch Set: Add comments Created 5 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « mojo/python/mojo_utils/__init__.py ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: mojo/python/mojo_utils/data_pipe_utils.py
diff --git a/mojo/python/mojo_utils/data_pipe_utils.py b/mojo/python/mojo_utils/data_pipe_utils.py
new file mode 100644
index 0000000000000000000000000000000000000000..707c37e848f30bd9c815be0516df283b6f17a45b
--- /dev/null
+++ b/mojo/python/mojo_utils/data_pipe_utils.py
@@ -0,0 +1,34 @@
+# Copyright 2015 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import mojo_system
+
+def BlockingCopyFromDataPipeIntoString(data_pipe, deadline):
+ """
+ Reads from |data_pipe| until either (1) there is no more data to read or
+ (2) an error is encountered.
+ Returns a tuple of the result status and output string. If the status is not
+ mojo_system.RESULT_OK, the returned string will be None.
+ """
+ BUFFER_SIZE = 1000
qsr 2015/02/25 15:22:35 Use a power of 2, usual values are 1 to 4k
blundell 2015/02/26 16:33:53 Done.
+ input_bytes = bytearray(BUFFER_SIZE)
+ output_bytes = bytearray()
qsr 2015/02/25 15:22:35 Instead of transferring in BUFFER_SIZE chunk, beca
blundell 2015/02/26 16:33:53 Done.
+ while True:
+ status, read_bytes = data_pipe.ReadData(input_bytes)
+ if status == mojo_system.RESULT_OK:
+ output_bytes = output_bytes + read_bytes
+ continue
+
+ if status == mojo_system.RESULT_SHOULD_WAIT:
+ status, _ = data_pipe.Wait(mojo_system.HANDLE_SIGNAL_READABLE, deadline)
+
+ # If the wait results in RESULT_OK, try to read more bytes.
+ if status == mojo_system.RESULT_OK:
+ continue
qsr 2015/02/25 15:22:35 What happen is you get another result here? In par
blundell 2015/02/26 16:33:53 Any result other than OK or FAILED_PRECONDITION is
qsr 2015/02/26 16:45:19 Hum, misread the alignment. Ok.
+
+ # Treat a failed precondition as EOF.
+ if status == mojo_system.RESULT_FAILED_PRECONDITION:
+ return (mojo_system.RESULT_OK, output_bytes.decode("utf-8"))
qsr 2015/02/25 15:22:35 This method being a generic utility method, it sho
blundell 2015/02/26 16:33:52 Done.
+
+ return (status, None)
« no previous file with comments | « mojo/python/mojo_utils/__init__.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698