2 releases
0.1.1 | Nov 6, 2024 |
---|---|
0.1.0 | Nov 6, 2024 |
#655 in Database interfaces
23 downloads per month
69KB
1K
SLoC
SQL Tools
A rust crate meant to make creating SQL queries easy. The goal is to have multiple different versions of SQL (SQLite, Postgres, etc..) however it's only Oracle SQL right now. The original crate I made that this is an evolution of is oracle_sql_tools.
How To Use
In your cargo.toml
file:
[dependencies]
sql_tools = "0.1.0"
# chrono is required if you're working with dates
chrono = "0.4.26"
To start using SQL Tools, you need a new connection.
use sql_tools::sql_variations::OracleConnect;
let username = "new_user";
let password = "password123!";
let connection_string = "my-secure.connection/database";
let conn = OracleConnect::new(connection_string, username, password);
Once you established a connection type, you can use the various methods in this crate to interact with your database. These options are select, update, insert, and create.
SQLDataTypes
This is the enum that is used to apply the proper data type to the data that's being selected, updated, or inserted.
pub enum SQLDataTypes {
Varchar(String),
Number(i64),
Float(f64),
Date(chrono::NaiveDateTime),
NULL,
}
SELECT
For the select method, you add the table you want to select from, then the columns in a vector. If you want to select all, simply input vec!["*"]
. You can add a where clause to filter out the rows you want, just like writing a SQL query.
let conn = OracleConnect::new(connection_string, username, password);
let data: Vec<Vec<SQLDataTypes>> = conn
.select("regional_sales", vec!["product_id", "revenue", "sale_date"])
.where_in("product_id", vec!["1001", "4567"])
.and_not("city", vec!["Austin", "Dallas"])
.build()?;
data.iter().for_each(|row: &Vec<SQLDataTypes>| { println!("{:?}", row) });
UPDATE
Updates a table's column(s) based on criteria set in the optional where clauses. Updates can return Ok() or the number of rows that were updated.
let conn = OracleConnect::new(connection_string, username, password);
conn.update("global_sales")
.set("continent", "North America")
.where_in("country", vec!["Canada", "United States", "Mexico"])
.build()?;
// If you want to get the number of rows that were updated
let count: usize = conn
.update("global_sales")
.set("continent", "North America")
.where_in("country", vec!["Canada", "United States", "Mexico"])
.build_return_count()?;
INSERT
Inserts a grid (two-dimensional vector) of data into your database. Can take any type that has the ToSQLData
trait implemented. If the table does not exist, it will automatically create a new table (will have an abort option in a future update).
let conn = OracleConnect::new(connection_string, username, password);
let data: Vec<Vec<&str>> = vec![
vec!["Column_A", "Column_B", "Column_C"],
vec!["a1", "b1", "c1"],
vec!["a2", "b2", "c2"],
vec!["a3", "b3", "c3"],
];
conn.insert("my_table", data).build()?;
If you have a grid of strings that have integers, dates, etc.. that you want to be formatted properly before being inserted into a table then you want to add the .format_grid_strings()
method.
let data: Vec<Vec<&str>> = vec![
vec!["product_id", "date_sold", "price"],
vec!["P-001", "09/15/2024", "54.99"],
vec!["P-002", "10/22/2024", "19.99"],
vec!["P-003", "11/04/2024", "39.99"],
];
conn.insert("sales_data", data)
// Will convert the "date_sold" column into chrono::NaiveDateTime
// and the "price" column into f64.
.format_grid_strings()?
.build()?;
CREATE
Creates a table using a vector of the CreateColumns
struct and the CreateDataTypes
to apply the correct types to the new columns.
let conn = OracleConnect::new(connection_string, username, password);
let columns = vec![
CreateColumns{ name: "Column_A".to_string(), data_type: CreateDataTypes::VARCHAR(20 as usize) },
CreateColumns{ name: "Column_B".to_string(), data_type: CreateDataTypes::NUMBER },
CreateColumns{ name: "Column_C".to_string(), data_type: CreateDataTypes::FLOAT },
];
conn.create()
.table("my_table", columns)
.build()?;
You can add a column after you initiated the create table process.
let my_table: CreateTable = conn.create()
.table("my_table", columns);
if add_date_col == true {
my_table.add_column("Column_D".to_string(), CreateDataTypes::DATE);
}
my_table.build()?;
ToSQLData
ToSQLData
is the trait that is implemented for the insert()
method which is used to convert various data types to SQLDataTypes
.
To implement a local enum:
enum MyEnum {
Name(String),
Age(i64)
}
impl ToSQLData for MyEnum {
fn fmt_data(self) -> SQLDataTypes {
match self {
MyEnum::Name(val) => SQLDataTypes::Varchar(val.into()),
MyEnum::Age(val) => SQLDataTypes::Number(val.into()),
}
}
}
To implement a foreign enum:
use some_crate::SomeForeignType;
struct MyType<'a>(&'a SomeForeignType);
impl ToSQLData for MyType<'_> {
fn fmt_data(self) -> SQLDataTypes {
match self {
MyType(SomeForeignType::Int(val)) => SQLDataTypes::Number(*val),
MyType(SomeForeignType::Float(val)) => SQLDataTypes::Float(*val),
MyType(SomeForeignType::String(val)) => SQLDataTypes::Varchar(val.to_owned()),
MyType(SomeForeignType::None) => SQLDataTypes::NULL,
}
}
}
Dependencies
~13–22MB
~299K SLoC