如何在结构中的stdin上存储迭代器?

时间:2019-03-23 14:14:42

标签: rust iterator

我创建了一个结构,该结构中应存储文件或stdin上的迭代器,但编译器对我大喊:)

我认为Lines是我需要存储在我的结构中以在以后使用它进行迭代的结构,而Box将允许存储大小未知的变量,所以我这样定义我的结构:< / p>

pub struct A {
    pub input: Box<Lines<BufRead>>,
}

我以后想做这样的事情:

let mut a = A {
    input: /* don't know what should be here yet */,
};
if something {
    a.input = Box::new(io::stdin().lock().lines());
} else {
    a.input = Box::new(BufReader::new(file).lines());
}

最后

for line in a.input {
    // ...
}

但是我从编译器中得到了一个错误

error[E0277]: the size for values of type `(dyn std::io::BufRead + 'static)` cannot be known at compilation time
  --> src/context.rs:11:5
   |
11 |     pub input: Box<Lines<BufRead>>,
   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ doesn't have a size known at compile-time
   |
   = help: the trait `std::marker::Sized` is not implemented for `(dyn std::io::BufRead + 'static)`
   = note: to learn more, visit <https://doc.rust-lang.org/book/second-edition/ch19-04-advanced-types.html#dynamically-sized-types-and-sized>
   = note: required by `std::io::Lines`

我如何实现我的目标?

1 个答案:

答案 0 :(得分:1)

对该问题最通用的答案是您不/不能。锁定stdin会返回引用Stdin值的类型。您无法创建本地值(stdin()),无法对其进行引用(.lock()),然后返回该引用。

如果只想在函数中执行此操作而不返回它,则可以创建特征对象:

use std::io::{self, prelude::*, BufReader};

fn example(file: Option<std::fs::File>) {
    let stdin;
    let mut stdin_lines;
    let mut file_lines;

    let input: &mut Iterator<Item = _> = match file {
        None => {
            stdin = io::stdin();
            stdin_lines = stdin.lock().lines();
            &mut stdin_lines
        }
        Some(file) => {
            file_lines = BufReader::new(file).lines();
            &mut file_lines
        }
    };

    for line in input {
        // ...
    }
}

或创建一个新的泛型函数,您可以将任何一种具体的迭代器传递给:

use std::io::{self, prelude::*, BufReader};

fn example(file: Option<std::fs::File>) {
    match file {
        None => finally(io::stdin().lock().lines()),
        Some(file) => finally(BufReader::new(file).lines()),
    }
}

fn finally(input: impl Iterator<Item = io::Result<String>>) {
    for line in input {
        // ...
    }
}

您可以将特征对象或泛型类型放入结构中,即使您无法返回它:

struct A<'a> {
    input: &mut Iterator<Item = io::Result<String>>,
}
struct A<I> 
where
    I: Iterator<Item = io::Result<String>>,
{
    input: I,
}

如果您喜欢冒险,则可以使用一些不安全的代码/包装箱包装不安全的代码,以将Stdin值和迭代器一起引用,这并不是普遍安全的。

另请参阅:

input: Box<Lines<BufRead>>,

这是无效的,因为Lines不是特征。您想要:

use std::io::{prelude::*, Lines};

pub struct A {
    pub input: Lines<Box<BufRead>>,
}

use std::io;

pub struct A {
    pub input: Box<Iterator<Item = io::Result<String>>>,
}