caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

CLI Generate Entity for Postgres (#61)

Browse Source 
* cli generate entity for Postgres

* Fixup
This commit is contained in:
Billy Chan 2021-08-07 18:23:52 +08:00 committed by GitHub
parent 1c73ab0759
commit 3701c86637
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 30 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
sea-orm-cli/Cargo.toml
View File

@ -20,8 +20,9 @@ clap = { version = "^2.33.3" }
dotenv = { version = "^0.15" }
async-std = { version = "^1.9", features = [ "attributes" ] }
sea-orm-codegen = { path = "../sea-orm-codegen" }
sea-schema = { version = "^0.2", default-features = false, features = [
sea-schema = { version = "^0.2", git = "https://github.com/SeaQL/sea-schema.git", branch = "pg-discovery-writer", default-features = false, features = [
"sqlx-mysql",
"sqlx-postgres",
"discovery",
"writer",
] }

37
sea-orm-cli/src/main.rs
View File

@ -1,8 +1,6 @@
use clap::ArgMatches;
use dotenv::dotenv;
use sea_orm_codegen::{EntityTransformer, OutputFile};
use sea_schema::mysql::discovery::SchemaDiscovery;
use sqlx::MySqlPool;
use std::{error::Error, fmt::Display, fs, io::Write, path::Path, process::Command};
mod cli;
@ -28,14 +26,33 @@ async fn run_generate_command(matches: &ArgMatches<'_>) -> Result<(), Box<dyn Er
let schema = args.value_of("DATABASE_SCHEMA").unwrap();
let output_dir = args.value_of("OUTPUT_DIR").unwrap();
let connection = MySqlPool::connect(url).await?;
let schema_discovery = SchemaDiscovery::new(connection, schema);
let schema = schema_discovery.discover().await;
let table_stmts = schema
.tables
.into_iter()
.map(|schema| schema.write())
.collect();
let table_stmts = if url.starts_with("mysql://") {
use sea_schema::mysql::discovery::SchemaDiscovery;
use sqlx::MySqlPool;
let connection = MySqlPool::connect(url).await?;
let schema_discovery = SchemaDiscovery::new(connection, schema);
let schema = schema_discovery.discover().await;
schema
.tables
.into_iter()
.map(|schema| schema.write())
.collect()
} else if url.starts_with("postgres://") {
use sea_schema::postgres::discovery::SchemaDiscovery;
use sqlx::PgPool;
let connection = PgPool::connect(url).await?;
let schema_discovery = SchemaDiscovery::new(connection, schema);
let schema = schema_discovery.discover().await;
schema
.tables
.into_iter()
.map(|schema| schema.write())
.collect()
} else {
panic!("This database is not supported ({})", url)
};
let output = EntityTransformer::transform(table_stmts)?.generate();

2
sea-orm-codegen/Cargo.toml
View File

@ -16,7 +16,7 @@ name = "sea_orm_codegen"
path = "src/lib.rs"
[dependencies]
sea-query = { version = "~0.12.8" }
sea-query = { version = "~0.12.8", git = "https://github.com/SeaQL/sea-query.git" }
syn = { version = "^1", default-features = false, features = [
"derive",
"parsing",