table/
requests.rs

1// Copyright 2023 Greptime Team
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15//! Table and TableEngine requests
16
17use std::collections::{HashMap, HashSet};
18use std::fmt;
19use std::str::FromStr;
20
21use common_base::readable_size::ReadableSize;
22use common_datasource::object_store::oss::is_supported_in_oss;
23use common_datasource::object_store::s3::is_supported_in_s3;
24use common_query::AddColumnLocation;
25use common_time::TimeToLive;
26use common_time::range::TimestampRange;
27use datatypes::data_type::ConcreteDataType;
28use datatypes::json::JsonSettings;
29use datatypes::prelude::VectorRef;
30use datatypes::schema::{
31    ColumnDefaultConstraint, ColumnSchema, FulltextOptions, Schema, SkippingIndexOptions,
32};
33use greptime_proto::v1::region::compact_request;
34use once_cell::sync::Lazy;
35use serde::{Deserialize, Serialize};
36use store_api::metric_engine_consts::{
37    LOGICAL_TABLE_METADATA_KEY, PHYSICAL_TABLE_METADATA_KEY, is_metric_engine_option_key,
38};
39use store_api::mito_engine_options::{
40    APPEND_MODE_KEY, COMPACTION_TYPE, EXPERIMENTAL_SST_FLOAT_FIELD_ENCODING, FloatFieldEncoding,
41    MEMTABLE_BULK_ENCODE_BYTES_THRESHOLD, MEMTABLE_BULK_ENCODE_ROW_THRESHOLD,
42    MEMTABLE_BULK_MAX_MERGE_GROUPS, MEMTABLE_BULK_MERGE_THRESHOLD, MEMTABLE_TYPE, MERGE_MODE_KEY,
43    SST_FORMAT_KEY, TWCS_ACTIVE_WINDOW_L1_MERGE_TRIGGER, TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM,
44    TWCS_FALLBACK_TO_LOCAL, TWCS_INACTIVE_WINDOW_L1_MERGE_TRIGGER,
45    TWCS_INACTIVE_WINDOW_TRIGGER_FILE_NUM, TWCS_MAX_OUTPUT_FILE_SIZE, TWCS_TIME_WINDOW,
46    TWCS_TRIGGER_FILE_NUM, is_mito_engine_option_key, normalize_twcs_trigger_options,
47};
48use store_api::region_request::{SetRegionOption, UnsetRegionOption};
49
50use crate::error::{ConflictingTableOptionsSnafu, ParseTableOptionSnafu, Result};
51use crate::metadata::{TableId, TableVersion};
52use crate::table_reference::TableReference;
53
54mod semantic;
55pub use semantic::*;
56
57pub const FILE_TABLE_META_KEY: &str = "__private.file_table_meta";
58pub const FILE_TABLE_LOCATION_KEY: &str = "location";
59pub const FILE_TABLE_PATTERN_KEY: &str = "pattern";
60pub const FILE_TABLE_FORMAT_KEY: &str = "format";
61
62pub const TABLE_DATA_MODEL: &str = "table_data_model";
63pub const TABLE_DATA_MODEL_TRACE_V1: &str = "greptime_trace_v1";
64/// Table data model used by the JSON2-based OTLP trace pipeline.
65pub const TABLE_DATA_MODEL_TRACE_V2: &str = "greptime_trace_v2";
66
67/// Returns true for the Trace V1 and V2 data models supported
68/// by semantic graph derivation.
69pub fn is_trace_table(table_info: &crate::metadata::TableInfo) -> bool {
70    let table_data_model = table_info.meta.options.data_model();
71    matches!(
72        table_data_model,
73        Some(TABLE_DATA_MODEL_TRACE_V1 | TABLE_DATA_MODEL_TRACE_V2)
74    )
75}
76
77pub const OTLP_METRIC_COMPAT_KEY: &str = "otlp_metric_compat";
78pub const OTLP_METRIC_COMPAT_PROM: &str = "prom";
79
80pub const VALID_TABLE_OPTION_KEYS: [&str; 15] = [
81    // common keys:
82    WRITE_BUFFER_SIZE_KEY,
83    TTL_KEY,
84    STORAGE_KEY,
85    COMMENT_KEY,
86    SKIP_WAL_KEY,
87    SST_FORMAT_KEY,
88    // file engine keys:
89    FILE_TABLE_LOCATION_KEY,
90    FILE_TABLE_FORMAT_KEY,
91    FILE_TABLE_PATTERN_KEY,
92    // metric engine keys:
93    PHYSICAL_TABLE_METADATA_KEY,
94    LOGICAL_TABLE_METADATA_KEY,
95    // table model info
96    TABLE_DATA_MODEL,
97    OTLP_METRIC_COMPAT_KEY,
98    REPARTITION_COLUMN_HINT_KEY,
99    REPARTITION_PARTITION_NUM_HINT_KEY,
100];
101
102pub const DDL_TIMEOUT: &str = "timeout";
103pub const DDL_WAIT: &str = "wait";
104
105pub const VALID_DDL_OPTION_KEYS: [&str; 2] = [DDL_TIMEOUT, DDL_WAIT];
106
107/// The key of ingest rows rate limit option (rows per second, cluster-wide) in database options.
108pub const INGEST_ROWS_RATE_LIMIT_KEY: &str = "ingest_rows_rate_limit";
109
110// Valid option keys when creating a db.
111static VALID_DB_OPT_KEYS: Lazy<HashSet<&str>> = Lazy::new(|| {
112    let mut set = HashSet::new();
113    set.insert(TTL_KEY);
114    set.insert(STORAGE_KEY);
115    set.insert(MEMTABLE_TYPE);
116    set.insert(MEMTABLE_BULK_MERGE_THRESHOLD);
117    set.insert(MEMTABLE_BULK_ENCODE_ROW_THRESHOLD);
118    set.insert(MEMTABLE_BULK_ENCODE_BYTES_THRESHOLD);
119    set.insert(MEMTABLE_BULK_MAX_MERGE_GROUPS);
120    set.insert(APPEND_MODE_KEY);
121    set.insert(MERGE_MODE_KEY);
122    set.insert(SKIP_WAL_KEY);
123    set.insert(COMPACTION_TYPE);
124    set.insert(TWCS_FALLBACK_TO_LOCAL);
125    set.insert(TWCS_TIME_WINDOW);
126    set.insert(TWCS_TRIGGER_FILE_NUM);
127    set.insert(TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM);
128    set.insert(TWCS_ACTIVE_WINDOW_L1_MERGE_TRIGGER);
129    set.insert(TWCS_INACTIVE_WINDOW_TRIGGER_FILE_NUM);
130    set.insert(TWCS_INACTIVE_WINDOW_L1_MERGE_TRIGGER);
131    set.insert(TWCS_MAX_OUTPUT_FILE_SIZE);
132    set.insert(SST_FORMAT_KEY);
133    set.insert(INGEST_ROWS_RATE_LIMIT_KEY);
134    set
135});
136
137/// Returns true if the `key` is a valid key for database.
138pub fn validate_database_option(key: &str) -> bool {
139    VALID_DB_OPT_KEYS.contains(&key)
140}
141
142/// Validates a database option value, returning the violated constraint on error.
143pub fn validate_database_option_value(
144    key: &str,
145    value: Option<&str>,
146) -> std::result::Result<(), &'static str> {
147    if key == INGEST_ROWS_RATE_LIMIT_KEY {
148        return value
149            .and_then(|value| value.parse::<u64>().ok())
150            .map(|_| ())
151            .ok_or("expected a non-negative integer fitting in u64");
152    }
153    let (minimum, constraint) = match key {
154        TWCS_TRIGGER_FILE_NUM
155        | TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM
156        | TWCS_INACTIVE_WINDOW_TRIGGER_FILE_NUM => {
157            (0, "expected a non-negative integer fitting in usize")
158        }
159        TWCS_ACTIVE_WINDOW_L1_MERGE_TRIGGER | TWCS_INACTIVE_WINDOW_L1_MERGE_TRIGGER => {
160            (2, "expected an integer greater than or equal to 2")
161        }
162        _ => return Ok(()),
163    };
164    if value
165        .and_then(|value| value.parse::<usize>().ok())
166        .is_some_and(|files| files >= minimum)
167    {
168        Ok(())
169    } else {
170        Err(constraint)
171    }
172}
173
174/// Returns true if the `key` is a valid key for any engine or storage.
175pub fn validate_table_option(key: &str) -> bool {
176    if is_supported_in_s3(key) {
177        return true;
178    }
179
180    if is_supported_in_oss(key) {
181        return true;
182    }
183
184    if is_mito_engine_option_key(key) {
185        return true;
186    }
187
188    if is_metric_engine_option_key(key) {
189        return true;
190    }
191
192    // Semantic-layer keys share a reserved prefix instead of a fixed allowlist so
193    // the vocabulary can grow without touching this gate. See `semantic` module.
194    if is_semantic_option_key(key) {
195        return true;
196    }
197
198    VALID_TABLE_OPTION_KEYS.contains(&key) || VALID_DDL_OPTION_KEYS.contains(&key)
199}
200
201#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
202#[serde(default)]
203pub struct TableOptions {
204    /// Per-region write buffer stall threshold. Writes are rejected at twice this size.
205    pub write_buffer_size: Option<ReadableSize>,
206    /// Time-to-live of table. Expired data will be automatically purged.
207    pub ttl: Option<TimeToLive>,
208    /// Skip wal write for this table.
209    pub skip_wal: bool,
210    /// Extra options that may not applicable to all table engines.
211    pub extra_options: HashMap<String, String>,
212}
213
214pub const WRITE_BUFFER_SIZE_KEY: &str = store_api::mito_engine_options::WRITE_BUFFER_SIZE_KEY;
215pub const TTL_KEY: &str = store_api::mito_engine_options::TTL_KEY;
216pub const STORAGE_KEY: &str = "storage";
217pub const COMMENT_KEY: &str = "comment";
218pub const AUTO_CREATE_TABLE_KEY: &str = "auto_create_table";
219pub const SKIP_WAL_KEY: &str = store_api::mito_engine_options::SKIP_WAL_KEY;
220pub const TRACE_TABLE_PARTITIONS_HINT_KEY: &str = "trace_table_partitions";
221pub const REPARTITION_COLUMN_HINT_KEY: &str = "repartition.column.hint";
222
223/// Table-level partition count hint consumed by the auto-repartition planner.
224pub const REPARTITION_PARTITION_NUM_HINT_KEY: &str = "repartition.partition.num.hint";
225
226impl TableOptions {
227    /// Returns the table data model, if specified.
228    pub fn data_model(&self) -> Option<&str> {
229        self.extra_options.get(TABLE_DATA_MODEL).map(String::as_str)
230    }
231
232    pub fn try_from_iter<T: ToString, U: IntoIterator<Item = (T, T)>>(
233        iter: U,
234    ) -> Result<TableOptions> {
235        let mut options = TableOptions::default();
236
237        let mut kvs: HashMap<String, String> = iter
238            .into_iter()
239            .map(|(k, v)| (k.to_string(), v.to_string()))
240            .collect();
241
242        normalize_twcs_trigger_options(&mut kvs).map_err(|conflict| {
243            ConflictingTableOptionsSnafu {
244                first_key: TWCS_TRIGGER_FILE_NUM,
245                first_value: conflict.legacy_value,
246                second_key: TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM,
247                second_value: conflict.canonical_value,
248            }
249            .build()
250        })?;
251
252        if let Some(write_buffer_size) = kvs.get(WRITE_BUFFER_SIZE_KEY) {
253            let size = ReadableSize::from_str(write_buffer_size).map_err(|_| {
254                ParseTableOptionSnafu {
255                    key: WRITE_BUFFER_SIZE_KEY,
256                    value: write_buffer_size,
257                }
258                .build()
259            })?;
260            options.write_buffer_size = Some(size)
261        }
262
263        if let Some(ttl) = kvs.get(TTL_KEY) {
264            let ttl_value = TimeToLive::from_humantime_or_str(ttl).map_err(|_| {
265                ParseTableOptionSnafu {
266                    key: TTL_KEY,
267                    value: ttl,
268                }
269                .build()
270            })?;
271            options.ttl = Some(ttl_value);
272        }
273
274        if let Some(encoding) = kvs.get(EXPERIMENTAL_SST_FLOAT_FIELD_ENCODING) {
275            encoding.parse::<FloatFieldEncoding>().map_err(|_| {
276                ParseTableOptionSnafu {
277                    key: EXPERIMENTAL_SST_FLOAT_FIELD_ENCODING,
278                    value: encoding,
279                }
280                .build()
281            })?;
282        }
283
284        if let Some(skip_wal) = kvs.get(SKIP_WAL_KEY) {
285            options.skip_wal = skip_wal.parse().map_err(|_| {
286                ParseTableOptionSnafu {
287                    key: SKIP_WAL_KEY,
288                    value: skip_wal,
289                }
290                .build()
291            })?;
292        }
293
294        options.extra_options = HashMap::from_iter(
295            kvs.into_iter()
296                .filter(|(k, _)| k != WRITE_BUFFER_SIZE_KEY && k != TTL_KEY),
297        );
298
299        Ok(options)
300    }
301}
302
303impl fmt::Display for TableOptions {
304    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
305        let mut key_vals = vec![];
306        if let Some(size) = self.write_buffer_size {
307            key_vals.push(format!("{}={}", WRITE_BUFFER_SIZE_KEY, size));
308        }
309
310        if let Some(ttl) = self.ttl.map(|ttl| ttl.to_string()) {
311            key_vals.push(format!("{}={}", TTL_KEY, ttl));
312        }
313
314        if self.skip_wal && !self.extra_options.contains_key(SKIP_WAL_KEY) {
315            key_vals.push(format!("{}={}", SKIP_WAL_KEY, self.skip_wal));
316        }
317
318        for (k, v) in &self.extra_options {
319            key_vals.push(format!("{}={}", k, v));
320        }
321
322        write!(f, "{}", key_vals.join(" "))
323    }
324}
325
326impl From<&TableOptions> for HashMap<String, String> {
327    fn from(opts: &TableOptions) -> Self {
328        let mut res = HashMap::with_capacity(3 + opts.extra_options.len());
329        if let Some(write_buffer_size) = opts.write_buffer_size {
330            let _ = res.insert(
331                WRITE_BUFFER_SIZE_KEY.to_string(),
332                write_buffer_size.to_string(),
333            );
334        }
335        if let Some(ttl_str) = opts.ttl.map(|ttl| ttl.to_string()) {
336            let _ = res.insert(TTL_KEY.to_string(), ttl_str);
337        }
338        if opts.skip_wal {
339            let _ = res.insert(SKIP_WAL_KEY.to_string(), true.to_string());
340        }
341        res.extend(opts.extra_options.clone());
342        res
343    }
344}
345
346/// Alter table request
347#[derive(Debug, Clone, Serialize, Deserialize)]
348pub struct AlterTableRequest {
349    pub catalog_name: String,
350    pub schema_name: String,
351    pub table_name: String,
352    pub table_id: TableId,
353    pub alter_kind: AlterKind,
354    // None in standalone.
355    pub table_version: Option<TableVersion>,
356}
357
358/// Add column request
359#[derive(Debug, Clone, Serialize, Deserialize)]
360pub struct AddColumnRequest {
361    pub column_schema: ColumnSchema,
362    pub is_key: bool,
363    pub location: Option<AddColumnLocation>,
364    /// Add column if not exists.
365    pub add_if_not_exists: bool,
366}
367
368/// Change column datatype request
369#[derive(Debug, Clone, Serialize, Deserialize)]
370pub struct ModifyColumnTypeRequest {
371    pub column_name: String,
372    pub target_type: ConcreteDataType,
373}
374
375/// Set JSON2 settings request.
376#[derive(Debug, Clone, Serialize, Deserialize)]
377pub struct SetJsonSettingsRequest {
378    pub column_name: String,
379    pub settings: JsonSettings,
380}
381
382/// A family of annotation table options: pure metadata markers that no region
383/// consumes. Setting or unsetting them only rewrites the table's
384/// `extra_options`, so the alter skips region dispatch entirely.
385#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
386pub enum AnnotationFamily {
387    /// `greptime.semantic.*` options (see the [`semantic`] module).
388    Semantic,
389    /// Column and partition count hints consumed by the auto-repartition planner.
390    RepartitionHint,
391}
392
393impl AnnotationFamily {
394    /// The key namespace used in diagnostics; accepted keys are classified by [`Self::of_key`].
395    pub fn namespace(self) -> &'static str {
396        match self {
397            Self::Semantic => SEMANTIC_PREFIX,
398            Self::RepartitionHint => "repartition.",
399        }
400    }
401
402    pub fn of_key(key: &str) -> Option<Self> {
403        if key.starts_with(SEMANTIC_PREFIX) {
404            Some(Self::Semantic)
405        } else if matches!(
406            key,
407            REPARTITION_COLUMN_HINT_KEY | REPARTITION_PARTITION_NUM_HINT_KEY
408        ) {
409            Some(Self::RepartitionHint)
410        } else {
411            None
412        }
413    }
414
415    /// Whether this family may be altered on logical metric tables. Only
416    /// families whose values nothing on the physical side consumes qualify;
417    /// the repartition hint drives physical region repartitioning.
418    pub fn allows_logical_tables(self) -> bool {
419        match self {
420            Self::Semantic => true,
421            Self::RepartitionHint => false,
422        }
423    }
424
425    /// Whether duplicate keys are rejected in this family's SET/UNSET batch.
426    pub fn requires_unique_keys(self) -> bool {
427        matches!(self, Self::RepartitionHint)
428    }
429
430    /// The error for a SET/UNSET batch mixing this family with other options.
431    pub fn mixed_batch_error(self) -> String {
432        match self {
433            Self::Semantic => format!(
434                "`{SEMANTIC_PREFIX}*` options must be altered separately from other table options"
435            ),
436            Self::RepartitionHint => {
437                "repartition hints must be altered separately from other table options".to_string()
438            }
439        }
440    }
441}
442
443/// Why an annotation SET/UNSET key batch was rejected.
444#[derive(Debug, Clone, PartialEq, Eq)]
445pub enum AnnotationKeyError {
446    MixedFamilies { family: AnnotationFamily },
447    DuplicateKey,
448}
449
450impl fmt::Display for AnnotationKeyError {
451    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
452        match self {
453            Self::MixedFamilies { family } => f.write_str(&family.mixed_batch_error()),
454            Self::DuplicateKey => f.write_str("duplicate repartition hint keys"),
455        }
456    }
457}
458
459/// Validates a SET/UNSET batch and returns its annotation family.
460///
461/// Empty batches and batches containing only non-annotation keys return `None`.
462/// Annotation keys cannot share a batch with another family or region options.
463/// Duplicate keys are rejected only for families that require unique keys.
464pub fn validate_annotation_keys<'a>(
465    keys: impl IntoIterator<Item = &'a str>,
466) -> std::result::Result<Option<AnnotationFamily>, AnnotationKeyError> {
467    let mut keys = keys.into_iter();
468    let Some(first) = keys.next() else {
469        return Ok(None);
470    };
471    let family = AnnotationFamily::of_key(first);
472    let reject_duplicates = family.is_some_and(|family| family.requires_unique_keys());
473    let mut seen = HashSet::new();
474    if reject_duplicates {
475        seen.insert(first);
476    }
477    for key in keys {
478        let this = AnnotationFamily::of_key(key);
479        if this != family
480            && let Some(family) = family.or(this)
481        {
482            return Err(AnnotationKeyError::MixedFamilies { family });
483        }
484        if reject_duplicates && !seen.insert(key) {
485            return Err(AnnotationKeyError::DuplicateKey);
486        }
487    }
488    Ok(family)
489}
490
491/// Table shape an annotation option is validated against.
492pub struct AnnotationContext<'a> {
493    pub data_model: Option<&'a str>,
494    pub schema: &'a Schema,
495    pub partition_key_indices: &'a [usize],
496}
497
498/// Why an annotation option was rejected. Typed so each DDL entry point maps
499/// rules onto its existing error variants and status codes: ALTER keeps
500/// missing columns as `TableColumnNotFound` (4002), CREATE keeps its
501/// `InvalidArguments` family — the rules converge, the contracts do not.
502#[derive(Debug, Clone, PartialEq, Eq)]
503pub enum AnnotationValidationError {
504    UnknownKey {
505        key: String,
506    },
507    InvalidValue {
508        key: String,
509        value: String,
510    },
511    ColumnNotFound {
512        column: String,
513    },
514    ColumnNotStringForm {
515        key: String,
516        column: String,
517        ty: ConcreteDataType,
518    },
519    InvalidPartitionNumHint {
520        value: String,
521    },
522    NotSingleColumn,
523    PartitionMetadataConflict,
524    TimeIndexConflict,
525}
526
527impl fmt::Display for AnnotationValidationError {
528    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
529        match self {
530            Self::UnknownKey { key } => write!(f, "unknown semantic option `{key}`"),
531            Self::InvalidValue { key, value } => {
532                write!(f, "invalid value `{value}` for semantic option `{key}`")
533            }
534            Self::ColumnNotFound { column } => write!(f, "column `{column}` not found"),
535            Self::ColumnNotStringForm { key, column, ty } => write!(
536                f,
537                "entity column `{column}` (option `{key}`) has type `{ty}`, \
538                 which cannot render as a string"
539            ),
540            Self::InvalidPartitionNumHint { value } => write!(
541                f,
542                "{REPARTITION_PARTITION_NUM_HINT_KEY} expects a positive integer within u32 range, got `{value}`"
543            ),
544            Self::NotSingleColumn => write!(
545                f,
546                "{REPARTITION_COLUMN_HINT_KEY} expects exactly one column name"
547            ),
548            Self::PartitionMetadataConflict => write!(
549                f,
550                "cannot set {REPARTITION_COLUMN_HINT_KEY} on a table with partition metadata"
551            ),
552            Self::TimeIndexConflict => write!(
553                f,
554                "cannot set {REPARTITION_COLUMN_HINT_KEY} to the time index column"
555            ),
556        }
557    }
558}
559
560/// Validates one annotation option and returns the value to store — the
561/// repartition hints are trimmed, semantic values pass
562/// through unchanged.
563pub(crate) fn validate_and_normalize_annotation(
564    family: AnnotationFamily,
565    cx: &AnnotationContext<'_>,
566    key: &str,
567    value: &str,
568) -> std::result::Result<String, AnnotationValidationError> {
569    match family {
570        AnnotationFamily::Semantic => {
571            if !is_semantic_option_key(key) {
572                return Err(AnnotationValidationError::UnknownKey {
573                    key: key.to_string(),
574                });
575            }
576            if !validate_semantic_option(key, value) {
577                return Err(AnnotationValidationError::InvalidValue {
578                    key: key.to_string(),
579                    value: value.to_string(),
580                });
581            }
582            if parse_entity_option_key(key).is_some() {
583                for column in parse_entity_columns(value) {
584                    if trace_v2_attribute(cx.schema, cx.data_model, &column).is_some() {
585                        continue;
586                    }
587                    let schema = cx.schema.column_schema_by_name(&column).ok_or_else(|| {
588                        AnnotationValidationError::ColumnNotFound {
589                            column: column.clone(),
590                        }
591                    })?;
592                    if !has_stable_string_form(&schema.data_type) {
593                        return Err(AnnotationValidationError::ColumnNotStringForm {
594                            key: key.to_string(),
595                            column,
596                            ty: schema.data_type.clone(),
597                        });
598                    }
599                }
600            }
601            Ok(value.to_string())
602        }
603        AnnotationFamily::RepartitionHint if key == REPARTITION_PARTITION_NUM_HINT_KEY => {
604            let value = value.trim();
605            if !matches!(value.parse::<u32>(), Ok(1..)) {
606                return Err(AnnotationValidationError::InvalidPartitionNumHint {
607                    value: value.to_string(),
608                });
609            }
610            Ok(value.to_string())
611        }
612        AnnotationFamily::RepartitionHint => {
613            let column_name = value.trim();
614            if column_name.is_empty() || column_name.contains(',') {
615                return Err(AnnotationValidationError::NotSingleColumn);
616            }
617            if !cx.partition_key_indices.is_empty() {
618                return Err(AnnotationValidationError::PartitionMetadataConflict);
619            }
620            let column_index = cx.schema.column_index_by_name(column_name).ok_or_else(|| {
621                AnnotationValidationError::ColumnNotFound {
622                    column: column_name.to_string(),
623                }
624            })?;
625            if cx.schema.timestamp_index() == Some(column_index) {
626                return Err(AnnotationValidationError::TimeIndexConflict);
627            }
628            Ok(column_name.to_string())
629        }
630    }
631}
632
633/// CREATE-side entry: validates every annotation option present in `options`
634/// and writes normalized values back in place.
635pub fn validate_and_normalize_annotation_options(
636    options: &mut TableOptions,
637    schema: &Schema,
638    partition_key_indices: &[usize],
639) -> std::result::Result<(), AnnotationValidationError> {
640    let cx = AnnotationContext {
641        data_model: options.data_model(),
642        schema,
643        partition_key_indices,
644    };
645    let mut normalized = Vec::new();
646    for (key, value) in &options.extra_options {
647        let Some(family) = AnnotationFamily::of_key(key) else {
648            continue;
649        };
650        let checked = validate_and_normalize_annotation(family, &cx, key, value)?;
651        if checked != *value {
652            normalized.push((key.clone(), checked));
653        }
654    }
655    for (key, value) in normalized {
656        options.extra_options.insert(key, value);
657    }
658    Ok(())
659}
660
661#[derive(Debug, Clone, Serialize, Deserialize)]
662pub enum AlterKind {
663    AddColumns {
664        columns: Vec<AddColumnRequest>,
665    },
666    DropColumns {
667        names: Vec<String>,
668    },
669    ModifyColumnTypes {
670        columns: Vec<ModifyColumnTypeRequest>,
671    },
672    SetJsonSettings {
673        request: SetJsonSettingsRequest,
674    },
675    RenameTable {
676        new_table_name: String,
677    },
678    SetTableOptions {
679        options: Vec<SetRegionOption>,
680    },
681    UnsetTableOptions {
682        keys: Vec<UnsetRegionOption>,
683    },
684    SetAnnotations {
685        family: AnnotationFamily,
686        options: Vec<(String, String)>,
687    },
688    UnsetAnnotations {
689        family: AnnotationFamily,
690        keys: Vec<String>,
691    },
692    SetIndexes {
693        options: Vec<SetIndexOption>,
694    },
695    UnsetIndexes {
696        options: Vec<UnsetIndexOption>,
697    },
698    DropDefaults {
699        names: Vec<String>,
700    },
701    SetDefaults {
702        defaults: Vec<SetDefaultRequest>,
703    },
704}
705
706#[derive(Debug, Clone, Serialize, Deserialize)]
707pub struct SetDefaultRequest {
708    pub column_name: String,
709    pub default_constraint: Option<ColumnDefaultConstraint>,
710}
711
712#[derive(Debug, Clone, Serialize, Deserialize)]
713pub enum SetIndexOption {
714    Fulltext {
715        column_name: String,
716        options: FulltextOptions,
717    },
718    Inverted {
719        column_name: String,
720    },
721    Skipping {
722        column_name: String,
723        options: SkippingIndexOptions,
724    },
725}
726
727impl SetIndexOption {
728    /// Returns the column name of the index option.
729    pub fn column_name(&self) -> &str {
730        match self {
731            SetIndexOption::Fulltext { column_name, .. } => column_name,
732            SetIndexOption::Inverted { column_name, .. } => column_name,
733            SetIndexOption::Skipping { column_name, .. } => column_name,
734        }
735    }
736}
737
738#[derive(Debug, Clone, Serialize, Deserialize)]
739pub enum UnsetIndexOption {
740    Fulltext { column_name: String },
741    Inverted { column_name: String },
742    Skipping { column_name: String },
743}
744
745impl UnsetIndexOption {
746    /// Returns the column name of the index option.
747    pub fn column_name(&self) -> &str {
748        match self {
749            UnsetIndexOption::Fulltext { column_name, .. } => column_name,
750            UnsetIndexOption::Inverted { column_name, .. } => column_name,
751            UnsetIndexOption::Skipping { column_name, .. } => column_name,
752        }
753    }
754}
755
756#[derive(Debug)]
757pub struct InsertRequest {
758    pub catalog_name: String,
759    pub schema_name: String,
760    pub table_name: String,
761    pub columns_values: HashMap<String, VectorRef>,
762    /// Whether this insert should skip WAL.
763    pub skip_wal: bool,
764}
765
766/// Delete (by primary key) request
767#[derive(Debug)]
768pub struct DeleteRequest {
769    pub catalog_name: String,
770    pub schema_name: String,
771    pub table_name: String,
772    /// Values of each column in this table's primary key and time index.
773    ///
774    /// The key is the column name, and the value is the column value.
775    pub key_column_values: HashMap<String, VectorRef>,
776}
777
778#[derive(Debug)]
779pub enum CopyDirection {
780    Export,
781    Import,
782}
783
784/// Copy table request
785#[derive(Debug)]
786pub struct CopyTableRequest {
787    pub catalog_name: String,
788    pub schema_name: String,
789    pub table_name: String,
790    pub location: String,
791    pub with: HashMap<String, String>,
792    pub connection: HashMap<String, String>,
793    pub pattern: Option<String>,
794    pub direction: CopyDirection,
795    pub timestamp_range: Option<TimestampRange>,
796    pub limit: Option<u64>,
797}
798
799#[derive(Debug, Clone, Default)]
800pub struct FlushTableRequest {
801    pub catalog_name: String,
802    pub schema_name: String,
803    pub table_name: String,
804}
805
806#[derive(Debug, Clone, Default)]
807pub struct BuildIndexTableRequest {
808    pub catalog_name: String,
809    pub schema_name: String,
810    pub table_name: String,
811}
812
813#[derive(Debug, Clone, PartialEq)]
814pub struct CompactTableRequest {
815    pub catalog_name: String,
816    pub schema_name: String,
817    pub table_name: String,
818    pub compact_options: compact_request::Options,
819    pub parallelism: u32,
820    pub time_range: Option<TimestampRange>,
821}
822
823impl Default for CompactTableRequest {
824    fn default() -> Self {
825        Self {
826            catalog_name: Default::default(),
827            schema_name: Default::default(),
828            table_name: Default::default(),
829            compact_options: compact_request::Options::Regular(Default::default()),
830            parallelism: 1,
831            time_range: None,
832        }
833    }
834}
835
836/// Truncate table request
837#[derive(Debug, Clone, Serialize, Deserialize)]
838pub struct TruncateTableRequest {
839    pub catalog_name: String,
840    pub schema_name: String,
841    pub table_name: String,
842    pub table_id: TableId,
843}
844
845impl TruncateTableRequest {
846    pub fn table_ref(&self) -> TableReference<'_> {
847        TableReference {
848            catalog: &self.catalog_name,
849            schema: &self.schema_name,
850            table: &self.table_name,
851        }
852    }
853}
854
855#[derive(Debug, Clone, Default, Deserialize, Serialize)]
856pub struct CopyDatabaseRequest {
857    pub catalog_name: String,
858    pub schema_name: String,
859    pub location: String,
860    pub with: HashMap<String, String>,
861    pub connection: HashMap<String, String>,
862    pub time_range: Option<TimestampRange>,
863}
864
865#[derive(Debug, Clone, Default, Deserialize, Serialize)]
866pub struct CopyQueryToRequest {
867    pub location: String,
868    pub with: HashMap<String, String>,
869    pub connection: HashMap<String, String>,
870}
871
872#[cfg(test)]
873mod tests {
874    use std::time::Duration;
875
876    use common_error::ext::ErrorExt;
877    use common_error::status_code::StatusCode;
878
879    use super::*;
880
881    #[test]
882    fn test_validate_annotation_keys() {
883        let column = REPARTITION_COLUMN_HINT_KEY;
884        let count = REPARTITION_PARTITION_NUM_HINT_KEY;
885        for (keys, expected) in [
886            (vec![], None),
887            (vec![TTL_KEY, TTL_KEY], None),
888            (vec!["repartition.unknown.hint"], None),
889            (vec![column], Some(AnnotationFamily::RepartitionHint)),
890            (vec![count], Some(AnnotationFamily::RepartitionHint)),
891            (vec![column, count], Some(AnnotationFamily::RepartitionHint)),
892            (vec![count, column], Some(AnnotationFamily::RepartitionHint)),
893            (
894                vec!["greptime.semantic.source", "greptime.semantic.source"],
895                Some(AnnotationFamily::Semantic),
896            ),
897        ] {
898            assert_eq!(validate_annotation_keys(keys), Ok(expected));
899        }
900        for keys in [
901            vec![column, column],
902            vec![count, count],
903            vec![column, count, column],
904        ] {
905            assert_eq!(
906                validate_annotation_keys(keys),
907                Err(AnnotationKeyError::DuplicateKey)
908            );
909        }
910        for keys in [
911            vec![column, TTL_KEY],
912            vec![TTL_KEY, count],
913            vec![column, "greptime.semantic.source"],
914        ] {
915            assert_eq!(
916                validate_annotation_keys(keys),
917                Err(AnnotationKeyError::MixedFamilies {
918                    family: AnnotationFamily::RepartitionHint,
919                })
920            );
921        }
922    }
923
924    #[test]
925    fn test_validate_table_option() {
926        assert!(validate_table_option(FILE_TABLE_LOCATION_KEY));
927        assert!(validate_table_option(FILE_TABLE_FORMAT_KEY));
928        assert!(validate_table_option(FILE_TABLE_PATTERN_KEY));
929        assert!(validate_table_option(TTL_KEY));
930        assert!(validate_table_option(WRITE_BUFFER_SIZE_KEY));
931        assert!(validate_table_option(STORAGE_KEY));
932        assert!(validate_table_option(MEMTABLE_BULK_MERGE_THRESHOLD));
933        assert!(validate_table_option(EXPERIMENTAL_SST_FLOAT_FIELD_ENCODING));
934        assert!(validate_table_option(REPARTITION_COLUMN_HINT_KEY));
935        assert!(validate_table_option(REPARTITION_PARTITION_NUM_HINT_KEY));
936        assert_eq!(AnnotationFamily::of_key("repartition.unknown.hint"), None);
937        assert!(!validate_table_option("foo"));
938
939        // Only whitelisted semantic keys are accepted.
940        assert!(validate_table_option(SEMANTIC_SIGNAL_TYPE));
941        assert!(validate_table_option(SEMANTIC_METRIC_TYPE));
942        // Unknown semantic key, near-miss, and the internal transport key are rejected.
943        assert!(!validate_table_option("greptime.semantic.future.key"));
944        assert!(!validate_table_option("greptime.semanticx"));
945        assert!(!validate_table_option(SEMANTIC_PER_TABLE_INDEX_KEY));
946    }
947
948    #[test]
949    fn test_validate_database_option() {
950        assert!(validate_database_option(MEMTABLE_TYPE));
951        assert!(validate_database_option(MEMTABLE_BULK_MERGE_THRESHOLD));
952        assert!(validate_database_option(MEMTABLE_BULK_ENCODE_ROW_THRESHOLD));
953        assert!(validate_database_option(
954            MEMTABLE_BULK_ENCODE_BYTES_THRESHOLD
955        ));
956        assert!(validate_database_option(MEMTABLE_BULK_MAX_MERGE_GROUPS));
957        assert!(validate_database_option(
958            "compaction.twcs.active_window.trigger_file_num"
959        ));
960        assert!(validate_database_option(
961            "compaction.twcs.active_window.l1_merge_trigger"
962        ));
963        assert!(validate_database_option(
964            "compaction.twcs.inactive_window.trigger_file_num"
965        ));
966        assert!(validate_database_option(
967            "compaction.twcs.inactive_window.l1_merge_trigger"
968        ));
969        assert!(validate_database_option(INGEST_ROWS_RATE_LIMIT_KEY));
970        assert!(validate_database_option("ingest_rows_rate_limit"));
971        assert!(!validate_database_option("foo"));
972    }
973
974    #[test]
975    fn test_parse_float_field_encoding_option() {
976        let options = TableOptions::try_from_iter([(
977            EXPERIMENTAL_SST_FLOAT_FIELD_ENCODING,
978            "byte_stream_split",
979        )])
980        .unwrap();
981        assert_eq!(
982            options
983                .extra_options
984                .get(EXPERIMENTAL_SST_FLOAT_FIELD_ENCODING),
985            Some(&"byte_stream_split".to_string())
986        );
987        assert!(
988            TableOptions::try_from_iter([(EXPERIMENTAL_SST_FLOAT_FIELD_ENCODING, "invalid",)])
989                .is_err()
990        );
991    }
992
993    #[test]
994    fn test_database_trigger_value_boundaries() {
995        let maximum = usize::MAX.to_string();
996        let overflow = format!("{maximum}0");
997        for (key, minimum) in [
998            (TWCS_TRIGGER_FILE_NUM, 0),
999            (TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM, 0),
1000            (TWCS_INACTIVE_WINDOW_TRIGGER_FILE_NUM, 0),
1001            (TWCS_ACTIVE_WINDOW_L1_MERGE_TRIGGER, 2),
1002            (TWCS_INACTIVE_WINDOW_L1_MERGE_TRIGGER, 2),
1003        ] {
1004            for invalid in [
1005                None,
1006                Some(""),
1007                Some("invalid"),
1008                Some("-1"),
1009                Some(overflow.as_str()),
1010            ] {
1011                assert!(
1012                    validate_database_option_value(key, invalid).is_err(),
1013                    "{key}: {invalid:?}"
1014                );
1015            }
1016            for valid in ["2", maximum.as_str()] {
1017                assert!(
1018                    validate_database_option_value(key, Some(valid)).is_ok(),
1019                    "{key}: {valid}"
1020                );
1021            }
1022            for boundary in ["0", "1"] {
1023                assert_eq!(
1024                    validate_database_option_value(key, Some(boundary)).is_ok(),
1025                    minimum == 0,
1026                    "{key}: {boundary}"
1027                );
1028            }
1029        }
1030    }
1031
1032    #[test]
1033    fn test_database_ingest_rate_limit_value_boundaries() {
1034        for invalid in [
1035            None,
1036            Some(""),
1037            Some("abc"),
1038            Some("1000/s"),
1039            Some("-1"),
1040            Some("1.5"),
1041            Some("18446744073709551616"),
1042        ] {
1043            assert!(validate_database_option_value(INGEST_ROWS_RATE_LIMIT_KEY, invalid).is_err());
1044        }
1045        let maximum = u64::MAX.to_string();
1046        for valid in ["0", "1", maximum.as_str()] {
1047            assert!(
1048                validate_database_option_value(INGEST_ROWS_RATE_LIMIT_KEY, Some(valid)).is_ok()
1049            );
1050        }
1051    }
1052
1053    #[test]
1054    fn test_serialize_table_options() {
1055        let options = TableOptions {
1056            write_buffer_size: None,
1057            ttl: Some(Duration::from_secs(1000).into()),
1058            extra_options: HashMap::new(),
1059            skip_wal: false,
1060        };
1061        let serialized = serde_json::to_string(&options).unwrap();
1062        let deserialized: TableOptions = serde_json::from_str(&serialized).unwrap();
1063        assert_eq!(options, deserialized);
1064    }
1065
1066    #[test]
1067    fn test_convert_hashmap_between_table_options() {
1068        let options = TableOptions {
1069            write_buffer_size: Some(ReadableSize::mb(128)),
1070            ttl: Some(Duration::from_secs(1000).into()),
1071            extra_options: HashMap::new(),
1072            skip_wal: false,
1073        };
1074        let serialized_map = HashMap::from(&options);
1075        let serialized = TableOptions::try_from_iter(&serialized_map).unwrap();
1076        assert_eq!(options, serialized);
1077
1078        let options = TableOptions {
1079            write_buffer_size: None,
1080            ttl: None,
1081            extra_options: HashMap::from([(SKIP_WAL_KEY.to_string(), true.to_string())]),
1082            skip_wal: true,
1083        };
1084        let serialized_map = HashMap::from(&options);
1085        assert_eq!(
1086            Some("true"),
1087            serialized_map.get(SKIP_WAL_KEY).map(String::as_str)
1088        );
1089        let serialized = TableOptions::try_from_iter(&serialized_map).unwrap();
1090        assert_eq!(options, serialized);
1091
1092        let options = TableOptions {
1093            write_buffer_size: None,
1094            ttl: Default::default(),
1095            extra_options: HashMap::new(),
1096            skip_wal: false,
1097        };
1098        let serialized_map = HashMap::from(&options);
1099        let serialized = TableOptions::try_from_iter(&serialized_map).unwrap();
1100        assert_eq!(options, serialized);
1101
1102        let options = TableOptions {
1103            write_buffer_size: Some(ReadableSize::mb(128)),
1104            ttl: Some(Duration::from_secs(1000).into()),
1105            extra_options: HashMap::from([("a".to_string(), "A".to_string())]),
1106            skip_wal: false,
1107        };
1108        let serialized_map = HashMap::from(&options);
1109        let serialized = TableOptions::try_from_iter(&serialized_map).unwrap();
1110        assert_eq!(options, serialized);
1111
1112        let options = TableOptions {
1113            extra_options: HashMap::from([(SKIP_WAL_KEY.to_string(), false.to_string())]),
1114            skip_wal: false,
1115            ..Default::default()
1116        };
1117        let serialized_map = HashMap::from(&options);
1118        let serialized = TableOptions::try_from_iter(&serialized_map).unwrap();
1119        assert_eq!(options, serialized);
1120    }
1121
1122    #[test]
1123    fn test_table_options_normalizes_twcs_trigger_aliases() {
1124        for options in [
1125            vec![(TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM, "4")],
1126            vec![
1127                (TWCS_TRIGGER_FILE_NUM, "4"),
1128                (TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM, "4"),
1129            ],
1130        ] {
1131            let table_options = TableOptions::try_from_iter(options).unwrap();
1132            assert_eq!(
1133                HashMap::from([(TWCS_TRIGGER_FILE_NUM.to_string(), "4".to_string())]),
1134                table_options.extra_options
1135            );
1136        }
1137    }
1138
1139    #[test]
1140    fn test_table_options_rejects_conflicting_twcs_trigger_aliases() {
1141        let error = TableOptions::try_from_iter([
1142            (TWCS_TRIGGER_FILE_NUM, "4"),
1143            (TWCS_ACTIVE_WINDOW_TRIGGER_FILE_NUM, "8"),
1144        ])
1145        .unwrap_err();
1146        assert_eq!(StatusCode::InvalidArguments, error.status_code());
1147        assert_eq!(
1148            "Conflicting table options: compaction.twcs.trigger_file_num=4 and compaction.twcs.active_window.trigger_file_num=8",
1149            error.to_string()
1150        );
1151    }
1152
1153    #[test]
1154    fn test_table_options_to_string() {
1155        let options = TableOptions {
1156            write_buffer_size: Some(ReadableSize::mb(128)),
1157            ttl: Some(Duration::from_secs(1000).into()),
1158            extra_options: HashMap::new(),
1159            skip_wal: false,
1160        };
1161
1162        assert_eq!(
1163            "write_buffer_size=128.0MiB ttl=16m 40s",
1164            options.to_string()
1165        );
1166
1167        let options = TableOptions {
1168            write_buffer_size: Some(ReadableSize::mb(128)),
1169            ttl: Some(Duration::from_secs(1000).into()),
1170            extra_options: HashMap::from([("a".to_string(), "A".to_string())]),
1171            skip_wal: false,
1172        };
1173
1174        assert_eq!(
1175            "write_buffer_size=128.0MiB ttl=16m 40s a=A",
1176            options.to_string()
1177        );
1178
1179        let options = TableOptions {
1180            write_buffer_size: Some(ReadableSize::mb(128)),
1181            ttl: Some(Duration::from_secs(1000).into()),
1182            extra_options: HashMap::new(),
1183            skip_wal: true,
1184        };
1185        assert_eq!(
1186            "write_buffer_size=128.0MiB ttl=16m 40s skip_wal=true",
1187            options.to_string()
1188        );
1189
1190        let options = TableOptions {
1191            write_buffer_size: None,
1192            ttl: None,
1193            extra_options: HashMap::from([(SKIP_WAL_KEY.to_string(), "false".to_string())]),
1194            skip_wal: false,
1195        };
1196        assert_eq!("skip_wal=false", options.to_string());
1197    }
1198}
table/requests.rs

table/
requests.rs
