feat(indexes): Support for index column operator class (#1014)

cont'd #964

---------

Co-authored-by: daniel <[email protected]>

Author: tyt2y3

src/backend/postgres/index.rs

@@ -40,7 +40,7 @@ impl IndexBuilder for PostgresQueryBuilder {
         sql.write_str("INDEX ").unwrap();
 
         if create.concurrently {
-            write!(sql, "CONCURRENTLY ").unwrap();
+            sql.write_str("CONCURRENTLY ").unwrap();
         }
 
         if create.if_not_exists {
@@ -89,7 +89,7 @@ impl IndexBuilder for PostgresQueryBuilder {
         sql.write_str("DROP INDEX ").unwrap();
 
         if drop.concurrently {
-            write!(sql, "CONCURRENTLY ").unwrap();
+            sql.write_str("CONCURRENTLY ").unwrap();
         }
 
         if drop.if_exists {
@@ -167,6 +167,11 @@ impl IndexBuilder for PostgresQueryBuilder {
                         }
                     }
                 }
+
+                if let Some(operator_class) = col.operator_class() {
+                    sql.write_str(" ").unwrap();
+                    sql.write_str(&operator_class.0).unwrap();
+                }
             }
         );
 

src/index/common.rs

@@ -20,12 +20,14 @@ pub struct IndexColumnTableColumn {
     pub(crate) name: DynIden,
     pub(crate) prefix: Option<u32>,
     pub(crate) order: Option<IndexOrder>,
+    pub(crate) operator_class: Option<DynIden>,
 }
 
 #[derive(Debug, Clone)]
 pub struct IndexColumnExpr {
     pub(crate) expr: Expr,
     pub(crate) order: Option<IndexOrder>,
+    pub(crate) operator_class: Option<DynIden>,
 }
 
 impl IndexColumn {
@@ -36,6 +38,28 @@ impl IndexColumn {
             IndexColumn::Expr(_) => None,
         }
     }
+
+    pub(crate) fn operator_class(&self) -> &Option<DynIden> {
+        match self {
+            IndexColumn::TableColumn(IndexColumnTableColumn { operator_class, .. }) => {
+                operator_class
+            }
+            IndexColumn::Expr(IndexColumnExpr { operator_class, .. }) => operator_class,
+        }
+    }
+
+    /// Set index operator class. Only available on Postgres.
+    pub fn with_operator_class<I: IntoIden>(mut self, operator_class: I) -> Self {
+        match self {
+            IndexColumn::TableColumn(ref mut index_column_table_column) => {
+                index_column_table_column.operator_class = Some(operator_class.into_iden());
+            }
+            IndexColumn::Expr(ref mut index_column_expr) => {
+                index_column_expr.operator_class = Some(operator_class.into_iden())
+            }
+        };
+        self
+    }
 }
 
 #[derive(Debug, Clone)]
@@ -67,6 +91,7 @@ where
             name: value.into_iden(),
             prefix: None,
             order: None,
+            operator_class: None,
         })
     }
 }
@@ -80,6 +105,7 @@ where
             name: value.0.into_iden(),
             prefix: Some(value.1),
             order: None,
+            operator_class: None,
         })
     }
 }
@@ -93,6 +119,7 @@ where
             name: value.0.into_iden(),
             prefix: None,
             order: Some(value.1),
+            operator_class: None,
         })
     }
 }
@@ -106,6 +133,7 @@ where
             name: value.0.into_iden(),
             prefix: Some(value.1),
             order: Some(value.2),
+            operator_class: None,
         })
     }
 }
@@ -115,6 +143,7 @@ impl From<FunctionCall> for IndexColumn {
         IndexColumn::Expr(IndexColumnExpr {
             expr: value.into(),
             order: None,
+            operator_class: None,
         })
     }
 }
@@ -124,6 +153,7 @@ impl From<(FunctionCall, IndexOrder)> for IndexColumn {
         IndexColumn::Expr(IndexColumnExpr {
             expr: value.0.into(),
             order: Some(value.1),
+            operator_class: None,
         })
     }
 }
@@ -133,6 +163,7 @@ impl From<Expr> for IndexColumn {
         IndexColumn::Expr(IndexColumnExpr {
             expr: value,
             order: None,
+            operator_class: None,
         })
     }
 }
@@ -142,6 +173,7 @@ impl From<(Expr, IndexOrder)> for IndexColumn {
         IndexColumn::Expr(IndexColumnExpr {
             expr: value.0,
             order: Some(value.1),
+            operator_class: None,
         })
     }
 }

tests/postgres/index.rs

@@ -157,6 +157,23 @@ fn create_11() {
     )
 }
 
+#[test]
+fn create_12() {
+    assert_eq!(
+        Index::create()
+            .if_not_exists()
+            .name("idx-glyph-image")
+            .table(Glyph::Table)
+            .col(
+                Glyph::Image
+                    .into_index_column()
+                    .with_operator_class("text_pattern_ops")
+            )
+            .to_string(PostgresQueryBuilder),
+        r#"CREATE INDEX IF NOT EXISTS "idx-glyph-image" ON "glyph" ("image" text_pattern_ops)"#
+    );
+}
+
 #[test]
 fn drop_1() {
     assert_eq!(