aboutsummaryrefslogtreecommitdiffstats
path: root/src/sqlite.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/sqlite.rs')
-rw-r--r--src/sqlite.rs25
1 files changed, 25 insertions, 0 deletions
diff --git a/src/sqlite.rs b/src/sqlite.rs
index 10dbb61..8094398 100644
--- a/src/sqlite.rs
+++ b/src/sqlite.rs
@@ -6,4 +6,29 @@ use rusqlite;
/// Use the SQLite rules for type affinity described in:
/// https://sqlite.org/datatype3.html#determination_of_column_affinity
pub fn affinity(type_name: &str) -> rusqlite::types::Type {
+ use rusqlite::types::Type;
+
+ let type_name = type_name.to_uppercase();
+
+ if type_name.contains("INT") {
+ return Type::Integer;
+ } else if type_name.contains("CHAR")
+ || type_name.contains("CLOB")
+ || type_name.contains("TEXT")
+ {
+ return Type::Text;
+ } else if type_name.contains("BLOB")
+ || type_name.is_empty()
+ {
+ return Type::Blob;
+ } else if type_name.contains("REAL")
+ || type_name.contains("FLOA")
+ || type_name.contains("DOUB")
+ {
+ return Type::Real;
+ }
+
+ // TODO: Numeric affinity
+
+ Type::Text
}