snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-26 03:55:07 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #367 from meilisearch/support-stdin-example

Browse Source 
Allow users to send csv files from stdin in examples
This commit is contained in:
Clément Renault 2019-12-05 12:33:18 +01:00 committed by GitHub
commit b756fc382a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
meilisearch-core/examples/from_file.rs
View File

@ -1,7 +1,7 @@
use std::collections::btree_map::{BTreeMap, Entry}; use std::collections::btree_map::{BTreeMap, Entry};
use std::collections::HashSet; use std::collections::HashSet;
use std::error::Error; use std::error::Error;
use std::io::Write; use std::io::{Read, Write};
use std::iter::FromIterator; use std::iter::FromIterator;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use std::time::{Duration, Instant}; use std::time::{Duration, Instant};
@ -24,7 +24,7 @@ struct IndexCommand {
#[structopt(long, default_value = "default")] #[structopt(long, default_value = "default")]
index_uid: String, index_uid: String,
/// The csv file to index. /// The csv file path to index, you can also use `-` to specify the standard input.
#[structopt(parse(from_os_str))] #[structopt(parse(from_os_str))]
csv_data_path: PathBuf, csv_data_path: PathBuf,
@ -135,7 +135,13 @@ fn index_command(command: IndexCommand, database: Database) -> Result<(), Box<dy
} }
} }
let mut rdr = csv::Reader::from_path(command.csv_data_path)?; let mut rdr = if command.csv_data_path.as_os_str() == "-" {
csv::Reader::from_reader(Box::new(io::stdin()) as Box<dyn Read>)
} else {
let file = std::fs::File::open(command.csv_data_path)?;
csv::Reader::from_reader(Box::new(file) as Box<dyn Read>)
};
let mut raw_record = csv::StringRecord::new(); let mut raw_record = csv::StringRecord::new();
let headers = rdr.headers()?.clone(); let headers = rdr.headers()?.clone();