Implement SQL text -> Garble enum conversion in Postgres example

sine-fdn · Jun 11, 2024 · ce0cefa · ce0cefa
1 parent 11420c1
commit ce0cefa
Show file tree

Hide file tree

Showing 5 changed files with 47 additions and 12 deletions.
diff --git a/examples/postgres-integration/.example.garble.rs b/examples/postgres-integration/.example.garble.rs
@@ -3,19 +3,34 @@ const ROWS_1: usize = PARTY_1::ROWS;
 const ROWS_RESULT: usize = max(PARTY_0::ROWS, PARTY_1::ROWS);
 const STR_LEN: usize = max(PARTY_0::STR_LEN, PARTY_1::STR_LEN);
 
+enum InsuranceStatus {
+    Foo,
+    Bar,
+    FooBar,
+}
+
 pub fn main(
     residents: [(i32, [u8; STR_LEN], i32); ROWS_0],
-    insurance: [(i32, [u8; STR_LEN], bool); ROWS_1],
+    insurance: [(i32, [u8; STR_LEN], InsuranceStatus); ROWS_1],
 ) -> [[u8; STR_LEN]; ROWS_RESULT] {
     let mut result = [[0u8; STR_LEN]; ROWS_RESULT];
     let mut i = 0usize;
     for resident in residents {
         for insurance in insurance {
             let (_, name0, age) = resident;
-            let (_, name1, health_problems) = insurance;
-            if name0 == name1 && health_problems == true && age > 65i32 {
-                result[i] = name0;
-                i = i + 1usize;
+            let (_, name1, insurance_status) = insurance;
+            if name0 == name1 && age > 65i32 {
+                match insurance_status {
+                    InsuranceStatus::Foo => {
+                        result[i] = name0;
+                        i = i + 1usize;
+                    }
+                    InsuranceStatus::Bar => {
+                        result[i] = name0;
+                        i = i + 1usize;
+                    }
+                    _ => {}
+                }
             }
         }
     }

diff --git a/examples/postgres-integration/policies1.json b/examples/postgres-integration/policies1.json
@@ -9,7 +9,7 @@
       "program": ".example.garble.rs",
       "leader": 0,
       "party": 1,
-      "input": "SELECT id, name, disabled FROM insurance",
+      "input": "SELECT id, name, status FROM insurance",
       "input_db": "postgres://postgres:test@localhost:5551/postgres",
       "constants": {
         "ROWS": {

diff --git a/examples/postgres-integration/seeds/db1/init.sql b/examples/postgres-integration/seeds/db1/init.sql
@@ -1,7 +1,7 @@
 CREATE TABLE insurance (
     id SERIAL PRIMARY KEY,
     name TEXT,
-    disabled BOOL,
+    status TEXT,
     address TEXT
 );
 

diff --git a/examples/postgres-integration/seeds/db1/insurance.csv b/examples/postgres-integration/seeds/db1/insurance.csv
@@ -1,3 +1,4 @@
-Id,Name,Disabled,Address
-0,"Max Doe",true,"Parkstraße 1, 12345 Berlin"
-1,"Bob Doe",false,"Schlossallee 2, 12345 Berlin"
+Id,Name,Status,Address
+0,"Max Doe","foo","Parkstraße 1, 12345 Berlin"
+1,"Bob Doe","bar","Schlossallee 2, 12345 Berlin"
+2,"Jane Doe","foobar","Schlossallee 2, 12345 Berlin"
diff --git a/examples/postgres-integration/src/main.rs b/examples/postgres-integration/src/main.rs
@@ -12,9 +12,9 @@ use parlay::{
     channel::Channel,
     fpre::fpre,
     garble_lang::{
-        ast::Type,
+        ast::{Type, Variant},
         compile_with_constants,
-        literal::Literal,
+        literal::{Literal, VariantLiteral},
         token::{SignedNumType, UnsignedNumType},
     },
     protocol::{mpc, Preprocessor},
@@ -507,6 +507,25 @@ async fn execute_mpc(
                             literal = Some(Literal::Array(fixed_str))
                         }
                     }
+                    Type::Enum(name) => {
+                        let Some(enum_def) = prg.program.enum_defs.get(name) else {
+                            bail!("Could not find definition for enum {name} in program");
+                        };
+                        if let Ok(variant) = row.try_get::<String, _>(c) {
+                            for v in &enum_def.variants {
+                                if let Variant::Unit(variant_name) = v {
+                                    if variant_name.to_lowercase() == variant.to_lowercase() {
+                                        literal = Some(Literal::Enum(
+                                            name.clone(),
+                                            variant_name.clone(),
+                                            VariantLiteral::Unit,
+                                        ));
+                                        break;
+                                    }
+                                }
+                            }
+                        }
+                    }
                     _ => {}
                 }
                 if let Some(literal) = literal {