risingwave_meta_model/
index.rs

1// Copyright 2025 RisingWave Labs
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
15use risingwave_pb::catalog::PbIndex;
16use sea_orm::ActiveValue::Set;
17use sea_orm::entity::prelude::*;
18use serde::{Deserialize, Serialize};
19
20use crate::{ExprNodeArray, IndexColumnPropertiesArray, IndexId, TableId};
21
22#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
23#[sea_orm(table_name = "index")]
24pub struct Model {
25    #[sea_orm(primary_key, auto_increment = false)]
26    pub index_id: IndexId,
27    pub name: String,
28    pub index_table_id: TableId,
29    pub primary_table_id: TableId,
30    pub index_items: ExprNodeArray,
31    pub index_column_properties: Option<IndexColumnPropertiesArray>,
32    pub index_columns_len: i32,
33}
34
35#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
36pub enum Relation {
37    #[sea_orm(
38        belongs_to = "super::object::Entity",
39        from = "Column::IndexId",
40        to = "super::object::Column::Oid",
41        on_update = "NoAction",
42        on_delete = "Cascade"
43    )]
44    Object,
45    #[sea_orm(
46        belongs_to = "super::table::Entity",
47        from = "Column::IndexTableId",
48        to = "super::table::Column::TableId",
49        on_update = "NoAction",
50        on_delete = "NoAction"
51    )]
52    Table2,
53    #[sea_orm(
54        belongs_to = "super::table::Entity",
55        from = "Column::PrimaryTableId",
56        to = "super::table::Column::TableId",
57        on_update = "NoAction",
58        on_delete = "NoAction"
59    )]
60    Table1,
61}
62
63impl Related<super::object::Entity> for Entity {
64    fn to() -> RelationDef {
65        Relation::Object.def()
66    }
67}
68
69impl ActiveModelBehavior for ActiveModel {}
70
71impl From<PbIndex> for ActiveModel {
72    fn from(pb_index: PbIndex) -> Self {
73        Self {
74            index_id: Set(pb_index.id as _),
75            name: Set(pb_index.name),
76            index_table_id: Set(pb_index.index_table_id as _),
77            primary_table_id: Set(pb_index.primary_table_id as _),
78            index_items: Set(pb_index.index_item.into()),
79            index_columns_len: Set(pb_index.index_columns_len as _),
80            index_column_properties: Set(Some(pb_index.index_column_properties.into())),
81        }
82    }
83}
risingwave_meta_model/index.rs

risingwave_meta_model/
index.rs
