Compare commits
21 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 31519e8447 | |||
| 847e921b1c | |||
| e19e1921e5 | |||
| 94d011e729 | |||
| 263cf04ffb | |||
| 00375c2926 | |||
| 885b9b5e44 | |||
| 298645ffdb | |||
| 330280ba48 | |||
| 02e661d219 | |||
| f7163e2689 | |||
| 091007006d | |||
| 3d66a7fc3c | |||
| e1314496dd | |||
| 70a27b430d | |||
| e078b8a74b | |||
| c2c0e62c2d | |||
| ebb97b3509 | |||
| 5d18847f32 | |||
| 4a33e29628 | |||
| d8fc286e94 |
11
.test/tests.md
Normal file
11
.test/tests.md
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
# 🗒️ Test Report (punc/framework)
|
||||||
|
|
||||||
|
_Generated at Wed Mar 18 05:21:40 EDT 2026_
|
||||||
|
|
||||||
|
## Summary
|
||||||
|
|
||||||
|
| Lang | Status | Tests | Passed | Failed | Duration |
|
||||||
|
| :--- | :---: | :---: | :---: | :---: | ---: |
|
||||||
|
|
||||||
|
## Results
|
||||||
|
|
||||||
5
.vscode/extensions.json
vendored
Normal file
5
.vscode/extensions.json
vendored
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
{
|
||||||
|
"recommendations": [
|
||||||
|
"rust-lang.rust-analyzer"
|
||||||
|
]
|
||||||
|
}
|
||||||
@ -39,10 +39,6 @@ crate-type = ["cdylib", "lib"]
|
|||||||
name = "pgrx_embed_jspg"
|
name = "pgrx_embed_jspg"
|
||||||
path = "src/bin/pgrx_embed.rs"
|
path = "src/bin/pgrx_embed.rs"
|
||||||
|
|
||||||
[[bin]]
|
|
||||||
name = "ast_explore"
|
|
||||||
path = "src/bin/ast_explore.rs"
|
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
default = ["pg18"]
|
default = ["pg18"]
|
||||||
pg18 = ["pgrx/pg18", "pgrx-tests/pg18" ]
|
pg18 = ["pgrx/pg18", "pgrx-tests/pg18" ]
|
||||||
|
|||||||
@ -43,7 +43,7 @@ JSPG implements specific extensions to the Draft 2020-12 standard to support the
|
|||||||
#### A. Polymorphism & Referencing (`$ref`, `$family`, and Native Types)
|
#### A. Polymorphism & Referencing (`$ref`, `$family`, and Native Types)
|
||||||
* **Native Type Discrimination (`variations`)**: Schemas defined inside a Postgres `type` are Entities. The validator securely and implicitly manages their `"type"` property. If an entity inherits from `user`, incoming JSON can safely define `{"type": "person"}` without errors, thanks to `compiled_variations` inheritance.
|
* **Native Type Discrimination (`variations`)**: Schemas defined inside a Postgres `type` are Entities. The validator securely and implicitly manages their `"type"` property. If an entity inherits from `user`, incoming JSON can safely define `{"type": "person"}` without errors, thanks to `compiled_variations` inheritance.
|
||||||
* **Structural Inheritance & Viral Infection (`$ref`)**: `$ref` is used exclusively for structural inheritance, *never* for union creation. A Punc request schema that `$ref`s an Entity virally inherits all physical database polymorphism rules for that target.
|
* **Structural Inheritance & Viral Infection (`$ref`)**: `$ref` is used exclusively for structural inheritance, *never* for union creation. A Punc request schema that `$ref`s an Entity virally inherits all physical database polymorphism rules for that target.
|
||||||
* **Shape Polymorphism (`$family`)**: Auto-expands polymorphic API lists based on an abstract Descendants Graph. If `{"$family": "widget"}` is used, JSPG evaluates the JSON against every schema that `$ref`s widget.
|
* **Shape Polymorphism (`$family`)**: Auto-expands polymorphic API lists based on an abstract **Descendants Graph**. If `{"$family": "widget"}` is used, the Validator dynamically identifies *every* schema in the registry that `$ref`s `widget` (e.g., `stock.widget`, `task.widget`) and evaluates the JSON against all of them.
|
||||||
* **Strict Matches & Depth Heuristic**: Polymorphic structures MUST match exactly **one** schema permutation. If multiple inherited struct permutations pass, JSPG applies the **Depth Heuristic Tie-Breaker**, selecting the candidate deepest in the inheritance tree.
|
* **Strict Matches & Depth Heuristic**: Polymorphic structures MUST match exactly **one** schema permutation. If multiple inherited struct permutations pass, JSPG applies the **Depth Heuristic Tie-Breaker**, selecting the candidate deepest in the inheritance tree.
|
||||||
|
|
||||||
#### B. Dot-Notation Schema Resolution & Database Mapping
|
#### B. Dot-Notation Schema Resolution & Database Mapping
|
||||||
@ -103,6 +103,10 @@ The Queryer transforms Postgres into a pre-compiled Semantic Query Engine via th
|
|||||||
* **Array Inclusion**: `{"$in": [values]}`, `{"$nin": [values]}` use native `jsonb_array_elements_text()` bindings to enforce `IN` and `NOT IN` logic without runtime SQL injection risks.
|
* **Array Inclusion**: `{"$in": [values]}`, `{"$nin": [values]}` use native `jsonb_array_elements_text()` bindings to enforce `IN` and `NOT IN` logic without runtime SQL injection risks.
|
||||||
* **Text Matching (ILIKE)**: Evaluates `$eq` or `$ne` against string fields containing the `%` character natively into Postgres `ILIKE` and `NOT ILIKE` partial substring matches.
|
* **Text Matching (ILIKE)**: Evaluates `$eq` or `$ne` against string fields containing the `%` character natively into Postgres `ILIKE` and `NOT ILIKE` partial substring matches.
|
||||||
* **Type Casting**: Safely resolves dynamic combinations by casting values instantly into the physical database types mapped in the schema (e.g. parsing `uuid` bindings to `::uuid`, formatting DateTimes to `::timestamptz`, and numbers to `::numeric`).
|
* **Type Casting**: Safely resolves dynamic combinations by casting values instantly into the physical database types mapped in the schema (e.g. parsing `uuid` bindings to `::uuid`, formatting DateTimes to `::timestamptz`, and numbers to `::numeric`).
|
||||||
|
* **Polymorphic SQL Generation (`$family`)**: Compiles `$family` properties by analyzing the **Physical Database Variations**, *not* the schema descendants.
|
||||||
|
* **The Dot Convention**: When a schema requests `$family: "target.schema"`, the compiler extracts the base type (e.g. `schema`) and looks up its Physical Table definition.
|
||||||
|
* **Multi-Table Branching**: If the Physical Table is a parent to other tables (e.g. `organization` has variations `["organization", "bot", "person"]`), the compiler generates a dynamic `CASE WHEN type = '...' THEN ...` query, expanding into `JOIN`s for each variation.
|
||||||
|
* **Single-Table Bypass**: If the Physical Table is a leaf node with only one variation (e.g. `person` has variations `["person"]`), the compiler cleanly bypasses `CASE` generation and compiles a simple `SELECT` across the base table, as all schema extensions (e.g. `light.person`, `full.person`) are guaranteed to reside in the exact same physical row.
|
||||||
|
|
||||||
### The Stem Engine
|
### The Stem Engine
|
||||||
|
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@ -1,312 +0,0 @@
|
|||||||
[
|
|
||||||
{
|
|
||||||
"description": "Stem Engine Unit Tests",
|
|
||||||
"database": {
|
|
||||||
"puncs": [],
|
|
||||||
"enums": [],
|
|
||||||
"relations": [
|
|
||||||
{
|
|
||||||
"id": "rel1",
|
|
||||||
"type": "relation",
|
|
||||||
"constraint": "fk_contact_entity",
|
|
||||||
"source_type": "contact",
|
|
||||||
"source_columns": [
|
|
||||||
"entity_id"
|
|
||||||
],
|
|
||||||
"destination_type": "person",
|
|
||||||
"destination_columns": [
|
|
||||||
"id"
|
|
||||||
],
|
|
||||||
"prefix": null
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "rel2",
|
|
||||||
"type": "relation",
|
|
||||||
"constraint": "fk_relationship_target",
|
|
||||||
"source_type": "relationship",
|
|
||||||
"source_columns": [
|
|
||||||
"target_id",
|
|
||||||
"target_type"
|
|
||||||
],
|
|
||||||
"destination_type": "entity",
|
|
||||||
"destination_columns": [
|
|
||||||
"id",
|
|
||||||
"type"
|
|
||||||
],
|
|
||||||
"prefix": "target"
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"types": [
|
|
||||||
{
|
|
||||||
"name": "entity",
|
|
||||||
"hierarchy": [
|
|
||||||
"entity"
|
|
||||||
],
|
|
||||||
"schemas": [
|
|
||||||
{
|
|
||||||
"$id": "entity",
|
|
||||||
"type": "object",
|
|
||||||
"properties": {}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"name": "person",
|
|
||||||
"hierarchy": [
|
|
||||||
"person",
|
|
||||||
"entity"
|
|
||||||
],
|
|
||||||
"schemas": [
|
|
||||||
{
|
|
||||||
"$id": "person",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"name": "email_address",
|
|
||||||
"hierarchy": [
|
|
||||||
"email_address",
|
|
||||||
"entity"
|
|
||||||
],
|
|
||||||
"schemas": [
|
|
||||||
{
|
|
||||||
"$id": "email_address",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"name": "phone_number",
|
|
||||||
"hierarchy": [
|
|
||||||
"phone_number",
|
|
||||||
"entity"
|
|
||||||
],
|
|
||||||
"schemas": [
|
|
||||||
{
|
|
||||||
"$id": "phone_number",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"name": "relationship",
|
|
||||||
"relationship": true,
|
|
||||||
"hierarchy": [
|
|
||||||
"relationship",
|
|
||||||
"entity"
|
|
||||||
],
|
|
||||||
"schemas": [
|
|
||||||
{
|
|
||||||
"$id": "relationship",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"name": "contact",
|
|
||||||
"relationship": true,
|
|
||||||
"hierarchy": [
|
|
||||||
"contact",
|
|
||||||
"relationship",
|
|
||||||
"entity"
|
|
||||||
],
|
|
||||||
"schemas": [
|
|
||||||
{
|
|
||||||
"$id": "contact",
|
|
||||||
"$ref": "relationship",
|
|
||||||
"properties": {
|
|
||||||
"target": {
|
|
||||||
"oneOf": [
|
|
||||||
{
|
|
||||||
"$ref": "phone_number"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"$ref": "email_address"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"name": "save_person",
|
|
||||||
"schemas": [
|
|
||||||
{
|
|
||||||
"$id": "save_person.response",
|
|
||||||
"$ref": "person",
|
|
||||||
"properties": {
|
|
||||||
"contacts": {
|
|
||||||
"type": "array",
|
|
||||||
"items": {
|
|
||||||
"$ref": "contact"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"tests": [
|
|
||||||
{
|
|
||||||
"description": "correctly squashes deep oneOf refs through array paths",
|
|
||||||
"action": "compile",
|
|
||||||
"expect": {
|
|
||||||
"success": true,
|
|
||||||
"stems": {
|
|
||||||
"contact": {
|
|
||||||
"": {
|
|
||||||
"schema": {
|
|
||||||
"$id": "contact",
|
|
||||||
"$ref": "relationship",
|
|
||||||
"properties": {
|
|
||||||
"target": {
|
|
||||||
"oneOf": [
|
|
||||||
{
|
|
||||||
"$ref": "phone_number"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"$ref": "email_address"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"type": "contact"
|
|
||||||
},
|
|
||||||
"target#(type==\"email_address\")": {
|
|
||||||
"relation": "target_id",
|
|
||||||
"schema": {
|
|
||||||
"$id": "email_address",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "email_address"
|
|
||||||
},
|
|
||||||
"target#(type==\"phone_number\")": {
|
|
||||||
"relation": "target_id",
|
|
||||||
"schema": {
|
|
||||||
"$id": "phone_number",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "phone_number"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"email_address": {
|
|
||||||
"": {
|
|
||||||
"schema": {
|
|
||||||
"$id": "email_address",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "email_address"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"entity": {
|
|
||||||
"": {
|
|
||||||
"schema": {
|
|
||||||
"$id": "entity",
|
|
||||||
"properties": {},
|
|
||||||
"type": "object"
|
|
||||||
},
|
|
||||||
"type": "entity"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"person": {
|
|
||||||
"": {
|
|
||||||
"schema": {
|
|
||||||
"$id": "person",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "person"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"phone_number": {
|
|
||||||
"": {
|
|
||||||
"schema": {
|
|
||||||
"$id": "phone_number",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "phone_number"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"relationship": {
|
|
||||||
"": {
|
|
||||||
"schema": {
|
|
||||||
"$id": "relationship",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "relationship"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"save_person.response": {
|
|
||||||
"": {
|
|
||||||
"schema": {
|
|
||||||
"$id": "save_person.response",
|
|
||||||
"$ref": "person",
|
|
||||||
"properties": {
|
|
||||||
"contacts": {
|
|
||||||
"items": {
|
|
||||||
"$ref": "contact"
|
|
||||||
},
|
|
||||||
"type": "array"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"type": "person"
|
|
||||||
},
|
|
||||||
"contacts.#": {
|
|
||||||
"relation": "contacts_id",
|
|
||||||
"schema": {
|
|
||||||
"$id": "contact",
|
|
||||||
"$ref": "relationship",
|
|
||||||
"properties": {
|
|
||||||
"target": {
|
|
||||||
"oneOf": [
|
|
||||||
{
|
|
||||||
"$ref": "phone_number"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"$ref": "email_address"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"type": "contact"
|
|
||||||
},
|
|
||||||
"contacts.#.target#(type==\"email_address\")": {
|
|
||||||
"relation": "target_id",
|
|
||||||
"schema": {
|
|
||||||
"$id": "email_address",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "email_address"
|
|
||||||
},
|
|
||||||
"contacts.#.target#(type==\"phone_number\")": {
|
|
||||||
"relation": "target_id",
|
|
||||||
"schema": {
|
|
||||||
"$id": "phone_number",
|
|
||||||
"$ref": "entity",
|
|
||||||
"properties": {}
|
|
||||||
},
|
|
||||||
"type": "phone_number"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
@ -1,17 +0,0 @@
|
|||||||
use sqlparser::dialect::PostgreSqlDialect;
|
|
||||||
use sqlparser::parser::Parser;
|
|
||||||
use std::env;
|
|
||||||
|
|
||||||
fn main() {
|
|
||||||
let sql = "SELECT t1_obj_t1_addresses_t1_target_t2.archived, t1.id FROM person t1 JOIN address t1_obj_t1_addresses ON true";
|
|
||||||
let dialect = PostgreSqlDialect {};
|
|
||||||
|
|
||||||
match Parser::parse_sql(&dialect, sql) {
|
|
||||||
Ok(ast) => {
|
|
||||||
println!("{:#?}", ast);
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
println!("Error: {:?}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -24,20 +24,28 @@ impl DatabaseExecutor for SpiExecutor {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Spi::connect(|client| {
|
pgrx::PgTryBuilder::new(|| {
|
||||||
match client.select(sql, Some(args_with_oid.len() as i64), &args_with_oid) {
|
Spi::connect(|client| {
|
||||||
Ok(tup_table) => {
|
pgrx::notice!("JSPG_SQL: {}", sql);
|
||||||
let mut results = Vec::new();
|
match client.select(sql, Some(args_with_oid.len() as i64), &args_with_oid) {
|
||||||
for row in tup_table {
|
Ok(tup_table) => {
|
||||||
if let Ok(Some(jsonb)) = row.get::<pgrx::JsonB>(1) {
|
let mut results = Vec::new();
|
||||||
results.push(jsonb.0);
|
for row in tup_table {
|
||||||
|
if let Ok(Some(jsonb)) = row.get::<pgrx::JsonB>(1) {
|
||||||
|
results.push(jsonb.0);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Ok(Value::Array(results))
|
||||||
}
|
}
|
||||||
Ok(Value::Array(results))
|
Err(e) => Err(format!("SPI Query Fetch Failure: {}", e)),
|
||||||
}
|
}
|
||||||
Err(e) => Err(format!("SPI Query Fetch Failure: {}", e)),
|
})
|
||||||
}
|
|
||||||
})
|
})
|
||||||
|
.catch_others(|cause| {
|
||||||
|
pgrx::warning!("JSPG Caught Native Postgres Error: {:?}", cause);
|
||||||
|
Err(format!("{:?}", cause))
|
||||||
|
})
|
||||||
|
.execute()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn execute(&self, sql: &str, args: Option<&[Value]>) -> Result<(), String> {
|
fn execute(&self, sql: &str, args: Option<&[Value]>) -> Result<(), String> {
|
||||||
@ -52,12 +60,20 @@ impl DatabaseExecutor for SpiExecutor {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Spi::connect_mut(|client| {
|
pgrx::PgTryBuilder::new(|| {
|
||||||
match client.update(sql, Some(args_with_oid.len() as i64), &args_with_oid) {
|
Spi::connect_mut(|client| {
|
||||||
Ok(_) => Ok(()),
|
pgrx::notice!("JSPG_SQL: {}", sql);
|
||||||
Err(e) => Err(format!("SPI Execution Failure: {}", e)),
|
match client.update(sql, Some(args_with_oid.len() as i64), &args_with_oid) {
|
||||||
}
|
Ok(_) => Ok(()),
|
||||||
|
Err(e) => Err(format!("SPI Execution Failure: {}", e)),
|
||||||
|
}
|
||||||
|
})
|
||||||
})
|
})
|
||||||
|
.catch_others(|cause| {
|
||||||
|
pgrx::warning!("JSPG Caught Native Postgres Error: {:?}", cause);
|
||||||
|
Err(format!("{:?}", cause))
|
||||||
|
})
|
||||||
|
.execute()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn auth_user_id(&self) -> Result<String, String> {
|
fn auth_user_id(&self) -> Result<String, String> {
|
||||||
|
|||||||
@ -18,24 +18,19 @@ use executors::pgrx::SpiExecutor;
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
use executors::mock::MockExecutor;
|
use executors::mock::MockExecutor;
|
||||||
|
|
||||||
pub mod stem;
|
|
||||||
use punc::Punc;
|
use punc::Punc;
|
||||||
use relation::Relation;
|
use relation::Relation;
|
||||||
use schema::Schema;
|
use schema::Schema;
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
use std::collections::{HashMap, HashSet};
|
use std::collections::{HashMap, HashSet};
|
||||||
use std::sync::Arc;
|
|
||||||
use stem::Stem;
|
|
||||||
use r#type::Type;
|
use r#type::Type;
|
||||||
|
|
||||||
pub struct Database {
|
pub struct Database {
|
||||||
pub enums: HashMap<String, Enum>,
|
pub enums: HashMap<String, Enum>,
|
||||||
pub types: HashMap<String, Type>,
|
pub types: HashMap<String, Type>,
|
||||||
pub puncs: HashMap<String, Punc>,
|
pub puncs: HashMap<String, Punc>,
|
||||||
pub relations: HashMap<String, Relation>,
|
pub relations: Vec<Relation>,
|
||||||
pub schemas: HashMap<String, Schema>,
|
pub schemas: HashMap<String, Schema>,
|
||||||
// Map of Schema ID -> { Entity Type -> Target Subschema Arc }
|
|
||||||
pub stems: HashMap<String, HashMap<String, Arc<Stem>>>,
|
|
||||||
pub descendants: HashMap<String, Vec<String>>,
|
pub descendants: HashMap<String, Vec<String>>,
|
||||||
pub depths: HashMap<String, usize>,
|
pub depths: HashMap<String, usize>,
|
||||||
pub executor: Box<dyn DatabaseExecutor + Send + Sync>,
|
pub executor: Box<dyn DatabaseExecutor + Send + Sync>,
|
||||||
@ -46,10 +41,9 @@ impl Database {
|
|||||||
let mut db = Self {
|
let mut db = Self {
|
||||||
enums: HashMap::new(),
|
enums: HashMap::new(),
|
||||||
types: HashMap::new(),
|
types: HashMap::new(),
|
||||||
relations: HashMap::new(),
|
relations: Vec::new(),
|
||||||
puncs: HashMap::new(),
|
puncs: HashMap::new(),
|
||||||
schemas: HashMap::new(),
|
schemas: HashMap::new(),
|
||||||
stems: HashMap::new(),
|
|
||||||
descendants: HashMap::new(),
|
descendants: HashMap::new(),
|
||||||
depths: HashMap::new(),
|
depths: HashMap::new(),
|
||||||
#[cfg(not(test))]
|
#[cfg(not(test))]
|
||||||
@ -78,7 +72,11 @@ impl Database {
|
|||||||
for item in arr {
|
for item in arr {
|
||||||
match serde_json::from_value::<Relation>(item.clone()) {
|
match serde_json::from_value::<Relation>(item.clone()) {
|
||||||
Ok(def) => {
|
Ok(def) => {
|
||||||
db.relations.insert(def.constraint.clone(), def);
|
if db.types.contains_key(&def.source_type)
|
||||||
|
&& db.types.contains_key(&def.destination_type)
|
||||||
|
{
|
||||||
|
db.relations.push(def);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
Err(e) => println!("DATABASE RELATION PARSE FAILED: {:?}", e),
|
Err(e) => println!("DATABASE RELATION PARSE FAILED: {:?}", e),
|
||||||
}
|
}
|
||||||
@ -137,13 +135,11 @@ impl Database {
|
|||||||
self.executor.timestamp()
|
self.executor.timestamp()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Organizes the graph of the database, compiling regex, format functions, and caching relationships.
|
|
||||||
pub fn compile(&mut self) -> Result<(), crate::drop::Drop> {
|
pub fn compile(&mut self) -> Result<(), crate::drop::Drop> {
|
||||||
self.collect_schemas();
|
self.collect_schemas();
|
||||||
self.collect_depths();
|
self.collect_depths();
|
||||||
self.collect_descendants();
|
self.collect_descendants();
|
||||||
self.compile_schemas();
|
self.compile_schemas();
|
||||||
self.collect_stems()?;
|
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@ -226,6 +222,79 @@ impl Database {
|
|||||||
self.descendants = descendants;
|
self.descendants = descendants;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn get_relation(
|
||||||
|
&self,
|
||||||
|
parent_type: &str,
|
||||||
|
child_type: &str,
|
||||||
|
prop_name: &str,
|
||||||
|
relative_keys: Option<&Vec<String>>,
|
||||||
|
) -> Option<(&Relation, bool)> {
|
||||||
|
if parent_type == "entity" && child_type == "entity" {
|
||||||
|
return None; // Ignore entity <-> entity generic fallbacks, they aren't useful edges
|
||||||
|
}
|
||||||
|
|
||||||
|
let p_def = self.types.get(parent_type)?;
|
||||||
|
let c_def = self.types.get(child_type)?;
|
||||||
|
|
||||||
|
let mut matching_rels = Vec::new();
|
||||||
|
let mut directions = Vec::new();
|
||||||
|
|
||||||
|
for rel in &self.relations {
|
||||||
|
let is_forward = p_def.hierarchy.contains(&rel.source_type)
|
||||||
|
&& c_def.hierarchy.contains(&rel.destination_type);
|
||||||
|
let is_reverse = p_def.hierarchy.contains(&rel.destination_type)
|
||||||
|
&& c_def.hierarchy.contains(&rel.source_type);
|
||||||
|
|
||||||
|
if is_forward {
|
||||||
|
matching_rels.push(rel);
|
||||||
|
directions.push(true);
|
||||||
|
} else if is_reverse {
|
||||||
|
matching_rels.push(rel);
|
||||||
|
directions.push(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if matching_rels.is_empty() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
if matching_rels.len() == 1 {
|
||||||
|
return Some((matching_rels[0], directions[0]));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut chosen_idx = 0;
|
||||||
|
let mut resolved = false;
|
||||||
|
|
||||||
|
// Reduce ambiguity with prefix
|
||||||
|
for (i, rel) in matching_rels.iter().enumerate() {
|
||||||
|
if let Some(prefix) = &rel.prefix {
|
||||||
|
if prefix == prop_name {
|
||||||
|
chosen_idx = i;
|
||||||
|
resolved = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reduce ambiguity by checking if relative payload OMITS the prefix (M:M heuristic)
|
||||||
|
if !resolved && relative_keys.is_some() {
|
||||||
|
let keys = relative_keys.unwrap();
|
||||||
|
let mut missing_prefix_ids = Vec::new();
|
||||||
|
for (i, rel) in matching_rels.iter().enumerate() {
|
||||||
|
if let Some(prefix) = &rel.prefix {
|
||||||
|
if !keys.contains(prefix) {
|
||||||
|
missing_prefix_ids.push(i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if missing_prefix_ids.len() == 1 {
|
||||||
|
chosen_idx = missing_prefix_ids[0];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Some((matching_rels[chosen_idx], directions[chosen_idx]))
|
||||||
|
}
|
||||||
|
|
||||||
fn collect_descendants_recursively(
|
fn collect_descendants_recursively(
|
||||||
target: &str,
|
target: &str,
|
||||||
direct_refs: &HashMap<String, Vec<String>>,
|
direct_refs: &HashMap<String, Vec<String>>,
|
||||||
@ -249,239 +318,4 @@ impl Database {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn collect_stems(&mut self) -> Result<(), crate::drop::Drop> {
|
|
||||||
let mut db_stems: HashMap<String, HashMap<String, Arc<Stem>>> = HashMap::new();
|
|
||||||
let mut errors: Vec<crate::drop::Error> = Vec::new();
|
|
||||||
|
|
||||||
let schema_ids: Vec<String> = self.schemas.keys().cloned().collect();
|
|
||||||
for schema_id in schema_ids {
|
|
||||||
if let Some(schema) = self.schemas.get(&schema_id) {
|
|
||||||
let mut inner_map = HashMap::new();
|
|
||||||
Self::discover_stems(
|
|
||||||
self,
|
|
||||||
&schema_id,
|
|
||||||
schema,
|
|
||||||
String::from(""),
|
|
||||||
None,
|
|
||||||
None,
|
|
||||||
false,
|
|
||||||
&mut inner_map,
|
|
||||||
Vec::new(),
|
|
||||||
&mut errors,
|
|
||||||
);
|
|
||||||
if !inner_map.is_empty() {
|
|
||||||
db_stems.insert(schema_id, inner_map);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
self.stems = db_stems;
|
|
||||||
|
|
||||||
if !errors.is_empty() {
|
|
||||||
return Err(crate::drop::Drop::with_errors(errors));
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn discover_stems(
|
|
||||||
db: &Database,
|
|
||||||
root_schema_id: &str,
|
|
||||||
schema: &Schema,
|
|
||||||
current_path: String,
|
|
||||||
parent_type: Option<String>,
|
|
||||||
property_name: Option<String>,
|
|
||||||
is_polymorphic: bool,
|
|
||||||
inner_map: &mut HashMap<String, Arc<Stem>>,
|
|
||||||
seen_entities: Vec<String>,
|
|
||||||
errors: &mut Vec<crate::drop::Error>,
|
|
||||||
) {
|
|
||||||
let mut is_entity = false;
|
|
||||||
let mut entity_type = String::new();
|
|
||||||
|
|
||||||
// First check if the Schema's $id is a native Database Type
|
|
||||||
if let Some(ref id) = schema.obj.id {
|
|
||||||
let parts: Vec<&str> = id.split('.').collect();
|
|
||||||
if let Some(last_seg) = parts.last() {
|
|
||||||
if db.types.contains_key(*last_seg) {
|
|
||||||
is_entity = true;
|
|
||||||
entity_type = last_seg.to_string();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// If not found via $id, check the $ref pointer
|
|
||||||
// This allows ad-hoc schemas (like `save_person.response`) to successfully adopt the Type of what they $ref
|
|
||||||
if !is_entity {
|
|
||||||
if let Some(ref r) = schema.obj.r#ref {
|
|
||||||
let parts: Vec<&str> = r.split('.').collect();
|
|
||||||
if let Some(last_seg) = parts.last() {
|
|
||||||
if db.types.contains_key(*last_seg) {
|
|
||||||
is_entity = true;
|
|
||||||
entity_type = last_seg.to_string();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if is_entity {
|
|
||||||
if seen_entities.contains(&entity_type) {
|
|
||||||
return; // Break cyclical schemas!
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut relation_col = None;
|
|
||||||
if is_entity {
|
|
||||||
if let (Some(pt), Some(prop)) = (&parent_type, &property_name) {
|
|
||||||
let expected_col = format!("{}_id", prop);
|
|
||||||
let mut found = false;
|
|
||||||
for rel in db.relations.values() {
|
|
||||||
if (rel.source_type == *pt && rel.destination_type == entity_type)
|
|
||||||
|| (rel.source_type == entity_type && rel.destination_type == *pt)
|
|
||||||
{
|
|
||||||
if rel.source_columns.contains(&expected_col) {
|
|
||||||
relation_col = Some(expected_col.clone());
|
|
||||||
found = true;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
relation_col = Some(expected_col);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut final_path = current_path.clone();
|
|
||||||
if is_polymorphic && !final_path.is_empty() && !final_path.ends_with(&entity_type) {
|
|
||||||
if final_path.ends_with(".#") {
|
|
||||||
final_path = format!("{}(type==\"{}\")", final_path, entity_type);
|
|
||||||
} else {
|
|
||||||
final_path = format!("{}#(type==\"{}\")", final_path, entity_type);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let stem = Stem {
|
|
||||||
r#type: entity_type.clone(),
|
|
||||||
relation: relation_col,
|
|
||||||
schema: Arc::new(schema.clone()),
|
|
||||||
};
|
|
||||||
inner_map.insert(final_path, Arc::new(stem));
|
|
||||||
}
|
|
||||||
|
|
||||||
let next_parent = if is_entity {
|
|
||||||
Some(entity_type.clone())
|
|
||||||
} else {
|
|
||||||
parent_type.clone()
|
|
||||||
};
|
|
||||||
|
|
||||||
let pass_seen = if is_entity {
|
|
||||||
let mut ns = seen_entities.clone();
|
|
||||||
ns.push(entity_type.clone());
|
|
||||||
ns
|
|
||||||
} else {
|
|
||||||
seen_entities.clone()
|
|
||||||
};
|
|
||||||
|
|
||||||
// Properties branch
|
|
||||||
if let Some(props) = &schema.obj.properties {
|
|
||||||
for (k, v) in props {
|
|
||||||
// Standard Property Pathing
|
|
||||||
let next_path = if current_path.is_empty() {
|
|
||||||
k.clone()
|
|
||||||
} else {
|
|
||||||
format!("{}.{}", current_path, k)
|
|
||||||
};
|
|
||||||
|
|
||||||
Self::discover_stems(
|
|
||||||
db,
|
|
||||||
root_schema_id,
|
|
||||||
v,
|
|
||||||
next_path,
|
|
||||||
next_parent.clone(),
|
|
||||||
Some(k.clone()),
|
|
||||||
false,
|
|
||||||
inner_map,
|
|
||||||
pass_seen.clone(),
|
|
||||||
errors,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Array Item branch
|
|
||||||
if let Some(items) = &schema.obj.items {
|
|
||||||
let next_path = if current_path.is_empty() {
|
|
||||||
String::from("#")
|
|
||||||
} else {
|
|
||||||
format!("{}.#", current_path)
|
|
||||||
};
|
|
||||||
|
|
||||||
Self::discover_stems(
|
|
||||||
db,
|
|
||||||
root_schema_id,
|
|
||||||
items,
|
|
||||||
next_path,
|
|
||||||
next_parent.clone(),
|
|
||||||
property_name.clone(),
|
|
||||||
false,
|
|
||||||
inner_map,
|
|
||||||
pass_seen.clone(),
|
|
||||||
errors,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Follow external reference if we didn't just crawl local properties
|
|
||||||
if schema.obj.properties.is_none() && schema.obj.items.is_none() && schema.obj.one_of.is_none()
|
|
||||||
{
|
|
||||||
if let Some(ref r) = schema.obj.r#ref {
|
|
||||||
if let Some(target_schema) = db.schemas.get(r) {
|
|
||||||
Self::discover_stems(
|
|
||||||
db,
|
|
||||||
root_schema_id,
|
|
||||||
target_schema,
|
|
||||||
current_path.clone(),
|
|
||||||
next_parent.clone(),
|
|
||||||
property_name.clone(),
|
|
||||||
is_polymorphic,
|
|
||||||
inner_map,
|
|
||||||
seen_entities.clone(),
|
|
||||||
errors,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Polymorphism branch
|
|
||||||
if let Some(arr) = &schema.obj.one_of {
|
|
||||||
for v in arr {
|
|
||||||
Self::discover_stems(
|
|
||||||
db,
|
|
||||||
root_schema_id,
|
|
||||||
v.as_ref(),
|
|
||||||
current_path.clone(),
|
|
||||||
next_parent.clone(),
|
|
||||||
property_name.clone(),
|
|
||||||
true,
|
|
||||||
inner_map,
|
|
||||||
pass_seen.clone(),
|
|
||||||
errors,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if let Some(arr) = &schema.obj.all_of {
|
|
||||||
for v in arr {
|
|
||||||
Self::discover_stems(
|
|
||||||
db,
|
|
||||||
root_schema_id,
|
|
||||||
v.as_ref(),
|
|
||||||
current_path.clone(),
|
|
||||||
next_parent.clone(),
|
|
||||||
property_name.clone(),
|
|
||||||
is_polymorphic,
|
|
||||||
inner_map,
|
|
||||||
pass_seen.clone(),
|
|
||||||
errors,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1,12 +0,0 @@
|
|||||||
use crate::database::schema::Schema;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct Stem {
|
|
||||||
pub r#type: String,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
pub relation: Option<String>,
|
|
||||||
|
|
||||||
pub schema: Arc<Schema>,
|
|
||||||
}
|
|
||||||
@ -67,6 +67,10 @@ pub struct Error {
|
|||||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||||
pub struct ErrorDetails {
|
pub struct ErrorDetails {
|
||||||
pub path: String,
|
pub path: String,
|
||||||
// Extensions can be added here (package, cause, etc)
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
// For now, validator only provides path
|
pub cause: Option<String>,
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub context: Option<Value>,
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub schema: Option<String>,
|
||||||
}
|
}
|
||||||
|
|||||||
25
src/lib.rs
25
src/lib.rs
@ -31,6 +31,9 @@ fn jspg_failure() -> JsonB {
|
|||||||
message: "JSPG extension has not been initialized via jspg_setup".to_string(),
|
message: "JSPG extension has not been initialized via jspg_setup".to_string(),
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: "".to_string(),
|
path: "".to_string(),
|
||||||
|
cause: None,
|
||||||
|
context: None,
|
||||||
|
schema: None,
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
let drop = crate::drop::Drop::with_errors(vec![error]);
|
let drop = crate::drop::Drop::with_errors(vec![error]);
|
||||||
@ -74,7 +77,7 @@ pub fn jspg_merge(data: JsonB) -> JsonB {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[cfg_attr(not(test), pg_extern)]
|
#[cfg_attr(not(test), pg_extern)]
|
||||||
pub fn jspg_query(schema_id: &str, stem: Option<&str>, filters: Option<JsonB>) -> JsonB {
|
pub fn jspg_query(schema_id: &str, filters: Option<JsonB>) -> JsonB {
|
||||||
let engine_opt = {
|
let engine_opt = {
|
||||||
let lock = GLOBAL_JSPG.read().unwrap();
|
let lock = GLOBAL_JSPG.read().unwrap();
|
||||||
lock.clone()
|
lock.clone()
|
||||||
@ -84,7 +87,7 @@ pub fn jspg_query(schema_id: &str, stem: Option<&str>, filters: Option<JsonB>) -
|
|||||||
Some(engine) => {
|
Some(engine) => {
|
||||||
let drop = engine
|
let drop = engine
|
||||||
.queryer
|
.queryer
|
||||||
.query(schema_id, stem, filters.as_ref().map(|f| &f.0));
|
.query(schema_id, filters.as_ref().map(|f| &f.0));
|
||||||
JsonB(serde_json::to_value(drop).unwrap())
|
JsonB(serde_json::to_value(drop).unwrap())
|
||||||
}
|
}
|
||||||
None => jspg_failure(),
|
None => jspg_failure(),
|
||||||
@ -128,24 +131,6 @@ pub fn jspg_schemas() -> JsonB {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg_attr(not(test), pg_extern)]
|
|
||||||
pub fn jspg_stems() -> JsonB {
|
|
||||||
let engine_opt = {
|
|
||||||
let lock = GLOBAL_JSPG.read().unwrap();
|
|
||||||
lock.clone()
|
|
||||||
};
|
|
||||||
|
|
||||||
match engine_opt {
|
|
||||||
Some(engine) => {
|
|
||||||
let stems_json = serde_json::to_value(&engine.database.stems)
|
|
||||||
.unwrap_or(serde_json::Value::Object(serde_json::Map::new()));
|
|
||||||
let drop = crate::drop::Drop::success_with_val(stems_json);
|
|
||||||
JsonB(serde_json::to_value(drop).unwrap())
|
|
||||||
}
|
|
||||||
None => jspg_failure(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg_attr(not(test), pg_extern(strict))]
|
#[cfg_attr(not(test), pg_extern(strict))]
|
||||||
pub fn jspg_teardown() -> JsonB {
|
pub fn jspg_teardown() -> JsonB {
|
||||||
let mut lock = GLOBAL_JSPG.write().unwrap();
|
let mut lock = GLOBAL_JSPG.write().unwrap();
|
||||||
|
|||||||
@ -21,27 +21,27 @@ impl Merger {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn merge(&self, data: Value) -> crate::drop::Drop {
|
pub fn merge(&self, data: Value) -> crate::drop::Drop {
|
||||||
let mut val_resolved = Value::Null;
|
|
||||||
let mut notifications_queue = Vec::new();
|
let mut notifications_queue = Vec::new();
|
||||||
|
|
||||||
let result = self.merge_internal(data, &mut notifications_queue);
|
|
||||||
|
|
||||||
match result {
|
let result = self.merge_internal(data.clone(), &mut notifications_queue);
|
||||||
Ok(val) => {
|
|
||||||
val_resolved = val;
|
let val_resolved = match result {
|
||||||
}
|
Ok(val) => val,
|
||||||
Err(msg) => {
|
Err(msg) => {
|
||||||
return crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
return crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
||||||
code: "MERGE_FAILED".to_string(),
|
code: "MERGE_FAILED".to_string(),
|
||||||
message: msg,
|
message: msg,
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: "".to_string(),
|
path: "".to_string(),
|
||||||
|
cause: None,
|
||||||
|
context: Some(data),
|
||||||
|
schema: None,
|
||||||
},
|
},
|
||||||
}]);
|
}]);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// Execute the globally collected, pre-ordered notifications last!
|
// Execute the globally collected, pre-ordered notifications last!
|
||||||
for notify_sql in notifications_queue {
|
for notify_sql in notifications_queue {
|
||||||
if let Err(e) = self.db.execute(¬ify_sql, None) {
|
if let Err(e) = self.db.execute(¬ify_sql, None) {
|
||||||
return crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
return crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
||||||
@ -49,6 +49,9 @@ impl Merger {
|
|||||||
message: format!("Executor Error in pre-ordered notify: {:?}", e),
|
message: format!("Executor Error in pre-ordered notify: {:?}", e),
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: "".to_string(),
|
path: "".to_string(),
|
||||||
|
cause: None,
|
||||||
|
context: None,
|
||||||
|
schema: None,
|
||||||
},
|
},
|
||||||
}]);
|
}]);
|
||||||
}
|
}
|
||||||
@ -82,7 +85,11 @@ impl Merger {
|
|||||||
crate::drop::Drop::success_with_val(stripped_val)
|
crate::drop::Drop::success_with_val(stripped_val)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn merge_internal(&self, data: Value, notifications: &mut Vec<String>) -> Result<Value, String> {
|
pub(crate) fn merge_internal(
|
||||||
|
&self,
|
||||||
|
data: Value,
|
||||||
|
notifications: &mut Vec<String>,
|
||||||
|
) -> Result<Value, String> {
|
||||||
match data {
|
match data {
|
||||||
Value::Array(items) => self.merge_array(items, notifications),
|
Value::Array(items) => self.merge_array(items, notifications),
|
||||||
Value::Object(map) => self.merge_object(map, notifications),
|
Value::Object(map) => self.merge_object(map, notifications),
|
||||||
@ -90,7 +97,11 @@ impl Merger {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn merge_array(&self, items: Vec<Value>, notifications: &mut Vec<String>) -> Result<Value, String> {
|
fn merge_array(
|
||||||
|
&self,
|
||||||
|
items: Vec<Value>,
|
||||||
|
notifications: &mut Vec<String>,
|
||||||
|
) -> Result<Value, String> {
|
||||||
let mut resolved_items = Vec::new();
|
let mut resolved_items = Vec::new();
|
||||||
for item in items {
|
for item in items {
|
||||||
let resolved = self.merge_internal(item, notifications)?;
|
let resolved = self.merge_internal(item, notifications)?;
|
||||||
@ -99,7 +110,11 @@ impl Merger {
|
|||||||
Ok(Value::Array(resolved_items))
|
Ok(Value::Array(resolved_items))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn merge_object(&self, obj: serde_json::Map<String, Value>, notifications: &mut Vec<String>) -> Result<Value, String> {
|
fn merge_object(
|
||||||
|
&self,
|
||||||
|
obj: serde_json::Map<String, Value>,
|
||||||
|
notifications: &mut Vec<String>,
|
||||||
|
) -> Result<Value, String> {
|
||||||
let queue_start = notifications.len();
|
let queue_start = notifications.len();
|
||||||
|
|
||||||
let type_name = match obj.get("type").and_then(|v| v.as_str()) {
|
let type_name = match obj.get("type").and_then(|v| v.as_str()) {
|
||||||
@ -158,10 +173,23 @@ impl Merger {
|
|||||||
_ => continue,
|
_ => continue,
|
||||||
};
|
};
|
||||||
|
|
||||||
let relative_relation = self.get_entity_relation(type_def, &relative, &relation_name)?;
|
// Attempt to extract relative object type name
|
||||||
|
let relative_type_name = match relative.get("type").and_then(|v| v.as_str()) {
|
||||||
|
Some(t) => t,
|
||||||
|
None => continue,
|
||||||
|
};
|
||||||
|
|
||||||
if let Some(relation) = relative_relation {
|
let relative_keys: Vec<String> = relative.keys().cloned().collect();
|
||||||
let parent_is_source = type_def.hierarchy.contains(&relation.source_type);
|
|
||||||
|
// Call central Database O(1) graph logic
|
||||||
|
let relative_relation = self.db.get_relation(
|
||||||
|
&type_def.name,
|
||||||
|
relative_type_name,
|
||||||
|
&relation_name,
|
||||||
|
Some(&relative_keys),
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Some((relation, parent_is_source)) = relative_relation {
|
||||||
|
|
||||||
if parent_is_source {
|
if parent_is_source {
|
||||||
// Parent holds FK to Child. Child MUST be generated FIRST.
|
// Parent holds FK to Child. Child MUST be generated FIRST.
|
||||||
@ -247,9 +275,23 @@ impl Merger {
|
|||||||
_ => continue,
|
_ => continue,
|
||||||
};
|
};
|
||||||
|
|
||||||
let relative_relation = self.get_entity_relation(type_def, first_relative, &relation_name)?;
|
// Attempt to extract relative object type name
|
||||||
|
let relative_type_name = match first_relative.get("type").and_then(|v| v.as_str()) {
|
||||||
|
Some(t) => t,
|
||||||
|
None => continue,
|
||||||
|
};
|
||||||
|
|
||||||
if let Some(relation) = relative_relation {
|
let relative_keys: Vec<String> = first_relative.keys().cloned().collect();
|
||||||
|
|
||||||
|
// Call central Database O(1) graph logic
|
||||||
|
let relative_relation = self.db.get_relation(
|
||||||
|
&type_def.name,
|
||||||
|
relative_type_name,
|
||||||
|
&relation_name,
|
||||||
|
Some(&relative_keys),
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Some((relation, _)) = relative_relation {
|
||||||
let mut relative_responses = Vec::new();
|
let mut relative_responses = Vec::new();
|
||||||
for relative_item_val in relative_arr {
|
for relative_item_val in relative_arr {
|
||||||
if let Value::Object(mut relative_item) = relative_item_val {
|
if let Value::Object(mut relative_item) = relative_item_val {
|
||||||
@ -266,10 +308,11 @@ impl Merger {
|
|||||||
&entity_fields,
|
&entity_fields,
|
||||||
);
|
);
|
||||||
|
|
||||||
let merged_relative = match self.merge_internal(Value::Object(relative_item), notifications)? {
|
let merged_relative =
|
||||||
Value::Object(m) => m,
|
match self.merge_internal(Value::Object(relative_item), notifications)? {
|
||||||
_ => continue,
|
Value::Object(m) => m,
|
||||||
};
|
_ => continue,
|
||||||
|
};
|
||||||
|
|
||||||
relative_responses.push(Value::Object(merged_relative));
|
relative_responses.push(Value::Object(merged_relative));
|
||||||
}
|
}
|
||||||
@ -760,101 +803,7 @@ impl Merger {
|
|||||||
changes
|
changes
|
||||||
}
|
}
|
||||||
|
|
||||||
fn reduce_entity_relations(
|
// Helper Functions
|
||||||
&self,
|
|
||||||
mut matching_relations: Vec<crate::database::relation::Relation>,
|
|
||||||
relative: &serde_json::Map<String, Value>,
|
|
||||||
relation_name: &str,
|
|
||||||
) -> Result<Option<crate::database::relation::Relation>, String> {
|
|
||||||
if matching_relations.is_empty() {
|
|
||||||
return Ok(None);
|
|
||||||
}
|
|
||||||
if matching_relations.len() == 1 {
|
|
||||||
return Ok(Some(matching_relations.pop().unwrap()));
|
|
||||||
}
|
|
||||||
|
|
||||||
let exact_match: Vec<_> = matching_relations
|
|
||||||
.iter()
|
|
||||||
.filter(|r| r.prefix.as_deref() == Some(relation_name))
|
|
||||||
.cloned()
|
|
||||||
.collect();
|
|
||||||
if exact_match.len() == 1 {
|
|
||||||
return Ok(Some(exact_match.into_iter().next().unwrap()));
|
|
||||||
}
|
|
||||||
|
|
||||||
matching_relations.retain(|r| {
|
|
||||||
if let Some(prefix) = &r.prefix {
|
|
||||||
!relative.contains_key(prefix)
|
|
||||||
} else {
|
|
||||||
true
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
if matching_relations.len() == 1 {
|
|
||||||
Ok(Some(matching_relations.pop().unwrap()))
|
|
||||||
} else {
|
|
||||||
let constraints: Vec<_> = matching_relations
|
|
||||||
.iter()
|
|
||||||
.map(|r| r.constraint.clone())
|
|
||||||
.collect();
|
|
||||||
Err(format!(
|
|
||||||
"AMBIGUOUS_TYPE_RELATIONS: Could not reduce ambiguous type relations: {}",
|
|
||||||
constraints.join(", ")
|
|
||||||
))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_entity_relation(
|
|
||||||
&self,
|
|
||||||
entity_type: &crate::database::r#type::Type,
|
|
||||||
relative: &serde_json::Map<String, Value>,
|
|
||||||
relation_name: &str,
|
|
||||||
) -> Result<Option<crate::database::relation::Relation>, String> {
|
|
||||||
let relative_type_name = match relative.get("type").and_then(|v| v.as_str()) {
|
|
||||||
Some(t) => t,
|
|
||||||
None => return Ok(None),
|
|
||||||
};
|
|
||||||
|
|
||||||
let relative_type = match self.db.types.get(relative_type_name) {
|
|
||||||
Some(t) => t,
|
|
||||||
None => return Ok(None),
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut relative_relations: Vec<crate::database::relation::Relation> = Vec::new();
|
|
||||||
|
|
||||||
for r in self.db.relations.values() {
|
|
||||||
if r.source_type != "entity" && r.destination_type != "entity" {
|
|
||||||
let condition1 = relative_type.hierarchy.contains(&r.source_type)
|
|
||||||
&& entity_type.hierarchy.contains(&r.destination_type);
|
|
||||||
let condition2 = entity_type.hierarchy.contains(&r.source_type)
|
|
||||||
&& relative_type.hierarchy.contains(&r.destination_type);
|
|
||||||
|
|
||||||
if condition1 || condition2 {
|
|
||||||
relative_relations.push(r.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut relative_relation =
|
|
||||||
self.reduce_entity_relations(relative_relations, relative, relation_name)?;
|
|
||||||
|
|
||||||
if relative_relation.is_none() {
|
|
||||||
let mut poly_relations: Vec<crate::database::relation::Relation> = Vec::new();
|
|
||||||
for r in self.db.relations.values() {
|
|
||||||
if r.destination_type == "entity" {
|
|
||||||
let condition1 = relative_type.hierarchy.contains(&r.source_type);
|
|
||||||
let condition2 = entity_type.hierarchy.contains(&r.source_type);
|
|
||||||
|
|
||||||
if condition1 || condition2 {
|
|
||||||
poly_relations.push(r.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
relative_relation = self.reduce_entity_relations(poly_relations, relative, relation_name)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(relative_relation)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn apply_entity_relation(
|
fn apply_entity_relation(
|
||||||
source_entity: &mut serde_json::Map<String, Value>,
|
source_entity: &mut serde_json::Map<String, Value>,
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@ -21,7 +21,6 @@ impl Queryer {
|
|||||||
pub fn query(
|
pub fn query(
|
||||||
&self,
|
&self,
|
||||||
schema_id: &str,
|
schema_id: &str,
|
||||||
stem_opt: Option<&str>,
|
|
||||||
filters: Option<&serde_json::Value>,
|
filters: Option<&serde_json::Value>,
|
||||||
) -> crate::drop::Drop {
|
) -> crate::drop::Drop {
|
||||||
let filters_map = filters.and_then(|f| f.as_object());
|
let filters_map = filters.and_then(|f| f.as_object());
|
||||||
@ -32,19 +31,21 @@ impl Queryer {
|
|||||||
Err(msg) => {
|
Err(msg) => {
|
||||||
return crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
return crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
||||||
code: "FILTER_PARSE_FAILED".to_string(),
|
code: "FILTER_PARSE_FAILED".to_string(),
|
||||||
message: msg,
|
message: msg.clone(),
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: schema_id.to_string(),
|
path: "".to_string(), // filters apply to the root query
|
||||||
|
cause: Some(msg),
|
||||||
|
context: filters.cloned(),
|
||||||
|
schema: Some(schema_id.to_string()),
|
||||||
},
|
},
|
||||||
}]);
|
}]);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let stem_key = stem_opt.unwrap_or("/");
|
let cache_key = format!("{}:{}", schema_id, filter_keys.join(","));
|
||||||
let cache_key = format!("{}(Stem:{}):{}", schema_id, stem_key, filter_keys.join(","));
|
|
||||||
|
|
||||||
// 2. Fetch from cache or compile
|
// 2. Fetch from cache or compile
|
||||||
let sql = match self.get_or_compile_sql(&cache_key, schema_id, stem_opt, &filter_keys) {
|
let sql = match self.get_or_compile_sql(&cache_key, schema_id, &filter_keys) {
|
||||||
Ok(sql) => sql,
|
Ok(sql) => sql,
|
||||||
Err(drop) => return drop,
|
Err(drop) => return drop,
|
||||||
};
|
};
|
||||||
@ -87,15 +88,19 @@ impl Queryer {
|
|||||||
&self,
|
&self,
|
||||||
cache_key: &str,
|
cache_key: &str,
|
||||||
schema_id: &str,
|
schema_id: &str,
|
||||||
stem_opt: Option<&str>,
|
|
||||||
filter_keys: &[String],
|
filter_keys: &[String],
|
||||||
) -> Result<String, crate::drop::Drop> {
|
) -> Result<String, crate::drop::Drop> {
|
||||||
if let Some(cached_sql) = self.cache.get(cache_key) {
|
if let Some(cached_sql) = self.cache.get(cache_key) {
|
||||||
return Ok(cached_sql.value().clone());
|
return Ok(cached_sql.value().clone());
|
||||||
}
|
}
|
||||||
|
|
||||||
let compiler = compiler::SqlCompiler::new(self.db.clone());
|
let compiler = compiler::Compiler {
|
||||||
match compiler.compile(schema_id, stem_opt, filter_keys) {
|
db: &self.db,
|
||||||
|
filter_keys: filter_keys,
|
||||||
|
alias_counter: 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
match compiler.compile(schema_id, filter_keys) {
|
||||||
Ok(compiled_sql) => {
|
Ok(compiled_sql) => {
|
||||||
self
|
self
|
||||||
.cache
|
.cache
|
||||||
@ -104,9 +109,12 @@ impl Queryer {
|
|||||||
}
|
}
|
||||||
Err(e) => Err(crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
Err(e) => Err(crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
||||||
code: "QUERY_COMPILATION_FAILED".to_string(),
|
code: "QUERY_COMPILATION_FAILED".to_string(),
|
||||||
message: e,
|
message: e.clone(),
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: schema_id.to_string(),
|
path: "".to_string(),
|
||||||
|
cause: Some(e),
|
||||||
|
context: None,
|
||||||
|
schema: Some(schema_id.to_string()),
|
||||||
},
|
},
|
||||||
}])),
|
}])),
|
||||||
}
|
}
|
||||||
@ -130,14 +138,20 @@ impl Queryer {
|
|||||||
code: "QUERY_FAILED".to_string(),
|
code: "QUERY_FAILED".to_string(),
|
||||||
message: format!("Expected array from generic query, got: {:?}", other),
|
message: format!("Expected array from generic query, got: {:?}", other),
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: schema_id.to_string(),
|
path: "".to_string(),
|
||||||
|
cause: Some(format!("Expected array, got {}", other)),
|
||||||
|
context: Some(serde_json::json!([sql])),
|
||||||
|
schema: Some(schema_id.to_string()),
|
||||||
},
|
},
|
||||||
}]),
|
}]),
|
||||||
Err(e) => crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
Err(e) => crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
||||||
code: "QUERY_FAILED".to_string(),
|
code: "QUERY_FAILED".to_string(),
|
||||||
message: format!("SPI error in queryer: {}", e),
|
message: format!("SPI error in queryer: {}", e),
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: schema_id.to_string(),
|
path: "".to_string(),
|
||||||
|
cause: Some(format!("SPI error in queryer: {}", e)),
|
||||||
|
context: Some(serde_json::json!([sql])),
|
||||||
|
schema: Some(schema_id.to_string()),
|
||||||
},
|
},
|
||||||
}]),
|
}]),
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1451,30 +1451,6 @@ fn test_queryer_0_6() {
|
|||||||
crate::tests::runner::run_test_case(&path, 0, 6).unwrap();
|
crate::tests::runner::run_test_case(&path, 0, 6).unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_queryer_0_7() {
|
|
||||||
let path = format!("{}/fixtures/queryer.json", env!("CARGO_MANIFEST_DIR"));
|
|
||||||
crate::tests::runner::run_test_case(&path, 0, 7).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_queryer_0_8() {
|
|
||||||
let path = format!("{}/fixtures/queryer.json", env!("CARGO_MANIFEST_DIR"));
|
|
||||||
crate::tests::runner::run_test_case(&path, 0, 8).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_queryer_0_9() {
|
|
||||||
let path = format!("{}/fixtures/queryer.json", env!("CARGO_MANIFEST_DIR"));
|
|
||||||
crate::tests::runner::run_test_case(&path, 0, 9).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_queryer_0_10() {
|
|
||||||
let path = format!("{}/fixtures/queryer.json", env!("CARGO_MANIFEST_DIR"));
|
|
||||||
crate::tests::runner::run_test_case(&path, 0, 10).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_not_0_0() {
|
fn test_not_0_0() {
|
||||||
let path = format!("{}/fixtures/not.json", env!("CARGO_MANIFEST_DIR"));
|
let path = format!("{}/fixtures/not.json", env!("CARGO_MANIFEST_DIR"));
|
||||||
@ -3449,12 +3425,6 @@ fn test_if_then_else_13_1() {
|
|||||||
crate::tests::runner::run_test_case(&path, 13, 1).unwrap();
|
crate::tests::runner::run_test_case(&path, 13, 1).unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_stems_0_0() {
|
|
||||||
let path = format!("{}/fixtures/stems.json", env!("CARGO_MANIFEST_DIR"));
|
|
||||||
crate::tests::runner::run_test_case(&path, 0, 0).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_empty_string_0_0() {
|
fn test_empty_string_0_0() {
|
||||||
let path = format!("{}/fixtures/emptyString.json", env!("CARGO_MANIFEST_DIR"));
|
let path = format!("{}/fixtures/emptyString.json", env!("CARGO_MANIFEST_DIR"));
|
||||||
|
|||||||
@ -2,7 +2,6 @@ use crate::*;
|
|||||||
pub mod runner;
|
pub mod runner;
|
||||||
pub mod types;
|
pub mod types;
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
pub mod sql_validator;
|
|
||||||
|
|
||||||
// Database module tests moved to src/database/executors/mock.rs
|
// Database module tests moved to src/database/executors/mock.rs
|
||||||
|
|
||||||
|
|||||||
@ -1,19 +1,10 @@
|
|||||||
|
use crate::tests::types::Suite;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
|
use serde_json::Value;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::fs;
|
use std::fs;
|
||||||
use std::sync::{Arc, OnceLock, RwLock};
|
use std::sync::{Arc, OnceLock, RwLock};
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
pub struct TestSuite {
|
|
||||||
#[allow(dead_code)]
|
|
||||||
pub description: String,
|
|
||||||
pub database: serde_json::Value,
|
|
||||||
pub tests: Vec<TestCase>,
|
|
||||||
}
|
|
||||||
|
|
||||||
use crate::tests::types::TestCase;
|
|
||||||
use serde_json::Value;
|
|
||||||
|
|
||||||
pub fn deserialize_some<'de, D>(deserializer: D) -> Result<Option<Value>, D::Error>
|
pub fn deserialize_some<'de, D>(deserializer: D) -> Result<Option<Value>, D::Error>
|
||||||
where
|
where
|
||||||
D: serde::Deserializer<'de>,
|
D: serde::Deserializer<'de>,
|
||||||
@ -23,7 +14,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Type alias for easier reading
|
// Type alias for easier reading
|
||||||
type CompiledSuite = Arc<Vec<(TestSuite, Arc<crate::database::Database>)>>;
|
type CompiledSuite = Arc<Vec<(Suite, Arc<crate::database::Database>)>>;
|
||||||
|
|
||||||
// Global cache mapping filename -> Vector of (Parsed JSON suite, Compiled Database)
|
// Global cache mapping filename -> Vector of (Parsed JSON suite, Compiled Database)
|
||||||
static CACHE: OnceLock<RwLock<HashMap<String, CompiledSuite>>> = OnceLock::new();
|
static CACHE: OnceLock<RwLock<HashMap<String, CompiledSuite>>> = OnceLock::new();
|
||||||
@ -46,7 +37,7 @@ fn get_cached_file(path: &str) -> CompiledSuite {
|
|||||||
} else {
|
} else {
|
||||||
let content =
|
let content =
|
||||||
fs::read_to_string(path).unwrap_or_else(|_| panic!("Failed to read file: {}", path));
|
fs::read_to_string(path).unwrap_or_else(|_| panic!("Failed to read file: {}", path));
|
||||||
let suites: Vec<TestSuite> = serde_json::from_str(&content)
|
let suites: Vec<Suite> = serde_json::from_str(&content)
|
||||||
.unwrap_or_else(|e| panic!("Failed to parse JSON in {}: {}", path, e));
|
.unwrap_or_else(|e| panic!("Failed to parse JSON in {}: {}", path, e));
|
||||||
|
|
||||||
let mut compiled_suites = Vec::new();
|
let mut compiled_suites = Vec::new();
|
||||||
|
|||||||
@ -1,156 +0,0 @@
|
|||||||
use sqlparser::ast::{
|
|
||||||
Expr, Join, JoinConstraint, JoinOperator, Query, Select, SelectItem, SetExpr, Statement,
|
|
||||||
TableFactor, TableWithJoins, Ident,
|
|
||||||
};
|
|
||||||
use sqlparser::dialect::PostgreSqlDialect;
|
|
||||||
use sqlparser::parser::Parser;
|
|
||||||
use std::collections::HashSet;
|
|
||||||
|
|
||||||
pub fn validate_semantic_sql(sql: &str) -> Result<(), String> {
|
|
||||||
let dialect = PostgreSqlDialect {};
|
|
||||||
let statements = match Parser::parse_sql(&dialect, sql) {
|
|
||||||
Ok(s) => s,
|
|
||||||
Err(e) => return Err(format!("SQL Syntax Error: {}\nSQL: {}", e, sql)),
|
|
||||||
};
|
|
||||||
|
|
||||||
for statement in statements {
|
|
||||||
validate_statement(&statement, sql)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn validate_statement(stmt: &Statement, original_sql: &str) -> Result<(), String> {
|
|
||||||
match stmt {
|
|
||||||
Statement::Query(query) => validate_query(query, original_sql)?,
|
|
||||||
Statement::Insert(insert) => {
|
|
||||||
if let Some(query) = &insert.source {
|
|
||||||
validate_query(query, original_sql)?
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Statement::Update(update) => {
|
|
||||||
if let Some(expr) = &update.selection {
|
|
||||||
validate_expr(expr, &HashSet::new(), original_sql)?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Statement::Delete(delete) => {
|
|
||||||
if let Some(expr) = &delete.selection {
|
|
||||||
validate_expr(expr, &HashSet::new(), original_sql)?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn validate_query(query: &Query, original_sql: &str) -> Result<(), String> {
|
|
||||||
if let SetExpr::Select(select) = &*query.body {
|
|
||||||
validate_select(select, original_sql)?;
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn validate_select(select: &Select, original_sql: &str) -> Result<(), String> {
|
|
||||||
let mut available_aliases = HashSet::new();
|
|
||||||
|
|
||||||
// 1. Collect all declared table aliases in the FROM clause and JOINs
|
|
||||||
for table_with_joins in &select.from {
|
|
||||||
collect_aliases_from_table_factor(&table_with_joins.relation, &mut available_aliases);
|
|
||||||
for join in &table_with_joins.joins {
|
|
||||||
collect_aliases_from_table_factor(&join.relation, &mut available_aliases);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 2. Validate all SELECT projection fields
|
|
||||||
for projection in &select.projection {
|
|
||||||
if let SelectItem::UnnamedExpr(expr) | SelectItem::ExprWithAlias { expr, .. } = projection {
|
|
||||||
validate_expr(expr, &available_aliases, original_sql)?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 3. Validate ON conditions in joins
|
|
||||||
for table_with_joins in &select.from {
|
|
||||||
for join in &table_with_joins.joins {
|
|
||||||
if let JoinOperator::Inner(JoinConstraint::On(expr))
|
|
||||||
| JoinOperator::LeftOuter(JoinConstraint::On(expr))
|
|
||||||
| JoinOperator::RightOuter(JoinConstraint::On(expr))
|
|
||||||
| JoinOperator::FullOuter(JoinConstraint::On(expr))
|
|
||||||
| JoinOperator::Join(JoinConstraint::On(expr)) = &join.join_operator
|
|
||||||
{
|
|
||||||
validate_expr(expr, &available_aliases, original_sql)?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 4. Validate WHERE conditions
|
|
||||||
if let Some(selection) = &select.selection {
|
|
||||||
validate_expr(selection, &available_aliases, original_sql)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn collect_aliases_from_table_factor(tf: &TableFactor, aliases: &mut HashSet<String>) {
|
|
||||||
match tf {
|
|
||||||
TableFactor::Table { name, alias, .. } => {
|
|
||||||
if let Some(table_alias) = alias {
|
|
||||||
aliases.insert(table_alias.name.value.clone());
|
|
||||||
} else if let Some(last) = name.0.last() {
|
|
||||||
match last {
|
|
||||||
sqlparser::ast::ObjectNamePart::Identifier(i) => {
|
|
||||||
aliases.insert(i.value.clone());
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
TableFactor::Derived { alias: Some(table_alias), .. } => {
|
|
||||||
aliases.insert(table_alias.name.value.clone());
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn validate_expr(expr: &Expr, available_aliases: &HashSet<String>, sql: &str) -> Result<(), String> {
|
|
||||||
match expr {
|
|
||||||
Expr::CompoundIdentifier(idents) => {
|
|
||||||
if idents.len() == 2 {
|
|
||||||
let alias = &idents[0].value;
|
|
||||||
if !available_aliases.is_empty() && !available_aliases.contains(alias) {
|
|
||||||
return Err(format!(
|
|
||||||
"Semantic Error: Orchestrated query referenced table alias '{}' but it was not declared in the query's FROM/JOIN clauses.\nAvailable aliases: {:?}\nSQL: {}",
|
|
||||||
alias, available_aliases, sql
|
|
||||||
));
|
|
||||||
}
|
|
||||||
} else if idents.len() > 2 {
|
|
||||||
let alias = &idents[1].value; // In form schema.table.column, 'table' is idents[1]
|
|
||||||
if !available_aliases.is_empty() && !available_aliases.contains(alias) {
|
|
||||||
return Err(format!(
|
|
||||||
"Semantic Error: Orchestrated query referenced table '{}' but it was not mapped.\nAvailable aliases: {:?}\nSQL: {}",
|
|
||||||
alias, available_aliases, sql
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Expr::BinaryOp { left, right, .. } => {
|
|
||||||
validate_expr(left, available_aliases, sql)?;
|
|
||||||
validate_expr(right, available_aliases, sql)?;
|
|
||||||
}
|
|
||||||
Expr::IsFalse(e) | Expr::IsNotFalse(e) | Expr::IsTrue(e) | Expr::IsNotTrue(e)
|
|
||||||
| Expr::IsNull(e) | Expr::IsNotNull(e) | Expr::InList { expr: e, .. }
|
|
||||||
| Expr::Nested(e) | Expr::UnaryOp { expr: e, .. } | Expr::Cast { expr: e, .. }
|
|
||||||
| Expr::Like { expr: e, .. } | Expr::ILike { expr: e, .. } | Expr::AnyOp { left: e, .. }
|
|
||||||
| Expr::AllOp { left: e, .. } => {
|
|
||||||
validate_expr(e, available_aliases, sql)?;
|
|
||||||
}
|
|
||||||
Expr::Function(func) => {
|
|
||||||
if let sqlparser::ast::FunctionArguments::List(args) = &func.args {
|
|
||||||
if let Some(sqlparser::ast::FunctionArg::Unnamed(sqlparser::ast::FunctionArgExpr::Expr(e))) = args.args.get(0) {
|
|
||||||
validate_expr(e, available_aliases, sql)?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
@ -1,11 +1,11 @@
|
|||||||
use super::expect::ExpectBlock;
|
use super::expect::Expect;
|
||||||
use crate::database::Database;
|
use crate::database::Database;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
pub struct TestCase {
|
pub struct Case {
|
||||||
pub description: String,
|
pub description: String,
|
||||||
|
|
||||||
#[serde(default = "default_action")]
|
#[serde(default = "default_action")]
|
||||||
@ -16,9 +16,6 @@ pub struct TestCase {
|
|||||||
pub schema_id: String,
|
pub schema_id: String,
|
||||||
|
|
||||||
// For Query
|
// For Query
|
||||||
#[serde(default)]
|
|
||||||
pub stem: Option<String>,
|
|
||||||
|
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub filters: Option<serde_json::Value>,
|
pub filters: Option<serde_json::Value>,
|
||||||
|
|
||||||
@ -30,15 +27,15 @@ pub struct TestCase {
|
|||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub mocks: Option<serde_json::Value>,
|
pub mocks: Option<serde_json::Value>,
|
||||||
|
|
||||||
pub expect: Option<ExpectBlock>,
|
pub expect: Option<Expect>,
|
||||||
}
|
}
|
||||||
|
|
||||||
fn default_action() -> String {
|
fn default_action() -> String {
|
||||||
"validate".to_string()
|
"validate".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
impl TestCase {
|
impl Case {
|
||||||
pub fn run_compile(&self, db: Arc<Database>) -> Result<(), String> {
|
pub fn run_compile(&self, _db: Arc<Database>) -> Result<(), String> {
|
||||||
let expected_success = self.expect.as_ref().map(|e| e.success).unwrap_or(false);
|
let expected_success = self.expect.as_ref().map(|e| e.success).unwrap_or(false);
|
||||||
|
|
||||||
// We assume db has already been setup and compiled successfully by runner.rs's `jspg_setup`
|
// We assume db has already been setup and compiled successfully by runner.rs's `jspg_setup`
|
||||||
@ -52,24 +49,6 @@ impl TestCase {
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Assert stems
|
|
||||||
if let Some(expect) = &self.expect {
|
|
||||||
if let Some(expected_stems) = &expect.stems {
|
|
||||||
// Convert the Db stems (HashMap<String, HashMap<String, Arc<Stem>>>) to matching JSON shape
|
|
||||||
let db_stems_json = serde_json::to_value(&db.stems).unwrap();
|
|
||||||
let expect_stems_json = serde_json::to_value(expected_stems).unwrap();
|
|
||||||
|
|
||||||
if db_stems_json != expect_stems_json {
|
|
||||||
let expected_pretty = serde_json::to_string_pretty(&expect_stems_json).unwrap();
|
|
||||||
let got_pretty = serde_json::to_string_pretty(&db_stems_json).unwrap();
|
|
||||||
return Err(format!(
|
|
||||||
"Stem validation failed.\nExpected:\n{}\n\nGot:\n{}",
|
|
||||||
expected_pretty, got_pretty
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -138,6 +117,7 @@ impl TestCase {
|
|||||||
))
|
))
|
||||||
} else if let Some(expect) = &self.expect {
|
} else if let Some(expect) = &self.expect {
|
||||||
let queries = db.executor.get_queries();
|
let queries = db.executor.get_queries();
|
||||||
|
expect.assert_pattern(&queries)?;
|
||||||
expect.assert_sql(&queries)
|
expect.assert_sql(&queries)
|
||||||
} else {
|
} else {
|
||||||
Ok(())
|
Ok(())
|
||||||
@ -157,8 +137,7 @@ impl TestCase {
|
|||||||
use crate::queryer::Queryer;
|
use crate::queryer::Queryer;
|
||||||
let queryer = Queryer::new(db.clone());
|
let queryer = Queryer::new(db.clone());
|
||||||
|
|
||||||
let stem_opt = self.stem.as_deref();
|
let result = queryer.query(&self.schema_id, self.filters.as_ref());
|
||||||
let result = queryer.query(&self.schema_id, stem_opt, self.filters.as_ref());
|
|
||||||
|
|
||||||
let expected_success = self.expect.as_ref().map(|e| e.success).unwrap_or(false);
|
let expected_success = self.expect.as_ref().map(|e| e.success).unwrap_or(false);
|
||||||
let got_success = result.errors.is_empty();
|
let got_success = result.errors.is_empty();
|
||||||
@ -176,6 +155,7 @@ impl TestCase {
|
|||||||
))
|
))
|
||||||
} else if let Some(expect) = &self.expect {
|
} else if let Some(expect) = &self.expect {
|
||||||
let queries = db.executor.get_queries();
|
let queries = db.executor.get_queries();
|
||||||
|
expect.assert_pattern(&queries)?;
|
||||||
expect.assert_sql(&queries)
|
expect.assert_sql(&queries)
|
||||||
} else {
|
} else {
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
20
src/tests/types/expect/mod.rs
Normal file
20
src/tests/types/expect/mod.rs
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
pub mod pattern;
|
||||||
|
pub mod sql;
|
||||||
|
|
||||||
|
use serde::Deserialize;
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
#[serde(untagged)]
|
||||||
|
pub enum SqlExpectation {
|
||||||
|
Single(String),
|
||||||
|
Multi(Vec<String>),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
pub struct Expect {
|
||||||
|
pub success: bool,
|
||||||
|
pub result: Option<serde_json::Value>,
|
||||||
|
pub errors: Option<Vec<serde_json::Value>>,
|
||||||
|
#[serde(default)]
|
||||||
|
pub sql: Option<Vec<SqlExpectation>>,
|
||||||
|
}
|
||||||
@ -1,30 +1,13 @@
|
|||||||
|
use super::Expect;
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
use serde::Deserialize;
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
impl Expect {
|
||||||
#[serde(untagged)]
|
|
||||||
pub enum SqlExpectation {
|
|
||||||
Single(String),
|
|
||||||
Multi(Vec<String>),
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
pub struct ExpectBlock {
|
|
||||||
pub success: bool,
|
|
||||||
pub result: Option<serde_json::Value>,
|
|
||||||
pub errors: Option<Vec<serde_json::Value>>,
|
|
||||||
pub stems: Option<HashMap<String, HashMap<String, serde_json::Value>>>,
|
|
||||||
#[serde(default)]
|
|
||||||
pub sql: Option<Vec<SqlExpectation>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ExpectBlock {
|
|
||||||
/// Advanced SQL execution assertion algorithm ported from `assert.go`.
|
/// Advanced SQL execution assertion algorithm ported from `assert.go`.
|
||||||
/// This compares two arrays of strings, one containing {{uuid:name}} or {{timestamp}} placeholders,
|
/// This compares two arrays of strings, one containing {{uuid:name}} or {{timestamp}} placeholders,
|
||||||
/// and the other containing actual executed database queries. It ensures that placeholder UUIDs
|
/// and the other containing actual executed database queries. It ensures that placeholder UUIDs
|
||||||
/// are consistently mapped to the same actual UUIDs across all lines, and strictly validates line-by-line sequences.
|
/// are consistently mapped to the same actual UUIDs across all lines, and strictly validates line-by-line sequences.
|
||||||
pub fn assert_sql(&self, actual: &[String]) -> Result<(), String> {
|
pub fn assert_pattern(&self, actual: &[String]) -> Result<(), String> {
|
||||||
let patterns = match &self.sql {
|
let patterns = match &self.sql {
|
||||||
Some(s) => s,
|
Some(s) => s,
|
||||||
None => return Ok(()),
|
None => return Ok(()),
|
||||||
@ -39,12 +22,6 @@ impl ExpectBlock {
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
for query in actual {
|
|
||||||
if let Err(e) = crate::tests::sql_validator::validate_semantic_sql(query) {
|
|
||||||
return Err(e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let ws_re = Regex::new(r"\s+").unwrap();
|
let ws_re = Regex::new(r"\s+").unwrap();
|
||||||
|
|
||||||
let types = HashMap::from([
|
let types = HashMap::from([
|
||||||
@ -82,8 +59,8 @@ impl ExpectBlock {
|
|||||||
let aline = clean_str(aline_raw);
|
let aline = clean_str(aline_raw);
|
||||||
|
|
||||||
let pattern_str_raw = match pattern_expect {
|
let pattern_str_raw = match pattern_expect {
|
||||||
SqlExpectation::Single(s) => s.clone(),
|
super::SqlExpectation::Single(s) => s.clone(),
|
||||||
SqlExpectation::Multi(m) => m.join(" "),
|
super::SqlExpectation::Multi(m) => m.join(" "),
|
||||||
};
|
};
|
||||||
|
|
||||||
let pattern_str = clean_str(&pattern_str_raw);
|
let pattern_str = clean_str(&pattern_str_raw);
|
||||||
206
src/tests/types/expect/sql.rs
Normal file
206
src/tests/types/expect/sql.rs
Normal file
@ -0,0 +1,206 @@
|
|||||||
|
use super::Expect;
|
||||||
|
use sqlparser::ast::{Expr, Query, SelectItem, Statement, TableFactor};
|
||||||
|
use sqlparser::dialect::PostgreSqlDialect;
|
||||||
|
use sqlparser::parser::Parser;
|
||||||
|
use std::collections::HashSet;
|
||||||
|
|
||||||
|
impl Expect {
|
||||||
|
pub fn assert_sql(&self, actual: &[String]) -> Result<(), String> {
|
||||||
|
for query in actual {
|
||||||
|
if let Err(e) = Self::validate_semantic_sql(query) {
|
||||||
|
return Err(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn validate_semantic_sql(sql: &str) -> Result<(), String> {
|
||||||
|
let dialect = PostgreSqlDialect {};
|
||||||
|
let statements = match Parser::parse_sql(&dialect, sql) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => return Err(format!("SQL Syntax Error: {}\nSQL: {}", e, sql)),
|
||||||
|
};
|
||||||
|
|
||||||
|
for statement in statements {
|
||||||
|
Self::validate_statement(&statement, sql)?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn validate_statement(stmt: &Statement, original_sql: &str) -> Result<(), String> {
|
||||||
|
match stmt {
|
||||||
|
Statement::Query(query) => Self::validate_query(query, &HashSet::new(), original_sql)?,
|
||||||
|
Statement::Insert(insert) => {
|
||||||
|
if let Some(query) = &insert.source {
|
||||||
|
Self::validate_query(query, &HashSet::new(), original_sql)?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Statement::Update(update) => {
|
||||||
|
if let Some(expr) = &update.selection {
|
||||||
|
Self::validate_expr(expr, &HashSet::new(), original_sql)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Statement::Delete(delete) => {
|
||||||
|
if let Some(expr) = &delete.selection {
|
||||||
|
Self::validate_expr(expr, &HashSet::new(), original_sql)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn validate_query(
|
||||||
|
query: &Query,
|
||||||
|
available_aliases: &HashSet<String>,
|
||||||
|
original_sql: &str,
|
||||||
|
) -> Result<(), String> {
|
||||||
|
if let sqlparser::ast::SetExpr::Select(select) = &*query.body {
|
||||||
|
Self::validate_select(&select, available_aliases, original_sql)?;
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn validate_select(
|
||||||
|
select: &sqlparser::ast::Select,
|
||||||
|
parent_aliases: &HashSet<String>,
|
||||||
|
original_sql: &str,
|
||||||
|
) -> Result<(), String> {
|
||||||
|
let mut available_aliases = parent_aliases.clone();
|
||||||
|
|
||||||
|
// 1. Collect all declared table aliases in the FROM clause and JOINs
|
||||||
|
for table_with_joins in &select.from {
|
||||||
|
Self::collect_aliases_from_table_factor(&table_with_joins.relation, &mut available_aliases);
|
||||||
|
for join in &table_with_joins.joins {
|
||||||
|
Self::collect_aliases_from_table_factor(&join.relation, &mut available_aliases);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Validate all SELECT projection fields
|
||||||
|
for projection in &select.projection {
|
||||||
|
if let SelectItem::UnnamedExpr(expr) | SelectItem::ExprWithAlias { expr, .. } = projection {
|
||||||
|
Self::validate_expr(expr, &available_aliases, original_sql)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. Validate ON conditions in joins
|
||||||
|
for table_with_joins in &select.from {
|
||||||
|
for join in &table_with_joins.joins {
|
||||||
|
if let sqlparser::ast::JoinOperator::Inner(sqlparser::ast::JoinConstraint::On(expr))
|
||||||
|
| sqlparser::ast::JoinOperator::LeftOuter(sqlparser::ast::JoinConstraint::On(expr))
|
||||||
|
| sqlparser::ast::JoinOperator::RightOuter(sqlparser::ast::JoinConstraint::On(expr))
|
||||||
|
| sqlparser::ast::JoinOperator::FullOuter(sqlparser::ast::JoinConstraint::On(expr))
|
||||||
|
| sqlparser::ast::JoinOperator::Join(sqlparser::ast::JoinConstraint::On(expr)) =
|
||||||
|
&join.join_operator
|
||||||
|
{
|
||||||
|
Self::validate_expr(expr, &available_aliases, original_sql)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. Validate WHERE conditions
|
||||||
|
if let Some(selection) = &select.selection {
|
||||||
|
Self::validate_expr(selection, &available_aliases, original_sql)?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn collect_aliases_from_table_factor(tf: &TableFactor, aliases: &mut HashSet<String>) {
|
||||||
|
match tf {
|
||||||
|
TableFactor::Table { name, alias, .. } => {
|
||||||
|
if let Some(table_alias) = alias {
|
||||||
|
aliases.insert(table_alias.name.value.clone());
|
||||||
|
} else if let Some(last) = name.0.last() {
|
||||||
|
match last {
|
||||||
|
sqlparser::ast::ObjectNamePart::Identifier(i) => {
|
||||||
|
aliases.insert(i.value.clone());
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
TableFactor::Derived {
|
||||||
|
subquery,
|
||||||
|
alias: Some(table_alias),
|
||||||
|
..
|
||||||
|
} => {
|
||||||
|
aliases.insert(table_alias.name.value.clone());
|
||||||
|
// A derived table is technically a nested scope which is opaque outside, but for pure semantic checks
|
||||||
|
// its internal contents should be validated purely within its own scope (not leaking external aliases in, usually)
|
||||||
|
// but Postgres allows lateral correlation. We will validate its interior with an empty scope.
|
||||||
|
let _ = Self::validate_query(subquery, &HashSet::new(), "");
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn validate_expr(
|
||||||
|
expr: &Expr,
|
||||||
|
available_aliases: &HashSet<String>,
|
||||||
|
sql: &str,
|
||||||
|
) -> Result<(), String> {
|
||||||
|
match expr {
|
||||||
|
Expr::CompoundIdentifier(idents) => {
|
||||||
|
if idents.len() == 2 {
|
||||||
|
let alias = &idents[0].value;
|
||||||
|
if !available_aliases.is_empty() && !available_aliases.contains(alias) {
|
||||||
|
return Err(format!(
|
||||||
|
"Semantic Error: Orchestrated query referenced table alias '{}' but it was not declared in the query's FROM/JOIN clauses.\nAvailable aliases: {:?}\nSQL: {}",
|
||||||
|
alias, available_aliases, sql
|
||||||
|
));
|
||||||
|
}
|
||||||
|
} else if idents.len() > 2 {
|
||||||
|
let alias = &idents[1].value; // In form schema.table.column, 'table' is idents[1]
|
||||||
|
if !available_aliases.is_empty() && !available_aliases.contains(alias) {
|
||||||
|
return Err(format!(
|
||||||
|
"Semantic Error: Orchestrated query referenced table '{}' but it was not mapped.\nAvailable aliases: {:?}\nSQL: {}",
|
||||||
|
alias, available_aliases, sql
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Expr::Subquery(subquery) => Self::validate_query(subquery, available_aliases, sql)?,
|
||||||
|
Expr::Exists { subquery, .. } => Self::validate_query(subquery, available_aliases, sql)?,
|
||||||
|
Expr::InSubquery {
|
||||||
|
expr: e, subquery, ..
|
||||||
|
} => {
|
||||||
|
Self::validate_expr(e, available_aliases, sql)?;
|
||||||
|
Self::validate_query(subquery, available_aliases, sql)?;
|
||||||
|
}
|
||||||
|
Expr::BinaryOp { left, right, .. } => {
|
||||||
|
Self::validate_expr(left, available_aliases, sql)?;
|
||||||
|
Self::validate_expr(right, available_aliases, sql)?;
|
||||||
|
}
|
||||||
|
Expr::IsFalse(e)
|
||||||
|
| Expr::IsNotFalse(e)
|
||||||
|
| Expr::IsTrue(e)
|
||||||
|
| Expr::IsNotTrue(e)
|
||||||
|
| Expr::IsNull(e)
|
||||||
|
| Expr::IsNotNull(e)
|
||||||
|
| Expr::InList { expr: e, .. }
|
||||||
|
| Expr::Nested(e)
|
||||||
|
| Expr::UnaryOp { expr: e, .. }
|
||||||
|
| Expr::Cast { expr: e, .. }
|
||||||
|
| Expr::Like { expr: e, .. }
|
||||||
|
| Expr::ILike { expr: e, .. }
|
||||||
|
| Expr::AnyOp { left: e, .. }
|
||||||
|
| Expr::AllOp { left: e, .. } => {
|
||||||
|
Self::validate_expr(e, available_aliases, sql)?;
|
||||||
|
}
|
||||||
|
Expr::Function(func) => {
|
||||||
|
if let sqlparser::ast::FunctionArguments::List(args) = &func.args {
|
||||||
|
if let Some(sqlparser::ast::FunctionArg::Unnamed(
|
||||||
|
sqlparser::ast::FunctionArgExpr::Expr(e),
|
||||||
|
)) = args.args.get(0)
|
||||||
|
{
|
||||||
|
Self::validate_expr(e, available_aliases, sql)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -2,6 +2,6 @@ pub mod case;
|
|||||||
pub mod expect;
|
pub mod expect;
|
||||||
pub mod suite;
|
pub mod suite;
|
||||||
|
|
||||||
pub use case::TestCase;
|
pub use case::Case;
|
||||||
pub use expect::ExpectBlock;
|
pub use expect::Expect;
|
||||||
pub use suite::TestSuite;
|
pub use suite::Suite;
|
||||||
|
|||||||
@ -1,10 +1,10 @@
|
|||||||
use super::case::TestCase;
|
use super::case::Case;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
pub struct TestSuite {
|
pub struct Suite {
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
pub description: String,
|
pub description: String,
|
||||||
pub database: serde_json::Value,
|
pub database: serde_json::Value,
|
||||||
pub tests: Vec<TestCase>,
|
pub tests: Vec<Case>,
|
||||||
}
|
}
|
||||||
|
|||||||
@ -67,7 +67,12 @@ impl Validator {
|
|||||||
.map(|e| crate::drop::Error {
|
.map(|e| crate::drop::Error {
|
||||||
code: e.code,
|
code: e.code,
|
||||||
message: e.message,
|
message: e.message,
|
||||||
details: crate::drop::ErrorDetails { path: e.path },
|
details: crate::drop::ErrorDetails {
|
||||||
|
path: e.path,
|
||||||
|
cause: None,
|
||||||
|
context: None,
|
||||||
|
schema: None,
|
||||||
|
},
|
||||||
})
|
})
|
||||||
.collect();
|
.collect();
|
||||||
crate::drop::Drop::with_errors(errors)
|
crate::drop::Drop::with_errors(errors)
|
||||||
@ -76,7 +81,12 @@ impl Validator {
|
|||||||
Err(e) => crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
Err(e) => crate::drop::Drop::with_errors(vec![crate::drop::Error {
|
||||||
code: e.code,
|
code: e.code,
|
||||||
message: e.message,
|
message: e.message,
|
||||||
details: crate::drop::ErrorDetails { path: e.path },
|
details: crate::drop::ErrorDetails {
|
||||||
|
path: e.path,
|
||||||
|
cause: None,
|
||||||
|
context: None,
|
||||||
|
schema: None,
|
||||||
|
},
|
||||||
}]),
|
}]),
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
@ -84,7 +94,10 @@ impl Validator {
|
|||||||
code: "SCHEMA_NOT_FOUND".to_string(),
|
code: "SCHEMA_NOT_FOUND".to_string(),
|
||||||
message: format!("Schema {} not found", schema_id),
|
message: format!("Schema {} not found", schema_id),
|
||||||
details: crate::drop::ErrorDetails {
|
details: crate::drop::ErrorDetails {
|
||||||
path: "".to_string(),
|
path: "/".to_string(),
|
||||||
|
cause: None,
|
||||||
|
context: None,
|
||||||
|
schema: None,
|
||||||
},
|
},
|
||||||
}])
|
}])
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user