d803bfe2b1
git-subtree-dir: vendor/ruvector git-subtree-split: b64c21726f2bb37286d9ee36a7869fef60cc6900
328 lines
11 KiB
SQL
328 lines
11 KiB
SQL
-- Graph Operations Examples for ruvector-postgres
|
|
-- This file demonstrates the graph database capabilities
|
|
|
|
-- ============================================================================
|
|
-- Basic Graph Operations
|
|
-- ============================================================================
|
|
|
|
-- Create a new graph
|
|
SELECT ruvector_create_graph('social_network');
|
|
|
|
-- List all graphs
|
|
SELECT ruvector_list_graphs();
|
|
|
|
-- ============================================================================
|
|
-- Social Network Example
|
|
-- ============================================================================
|
|
|
|
-- Add users
|
|
SELECT ruvector_add_node(
|
|
'social_network',
|
|
ARRAY['Person'],
|
|
jsonb_build_object('name', 'Alice', 'age', 30, 'city', 'New York')
|
|
) AS alice_id;
|
|
|
|
SELECT ruvector_add_node(
|
|
'social_network',
|
|
ARRAY['Person'],
|
|
jsonb_build_object('name', 'Bob', 'age', 25, 'city', 'San Francisco')
|
|
) AS bob_id;
|
|
|
|
SELECT ruvector_add_node(
|
|
'social_network',
|
|
ARRAY['Person'],
|
|
jsonb_build_object('name', 'Charlie', 'age', 35, 'city', 'Boston')
|
|
) AS charlie_id;
|
|
|
|
SELECT ruvector_add_node(
|
|
'social_network',
|
|
ARRAY['Person'],
|
|
jsonb_build_object('name', 'Diana', 'age', 28, 'city', 'Seattle')
|
|
) AS diana_id;
|
|
|
|
-- Create friendships
|
|
SELECT ruvector_add_edge(
|
|
'social_network',
|
|
1, 2, -- Alice -> Bob
|
|
'FRIENDS',
|
|
jsonb_build_object('since', '2020-01-15', 'strength', 0.9)
|
|
);
|
|
|
|
SELECT ruvector_add_edge(
|
|
'social_network',
|
|
2, 3, -- Bob -> Charlie
|
|
'FRIENDS',
|
|
jsonb_build_object('since', '2019-06-20', 'strength', 0.8)
|
|
);
|
|
|
|
SELECT ruvector_add_edge(
|
|
'social_network',
|
|
1, 4, -- Alice -> Diana
|
|
'FRIENDS',
|
|
jsonb_build_object('since', '2021-03-10', 'strength', 0.7)
|
|
);
|
|
|
|
SELECT ruvector_add_edge(
|
|
'social_network',
|
|
3, 4, -- Charlie -> Diana
|
|
'FRIENDS',
|
|
jsonb_build_object('since', '2020-09-05', 'strength', 0.85)
|
|
);
|
|
|
|
-- Get graph statistics
|
|
SELECT ruvector_graph_stats('social_network');
|
|
|
|
-- Find nodes by label
|
|
SELECT ruvector_find_nodes_by_label('social_network', 'Person');
|
|
|
|
-- Get neighbors of Alice (node 1)
|
|
SELECT ruvector_get_neighbors('social_network', 1);
|
|
|
|
-- Find shortest path from Alice to Charlie
|
|
SELECT ruvector_shortest_path('social_network', 1, 3, 10);
|
|
|
|
-- Find weighted shortest path
|
|
SELECT ruvector_shortest_path_weighted('social_network', 1, 3, 'strength');
|
|
|
|
-- ============================================================================
|
|
-- Cypher Query Examples
|
|
-- ============================================================================
|
|
|
|
-- Create nodes with Cypher
|
|
SELECT ruvector_cypher(
|
|
'social_network',
|
|
'CREATE (n:Person {name: ''Eve'', age: 27, city: ''Austin''}) RETURN n',
|
|
NULL
|
|
);
|
|
|
|
-- Match all persons
|
|
SELECT ruvector_cypher(
|
|
'social_network',
|
|
'MATCH (n:Person) RETURN n.name, n.age',
|
|
NULL
|
|
);
|
|
|
|
-- Match with WHERE clause
|
|
SELECT ruvector_cypher(
|
|
'social_network',
|
|
'MATCH (n:Person) WHERE n.age > 28 RETURN n.name, n.age',
|
|
NULL
|
|
);
|
|
|
|
-- Parameterized query
|
|
SELECT ruvector_cypher(
|
|
'social_network',
|
|
'MATCH (n:Person) WHERE n.name = $name RETURN n',
|
|
jsonb_build_object('name', 'Alice')
|
|
);
|
|
|
|
-- Create relationship with Cypher
|
|
SELECT ruvector_cypher(
|
|
'social_network',
|
|
'CREATE (a:Person {name: ''Frank''})-[:KNOWS {since: 2022}]->(b:Person {name: ''Grace''}) RETURN a, b',
|
|
NULL
|
|
);
|
|
|
|
-- ============================================================================
|
|
-- Knowledge Graph Example
|
|
-- ============================================================================
|
|
|
|
SELECT ruvector_create_graph('knowledge');
|
|
|
|
-- Add concepts
|
|
SELECT ruvector_cypher(
|
|
'knowledge',
|
|
'CREATE (ml:Concept {name: ''Machine Learning'', category: ''AI''})
|
|
CREATE (nn:Concept {name: ''Neural Networks'', category: ''AI''})
|
|
CREATE (dl:Concept {name: ''Deep Learning'', category: ''AI''})
|
|
CREATE (cv:Concept {name: ''Computer Vision'', category: ''AI''})
|
|
CREATE (nlp:Concept {name: ''Natural Language Processing'', category: ''AI''})
|
|
RETURN ml, nn, dl, cv, nlp',
|
|
NULL
|
|
);
|
|
|
|
-- Create relationships between concepts
|
|
WITH ids AS (
|
|
SELECT generate_series(1, 5) AS id
|
|
)
|
|
SELECT
|
|
CASE
|
|
WHEN i.id = 1 THEN ruvector_add_edge('knowledge', 1, 2, 'INCLUDES', '{"strength": 0.9}'::jsonb)
|
|
WHEN i.id = 2 THEN ruvector_add_edge('knowledge', 2, 3, 'SPECIALIZES_IN', '{"strength": 0.95}'::jsonb)
|
|
WHEN i.id = 3 THEN ruvector_add_edge('knowledge', 3, 4, 'APPLIES_TO', '{"strength": 0.85}'::jsonb)
|
|
WHEN i.id = 4 THEN ruvector_add_edge('knowledge', 3, 5, 'APPLIES_TO', '{"strength": 0.9}'::jsonb)
|
|
END AS edge_id
|
|
FROM ids i
|
|
WHERE i.id <= 4;
|
|
|
|
-- Find path from Machine Learning to Computer Vision
|
|
SELECT ruvector_shortest_path('knowledge', 1, 4, 10);
|
|
|
|
-- ============================================================================
|
|
-- Recommendation System Example
|
|
-- ============================================================================
|
|
|
|
SELECT ruvector_create_graph('recommendations');
|
|
|
|
-- Add users and movies
|
|
SELECT ruvector_cypher(
|
|
'recommendations',
|
|
'CREATE (u1:User {name: ''Alice'', preference: ''SciFi''})
|
|
CREATE (u2:User {name: ''Bob'', preference: ''Action''})
|
|
CREATE (u3:User {name: ''Charlie'', preference: ''SciFi''})
|
|
CREATE (m1:Movie {title: ''Inception'', genre: ''SciFi''})
|
|
CREATE (m2:Movie {title: ''Interstellar'', genre: ''SciFi''})
|
|
CREATE (m3:Movie {title: ''The Matrix'', genre: ''SciFi''})
|
|
CREATE (m4:Movie {title: ''Die Hard'', genre: ''Action''})
|
|
RETURN u1, u2, u3, m1, m2, m3, m4',
|
|
NULL
|
|
);
|
|
|
|
-- Create watch history
|
|
SELECT ruvector_add_edge('recommendations', 1, 4, 'WATCHED', '{"rating": 5, "timestamp": "2024-01-15"}'::jsonb);
|
|
SELECT ruvector_add_edge('recommendations', 1, 5, 'WATCHED', '{"rating": 4, "timestamp": "2024-01-20"}'::jsonb);
|
|
SELECT ruvector_add_edge('recommendations', 2, 7, 'WATCHED', '{"rating": 5, "timestamp": "2024-01-18"}'::jsonb);
|
|
SELECT ruvector_add_edge('recommendations', 3, 4, 'WATCHED', '{"rating": 5, "timestamp": "2024-01-22"}'::jsonb);
|
|
SELECT ruvector_add_edge('recommendations', 3, 6, 'WATCHED', '{"rating": 4, "timestamp": "2024-01-25"}'::jsonb);
|
|
|
|
-- Get statistics
|
|
SELECT ruvector_graph_stats('recommendations');
|
|
|
|
-- ============================================================================
|
|
-- Organizational Hierarchy Example
|
|
-- ============================================================================
|
|
|
|
SELECT ruvector_create_graph('org_chart');
|
|
|
|
-- Create organizational structure
|
|
SELECT ruvector_cypher(
|
|
'org_chart',
|
|
'CREATE (ceo:Employee {name: ''Jane Doe'', title: ''CEO'', level: 1})
|
|
CREATE (cto:Employee {name: ''John Smith'', title: ''CTO'', level: 2})
|
|
CREATE (cfo:Employee {name: ''Emily Brown'', title: ''CFO'', level: 2})
|
|
CREATE (dev1:Employee {name: ''Alex Johnson'', title: ''Senior Dev'', level: 3})
|
|
CREATE (dev2:Employee {name: ''Sarah Wilson'', title: ''Senior Dev'', level: 3})
|
|
CREATE (acc1:Employee {name: ''Michael Davis'', title: ''Accountant'', level: 3})
|
|
RETURN ceo, cto, cfo, dev1, dev2, acc1',
|
|
NULL
|
|
);
|
|
|
|
-- Create reporting structure
|
|
SELECT ruvector_add_edge('org_chart', 2, 1, 'REPORTS_TO', '{}'::jsonb);
|
|
SELECT ruvector_add_edge('org_chart', 3, 1, 'REPORTS_TO', '{}'::jsonb);
|
|
SELECT ruvector_add_edge('org_chart', 4, 2, 'REPORTS_TO', '{}'::jsonb);
|
|
SELECT ruvector_add_edge('org_chart', 5, 2, 'REPORTS_TO', '{}'::jsonb);
|
|
SELECT ruvector_add_edge('org_chart', 6, 3, 'REPORTS_TO', '{}'::jsonb);
|
|
|
|
-- Find all employees reporting to CTO (directly or indirectly)
|
|
SELECT ruvector_shortest_path('org_chart', 4, 1, 5); -- Path from dev1 to CEO
|
|
SELECT ruvector_shortest_path('org_chart', 5, 1, 5); -- Path from dev2 to CEO
|
|
|
|
-- ============================================================================
|
|
-- Transport Network Example
|
|
-- ============================================================================
|
|
|
|
SELECT ruvector_create_graph('transport');
|
|
|
|
-- Add cities as nodes
|
|
SELECT ruvector_add_node('transport', ARRAY['City'], '{"name": "New York", "population": 8336817}'::jsonb);
|
|
SELECT ruvector_add_node('transport', ARRAY['City'], '{"name": "Boston", "population": 692600}'::jsonb);
|
|
SELECT ruvector_add_node('transport', ARRAY['City'], '{"name": "Philadelphia", "population": 1584064}'::jsonb);
|
|
SELECT ruvector_add_node('transport', ARRAY['City'], '{"name": "Washington DC", "population": 705749}'::jsonb);
|
|
|
|
-- Add routes with distances
|
|
SELECT ruvector_add_edge('transport', 1, 2, 'ROUTE', '{"distance": 215, "mode": "train", "duration": 4.5}'::jsonb);
|
|
SELECT ruvector_add_edge('transport', 1, 3, 'ROUTE', '{"distance": 95, "mode": "train", "duration": 1.5}'::jsonb);
|
|
SELECT ruvector_add_edge('transport', 3, 4, 'ROUTE', '{"distance": 140, "mode": "train", "duration": 2.5}'::jsonb);
|
|
SELECT ruvector_add_edge('transport', 2, 3, 'ROUTE', '{"distance": 310, "mode": "train", "duration": 5.5}'::jsonb);
|
|
|
|
-- Find shortest route by distance
|
|
SELECT ruvector_shortest_path_weighted('transport', 2, 4, 'distance');
|
|
|
|
-- Find fastest route by duration
|
|
SELECT ruvector_shortest_path_weighted('transport', 2, 4, 'duration');
|
|
|
|
-- ============================================================================
|
|
-- Analytics Queries
|
|
-- ============================================================================
|
|
|
|
-- Get all graphs with their statistics
|
|
SELECT
|
|
name,
|
|
(ruvector_graph_stats(name)::jsonb)->>'node_count' AS nodes,
|
|
(ruvector_graph_stats(name)::jsonb)->>'edge_count' AS edges
|
|
FROM (
|
|
SELECT unnest(ruvector_list_graphs()) AS name
|
|
) graphs;
|
|
|
|
-- ============================================================================
|
|
-- Cleanup
|
|
-- ============================================================================
|
|
|
|
-- Delete specific graph
|
|
-- SELECT ruvector_delete_graph('social_network');
|
|
|
|
-- Delete all graphs
|
|
-- SELECT ruvector_delete_graph(name)
|
|
-- FROM unnest(ruvector_list_graphs()) AS name;
|
|
|
|
-- ============================================================================
|
|
-- Performance Testing
|
|
-- ============================================================================
|
|
|
|
-- Create a larger graph for performance testing
|
|
SELECT ruvector_create_graph('perf_test');
|
|
|
|
-- Generate random nodes
|
|
DO $$
|
|
DECLARE
|
|
i INTEGER;
|
|
BEGIN
|
|
FOR i IN 1..1000 LOOP
|
|
PERFORM ruvector_add_node(
|
|
'perf_test',
|
|
ARRAY['Node'],
|
|
jsonb_build_object('id', i, 'value', random() * 100)
|
|
);
|
|
END LOOP;
|
|
END $$;
|
|
|
|
-- Generate random edges
|
|
DO $$
|
|
DECLARE
|
|
i INTEGER;
|
|
source_id INTEGER;
|
|
target_id INTEGER;
|
|
BEGIN
|
|
FOR i IN 1..5000 LOOP
|
|
source_id := 1 + floor(random() * 1000)::INTEGER;
|
|
target_id := 1 + floor(random() * 1000)::INTEGER;
|
|
IF source_id <> target_id THEN
|
|
BEGIN
|
|
PERFORM ruvector_add_edge(
|
|
'perf_test',
|
|
source_id,
|
|
target_id,
|
|
'CONNECTS',
|
|
jsonb_build_object('weight', random())
|
|
);
|
|
EXCEPTION WHEN OTHERS THEN
|
|
-- Ignore errors (e.g., duplicate edges)
|
|
NULL;
|
|
END;
|
|
END IF;
|
|
END LOOP;
|
|
END $$;
|
|
|
|
-- Check performance stats
|
|
SELECT ruvector_graph_stats('perf_test');
|
|
|
|
-- Test path finding performance
|
|
\timing on
|
|
SELECT ruvector_shortest_path('perf_test', 1, 500, 20);
|
|
SELECT ruvector_shortest_path_weighted('perf_test', 1, 500, 'weight');
|
|
\timing off
|
|
|
|
-- Cleanup performance test
|
|
-- SELECT ruvector_delete_graph('perf_test');
|