Update examples

This commit is contained in:
Loïc Lecrenier 2023-04-24 10:07:49 +02:00
parent 84d9c731f8
commit a7a0891210
2 changed files with 15 additions and 5 deletions

View File

@ -23,12 +23,14 @@ fn main() -> Result<(), Box<dyn Error>> {
let dataset_path = args let dataset_path = args
.next() .next()
.unwrap_or_else(|| panic!("{}", usage("Missing path to source dataset.", &program_name))); .unwrap_or_else(|| panic!("{}", usage("Missing path to source dataset.", &program_name)));
let primary_key = args.next().unwrap_or_else(|| "id".into()); // let primary_key = args.next().unwrap_or_else(|| "id".into());
// "title overview" // "title overview"
let searchable_fields: Vec<String> = args let searchable_fields: Vec<String> = args
.next() .next()
.map(|arg| arg.split_whitespace().map(ToString::to_string).collect()) .map(|arg| arg.split_whitespace().map(ToString::to_string).collect())
.unwrap_or_default(); .unwrap_or_default();
println!("{searchable_fields:?}");
// "release_date genres" // "release_date genres"
let filterable_fields: Vec<String> = args let filterable_fields: Vec<String> = args
.next() .next()
@ -44,17 +46,25 @@ fn main() -> Result<(), Box<dyn Error>> {
let config = IndexerConfig::default(); let config = IndexerConfig::default();
let mut builder = Settings::new(&mut wtxn, &index, &config); let mut builder = Settings::new(&mut wtxn, &index, &config);
builder.set_primary_key(primary_key); // builder.set_primary_key(primary_key);
let searchable_fields = searchable_fields.iter().map(|s| s.to_string()).collect(); let searchable_fields = searchable_fields.iter().map(|s| s.to_string()).collect();
builder.set_searchable_fields(searchable_fields); builder.set_searchable_fields(searchable_fields);
let filterable_fields = filterable_fields.iter().map(|s| s.to_string()).collect(); let filterable_fields = filterable_fields.iter().map(|s| s.to_string()).collect();
builder.set_filterable_fields(filterable_fields); builder.set_filterable_fields(filterable_fields);
builder.set_criteria(vec![Criterion::Words, Criterion::Typo, Criterion::Proximity]); builder.set_criteria(vec![
Criterion::Words,
Criterion::Typo,
Criterion::Proximity,
Criterion::Attribute,
]);
builder.execute(|_| (), || false).unwrap(); builder.execute(|_| (), || false).unwrap();
let config = IndexerConfig::default(); let config = IndexerConfig::default();
let indexing_config = IndexDocumentsConfig::default(); let mut indexing_config = IndexDocumentsConfig::default();
indexing_config.autogenerate_docids = true;
let builder = let builder =
IndexDocuments::new(&mut wtxn, &index, &config, indexing_config, |_| (), || false).unwrap(); IndexDocuments::new(&mut wtxn, &index, &config, indexing_config, |_| (), || false).unwrap();

View File

@ -10,7 +10,7 @@ fn main() {
let mut options = EnvOpenOptions::new(); let mut options = EnvOpenOptions::new();
options.map_size(100 * 1024 * 1024 * 1024); // 100 GB options.map_size(100 * 1024 * 1024 * 1024); // 100 GB
let index = Index::new(options, "data_movies").unwrap(); let index = Index::new(options, "data_movies.ms").unwrap();
let mut wtxn = index.write_txn().unwrap(); let mut wtxn = index.write_txn().unwrap();
let config = IndexerConfig::default(); let config = IndexerConfig::default();