1 // Copyright (c) 2016 Sandstorm Development Group, Inc. and contributors 2 // 3 // Permission is hereby granted, free of charge, to any person obtaining a copy 4 // of this software and associated documentation files (the "Software"), to deal 5 // in the Software without restriction, including without limitation the rights 6 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 // copies of the Software, and to permit persons to whom the Software is 8 // furnished to do so, subject to the following conditions: 9 // 10 // The above copyright notice and this permission notice shall be included in 11 // all copies or substantial portions of the Software. 12 // 13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 // THE SOFTWARE. 20 21 use std::io; 22 use futures::future::Future; 23 use futures::stream::Stream; 24 use futures::{Async, Poll}; 25 26 use capnp::{Error, message}; 27 28 #[must_use = "streams do nothing unless polled"] 29 pub struct ReadStream<R> where R: io::Read { 30 options: message::ReaderOptions, 31 read: ::serialize::Read<R>, 32 } 33 34 impl <R> ReadStream<R> where R: io::Read { new(reader: R, options: message::ReaderOptions) -> ReadStream<R>35 pub fn new(reader: R, options: message::ReaderOptions) -> ReadStream<R> { 36 ReadStream { 37 read: ::serialize::read_message(reader, options), 38 options: options, 39 } 40 } 41 } 42 43 impl <R> Stream for ReadStream<R> where R: io::Read { 44 type Item = message::Reader<::serialize::OwnedSegments>; 45 type Error = Error; 46 poll(&mut self) -> Poll<Option<Self::Item>, Error>47 fn poll(&mut self) -> Poll<Option<Self::Item>, Error> { 48 let (r, m) = try_ready!(Future::poll(&mut self.read)); 49 self.read = ::serialize::read_message(r, self.options); 50 Ok(Async::Ready(m)) 51 } 52 } 53