This tutorial shows you how build a simple Rust application with CockroachDB using a PostgreSQL-compatible driver. We've tested and can recommend the Rust Postgres driver, so that driver is featured here.
Before You Begin
Make sure you have already installed CockroachDB.
Step 1. Install the Rust Postgres driver
Install the Rust Postgres driver as described in the official documentation.
Step 2. Start a single-node cluster
For the purpose of this tutorial, you need only one CockroachDB node running in insecure mode:
$ cockroach start \
--insecure \
--store=hello-1 \
--host=localhost
Step 3. Create a user
In a new terminal, as the root user, use the cockroach user command to create a new user, maxroach.
$ cockroach user set maxroach --insecure
Step 4. Create a database and grant privileges
As the root user, use the built-in SQL client to create a bank database.
$ cockroach sql --insecure -e 'CREATE DATABASE bank'
Then grant privileges to the maxroach user.
$ cockroach sql --insecure -e 'GRANT ALL ON DATABASE bank TO maxroach'
Step 5. Create a table in the new database
As the maxroach user, use the built-in SQL client to create an accounts table in the new database.
$ cockroach sql --insecure \
--database=bank \
--user=maxroach \
-e 'CREATE TABLE accounts (id INT PRIMARY KEY, balance INT)'
Step 6. Run the Rust code
Now that you have a database and a user, you'll run code to create a table and insert some rows, and then you'll run code to read and update values as an atomic transaction.
Basic Statements
First, use the following code to connect as the maxroach user and execute some basic SQL statements, inserting rows and reading and printing the rows.
Download the basic-sample.rs file, or create the file yourself and copy the code into it.
extern crate postgres;
use postgres::{Connection, TlsMode};
fn main() {
let conn = Connection::connect("postgresql://maxroach@localhost:26257/bank", TlsMode::None)
.unwrap();
// Insert two rows into the "accounts" table.
conn.execute(
"INSERT INTO accounts (id, balance) VALUES (1, 1000), (2, 250)",
&[],
).unwrap();
// Print out the balances.
println!("Initial balances:");
for row in &conn.query("SELECT id, balance FROM accounts", &[]).unwrap() {
let id: i64 = row.get(0);
let balance: i64 = row.get(1);
println!("{} {}", id, balance);
}
}
Transaction (with retry logic)
Next, use the following code to again connect as the maxroach user but this time execute a batch of statements as an atomic transaction to transfer funds from one account to another, where all included statements are either committed or aborted.
Download the txn-sample.rs file, or create the file yourself and copy the code into it.
SERIALIZABLE isolation level, CockroachDB may require the client to retry a transaction in case of read/write contention. CockroachDB provides a generic retry function that runs inside a transaction and retries it as needed. You can copy and paste the retry function from here into your code.extern crate postgres;
use postgres::{Connection, TlsMode, Result};
use postgres::transaction::Transaction;
use self::postgres::error::T_R_SERIALIZATION_FAILURE;
/// Runs op inside a transaction and retries it as needed.
/// On non-retryable failures, the transaction is aborted and
/// rolled back; on success, the transaction is committed.
fn execute_txn<T, F>(conn: &Connection, mut op: F) -> Result<T>
where
F: FnMut(&Transaction) -> Result<T>,
{
let txn = conn.transaction()?;
loop {
let sp = txn.savepoint("cockroach_restart")?;
match op(&sp).and_then(|t| sp.commit().map(|_| t)) {
Err(ref err) if err.as_db()
.map(|e| e.code == T_R_SERIALIZATION_FAILURE)
.unwrap_or(false) => {},
r => break r,
}
}.and_then(|t| txn.commit().map(|_| t))
}
fn transfer_funds(txn: &Transaction, from: i64, to: i64, amount: i64) -> Result<()> {
// Read the balance.
let from_balance: i64 = txn.query("SELECT balance FROM accounts WHERE id = $1", &[&from])?
.get(0)
.get(0);
assert!(from_balance >= amount);
// Perform the transfer.
txn.execute(
"UPDATE accounts SET balance = balance - $1 WHERE id = $2",
&[&amount, &from],
)?;
txn.execute(
"UPDATE accounts SET balance = balance + $1 WHERE id = $2",
&[&amount, &to],
)?;
Ok(())
}
fn main() {
let conn = Connection::connect("postgresql://maxroach@localhost:26257/bank", TlsMode::None)
.unwrap();
// Run a transfer in a transaction.
execute_txn(&conn, |txn| transfer_funds(txn, 1, 2, 100)).unwrap();
// Check account balances after the transaction.
for row in &conn.query("SELECT id, balance FROM accounts", &[]).unwrap() {
let id: i64 = row.get(0);
let balance: i64 = row.get(1);
println!("{} {}", id, balance);
}
}
After running the code, use the built-in SQL client to verify that funds were transferred from one account to another:
$ cockroach sql --insecure -e 'SELECT id, balance FROM accounts' --database=bank
+----+---------+
| id | balance |
+----+---------+
| 1 | 900 |
| 2 | 350 |
+----+---------+
(2 rows)
What's Next?
Read more about using the Rust Postgres driver.
You might also be interested in using a local cluster to explore the following core CockroachDB features:
Was this helpful?
