fix: permission and sql for query sample

Signed-off-by: cutecutecat <junyuchen@tensorchord.ai>

Author: cutecutecat

src/index/gucs.rs

@@ -170,7 +170,7 @@ pub fn init() {
         c"`query_sampling_enable` argument of vchordrq.",
         &VCHORDRQ_QUERY_SAMPLING_ENABLE,
         GucContext::Userset,
-        GucFlags::default(),
+        GucFlags::default().intersection(GucFlags::SUPERUSER_ONLY),
     );
     GucRegistry::define_int_guc(
         c"vchordrq.query_sampling_max_records",
@@ -180,7 +180,7 @@ pub fn init() {
         0,
         10000,
         GucContext::Userset,
-        GucFlags::default(),
+        GucFlags::default().intersection(GucFlags::SUPERUSER_ONLY),
     );
     GucRegistry::define_float_guc(
         c"vchordrq.query_sampling_rate",
@@ -190,7 +190,7 @@ pub fn init() {
         0.0,
         1.0,
         GucContext::Userset,
-        GucFlags::default(),
+        GucFlags::default().intersection(GucFlags::SUPERUSER_ONLY),
     );
     unsafe {
         #[cfg(any(feature = "pg13", feature = "pg14"))]

src/sql/finalize.sql

@@ -152,7 +152,7 @@ RETURNS TABLE(
     index_name NAME,
     table_name NAME,
     column_name NAME,
-    operator TEXT,
+    operator NAME,
     value TEXT
 )
 LANGUAGE plpgsql
@@ -179,12 +179,7 @@ BEGIN
                 I.relname AS index_name,
                 C.relname AS table_name,
                 PA.attname AS column_name,
-                CASE
-                    WHEN OP.opcname LIKE '%%l2%%' THEN '<->'
-                    WHEN OP.opcname LIKE '%%ip%%' THEN '<#>'
-                    WHEN OP.opcname LIKE '%%cosine%%' THEN '<=>'
-                    ELSE ''
-                END AS operator
+                OP.oprname AS operator
             FROM
                 pg_catalog.pg_index X
             JOIN
@@ -196,11 +191,16 @@ BEGIN
             JOIN
                 pg_catalog.pg_am A ON A.oid = I.relam
             LEFT JOIN
-                pg_catalog.pg_opclass AS OP ON OP.oid = X.indclass[0]
+                pg_catalog.pg_opclass AS OPC ON OPC.oid = X.indclass[0]
+            LEFT JOIN
+                pg_catalog.pg_amop AO ON OPC.opcfamily = AO.amopfamily
+            LEFT JOIN
+                pg_catalog.pg_operator OP ON OP.oid = AO.amopopr
             LEFT JOIN
                 pg_catalog.pg_attribute PA ON PA.attrelid = X.indrelid AND PA.attnum = X.indkey[0]
             WHERE
                 A.amname = 'vchordrq'
+                AND AO.amoplefttype = AO.amoprighttype
                 AND C.relkind = 'r'
                 AND X.indnatts = 1
                 AND X.indexrelid = %1$s

tests/vchordrq/recall.slt

@@ -114,7 +114,7 @@ FROM (
     SELECT
         vchordrq_evaluate_query_recall(
             query => format(
-                'SELECT ctid FROM %I.%I ORDER BY %I %s ''%s'' LIMIT 10',
+                'SELECT ctid FROM %I.%I ORDER BY %I %s %L LIMIT 10',
                 lq.schema_name,
                 lq.table_name,
                 lq.column_name,