f002445c9e
- Add PostgreSQL client functionality for database interactions. - Add Redis client functionality for cache and data store operations. - Extend Rhai scripting with PostgreSQL and Redis client modules. - Add documentation and test cases for both clients.
133 lines
4.5 KiB
Plaintext
133 lines
4.5 KiB
Plaintext
// PostgreSQL Basic Operations Example
|
|
//
|
|
// This example demonstrates how to use the PostgreSQL client module to:
|
|
// - Connect to a PostgreSQL database
|
|
// - Create a table
|
|
// - Insert data
|
|
// - Query data
|
|
// - Update data
|
|
// - Delete data
|
|
// - Drop a table
|
|
//
|
|
// Prerequisites:
|
|
// - PostgreSQL server must be running
|
|
// - Environment variables should be set for connection details:
|
|
// - POSTGRES_HOST: PostgreSQL server host (default: localhost)
|
|
// - POSTGRES_PORT: PostgreSQL server port (default: 5432)
|
|
// - POSTGRES_USER: PostgreSQL username (default: postgres)
|
|
// - POSTGRES_PASSWORD: PostgreSQL password
|
|
// - POSTGRES_DB: PostgreSQL database name (default: postgres)
|
|
|
|
// Helper function to check if PostgreSQL is available
|
|
fn is_postgres_available() {
|
|
try {
|
|
// Try to execute a simple connection
|
|
let connect_result = pg_connect();
|
|
return connect_result;
|
|
} catch(err) {
|
|
print(`PostgreSQL connection error: ${err}`);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Main function
|
|
fn main() {
|
|
print("=== PostgreSQL Basic Operations Example ===");
|
|
|
|
// Check if PostgreSQL is available
|
|
let postgres_available = is_postgres_available();
|
|
if !postgres_available {
|
|
print("PostgreSQL server is not available. Please check your connection settings.");
|
|
return;
|
|
}
|
|
|
|
print("✓ Connected to PostgreSQL server");
|
|
|
|
// Define table name
|
|
let table_name = "rhai_example_users";
|
|
|
|
// Step 1: Create a table
|
|
print("\n1. Creating table...");
|
|
let create_table_query = `
|
|
CREATE TABLE IF NOT EXISTS ${table_name} (
|
|
id SERIAL PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
email TEXT UNIQUE NOT NULL,
|
|
age INTEGER,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
`;
|
|
|
|
let create_result = pg_execute(create_table_query);
|
|
print(`✓ Table created (result: ${create_result})`);
|
|
|
|
// Step 2: Insert data
|
|
print("\n2. Inserting data...");
|
|
let insert_queries = [
|
|
`INSERT INTO ${table_name} (name, email, age) VALUES ('Alice', 'alice@example.com', 30)`,
|
|
`INSERT INTO ${table_name} (name, email, age) VALUES ('Bob', 'bob@example.com', 25)`,
|
|
`INSERT INTO ${table_name} (name, email, age) VALUES ('Charlie', 'charlie@example.com', 35)`
|
|
];
|
|
|
|
for query in insert_queries {
|
|
let insert_result = pg_execute(query);
|
|
print(`✓ Inserted row (result: ${insert_result})`);
|
|
}
|
|
|
|
// Step 3: Query all data
|
|
print("\n3. Querying all data...");
|
|
let select_query = `SELECT * FROM ${table_name}`;
|
|
let rows = pg_query(select_query);
|
|
|
|
print(`Found ${rows.len()} rows:`);
|
|
for row in rows {
|
|
print(` ID: ${row.id}, Name: ${row.name}, Email: ${row.email}, Age: ${row.age}, Created: ${row.created_at}`);
|
|
}
|
|
|
|
// Step 4: Query specific data
|
|
print("\n4. Querying specific data...");
|
|
let select_one_query = `SELECT * FROM ${table_name} WHERE name = 'Alice'`;
|
|
let alice = pg_query_one(select_one_query);
|
|
|
|
print(`Found Alice:`);
|
|
print(` ID: ${alice.id}, Name: ${alice.name}, Email: ${alice.email}, Age: ${alice.age}`);
|
|
|
|
// Step 5: Update data
|
|
print("\n5. Updating data...");
|
|
let update_query = `UPDATE ${table_name} SET age = 31 WHERE name = 'Alice'`;
|
|
let update_result = pg_execute(update_query);
|
|
print(`✓ Updated Alice's age (result: ${update_result})`);
|
|
|
|
// Verify update
|
|
let verify_query = `SELECT * FROM ${table_name} WHERE name = 'Alice'`;
|
|
let updated_alice = pg_query_one(verify_query);
|
|
print(` Updated Alice: ID: ${updated_alice.id}, Name: ${updated_alice.name}, Age: ${updated_alice.age}`);
|
|
|
|
// Step 6: Delete data
|
|
print("\n6. Deleting data...");
|
|
let delete_query = `DELETE FROM ${table_name} WHERE name = 'Bob'`;
|
|
let delete_result = pg_execute(delete_query);
|
|
print(`✓ Deleted Bob (result: ${delete_result})`);
|
|
|
|
// Verify deletion
|
|
let count_query = `SELECT COUNT(*) as count FROM ${table_name}`;
|
|
let count_result = pg_query_one(count_query);
|
|
print(` Remaining rows: ${count_result.count}`);
|
|
|
|
// Step 7: Drop table
|
|
print("\n7. Dropping table...");
|
|
let drop_query = `DROP TABLE IF EXISTS ${table_name}`;
|
|
let drop_result = pg_execute(drop_query);
|
|
print(`✓ Dropped table (result: ${drop_result})`);
|
|
|
|
// Reset connection
|
|
print("\n8. Resetting connection...");
|
|
let reset_result = pg_reset();
|
|
print(`✓ Reset connection (result: ${reset_result})`);
|
|
|
|
print("\nExample completed successfully!");
|
|
}
|
|
|
|
// Run the main function
|
|
main();
|