allow to set the primary key in the cli

This commit is contained in:
Tamo 2022-03-07 14:55:13 +01:00
parent 3cbadf92b6
commit 8bb45956d4
No known key found for this signature in database
GPG Key ID: 20CD8020AFA88D69

View File

@ -11,7 +11,7 @@ use indicatif::{MultiProgress, ProgressBar, ProgressStyle};
use milli::update::UpdateIndexingStep::{ use milli::update::UpdateIndexingStep::{
ComputeIdsAndMergeDocuments, IndexDocuments, MergeDataIntoFinalDatabase, RemapDocumentAddition, ComputeIdsAndMergeDocuments, IndexDocuments, MergeDataIntoFinalDatabase, RemapDocumentAddition,
}; };
use milli::update::{IndexDocumentsConfig, IndexDocumentsMethod, IndexerConfig}; use milli::update::{self, IndexDocumentsConfig, IndexDocumentsMethod, IndexerConfig};
use milli::Index; use milli::Index;
use serde_json::{Map, Value}; use serde_json::{Map, Value};
use structopt::StructOpt; use structopt::StructOpt;
@ -191,6 +191,9 @@ struct DocumentAddition {
/// Path to the update file, if not present, will read from stdin. /// Path to the update file, if not present, will read from stdin.
#[structopt(short, long)] #[structopt(short, long)]
path: Option<PathBuf>, path: Option<PathBuf>,
/// Specify the primary key.
#[structopt(long)]
primary: Option<String>,
/// Whether to generate missing document ids. /// Whether to generate missing document ids.
#[structopt(short, long)] #[structopt(short, long)]
autogen_docids: bool, autogen_docids: bool,
@ -231,6 +234,12 @@ impl Performer for DocumentAddition {
IndexDocumentsMethod::ReplaceDocuments IndexDocumentsMethod::ReplaceDocuments
}; };
if let Some(primary) = self.primary {
let mut builder = update::Settings::new(&mut txn, &index, &config);
builder.set_primary_key(primary);
builder.execute(|_| ()).unwrap();
}
let indexing_config = IndexDocumentsConfig { let indexing_config = IndexDocumentsConfig {
update_method, update_method,
autogenerate_docids: self.autogen_docids, autogenerate_docids: self.autogen_docids,