Rustでミスマッチタイプエラーを解決できない。

Question

実現したいこと

rustでSeaOrmを使ってDBからデータをselectしたいのですが、コンパイルエラーを解決できなくなってしまいました。

SeaOrmの公式ドキュメントや、個人技術ブログなどを参考に、DBへの接続情報をSeaOrmのselectするメソッドに渡すように作っているのですが、後述するソースではうまくいかず、後述するコンパイルエラーがでてしまいます。

発生している問題・分からないこと

・エラー内容
（expected DatabaseConnectionと書かれているので、 let db:DatabaseConnection としているのですが解消できません。）
error[E0308]: mismatched types
--> src/main.rs:9:34
|
9 | let db:DatabaseConnection = check_connection();
| ------------------ ^^^^^^^^^^^^^^^^^^ expected DatabaseConnection, found future
| |
| expected due to this

該当のソースコード

mod database;
use sea_orm::{DatabaseConnection, DbConn, DbErr, EntityTrait};

use crate::database::db_connection::check_connection;
use crate::database::entities::*;
use crate::database::entities::prelude::Employees;

fn main() {
    let db:DatabaseConnection  = check_connection();
    select_employees(&db);
}

pub async fn select_employees(db: &DbConn) -> Result<Option<employees::Model>, DbErr> {
    let selected: Option<employees::Model> = Employees::find_by_id(1).one(db).await?;
    Ok(selected)
}

※DBコネクションに関する箇所のみ
use std::env;
use std::time::Duration;
use dotenv::dotenv;
use sea_orm::{ActiveModelTrait, ActiveValue, ConnectOptions, Database, DbConn, DbErr, DeleteResult, EntityTrait, QueryFilter};
use sea_orm::ActiveValue::Set;
use crate::database::entities::{todos, users};
use crate::database::entities::prelude::Todos;
use sea_orm::ColumnTrait;

pub async  fn check_connection() -> Result<(), DbErr> {
    // DB接続のためのコネクションを生成
    let db = establish_connection().await?;

    assert!(db.ping().await.is_ok());
    db.clone().close().await.expect("panic!");
    println!("OK");
    Ok(())
}

pub async fn establish_connection() -> Result<DbConn, DbErr> {
    dotenv().ok();

    let url = env::var("DATABASE_URL").expect("DATABASE_URL is not found.");

    let mut opt = ConnectOptions::new(url);
    opt.max_connections(100)
        .min_connections(5)
        .connect_timeout(Duration::from_secs(8))
        .acquire_timeout(Duration::from_secs(8))
        .idle_timeout(Duration::from_secs(8))
        .max_lifetime(Duration::from_secs(8))
        .sqlx_logging(true)
        .sqlx_logging_level(log::LevelFilter::Info);

    //  DB接続のためのコネクションを生成
    Database::connect(opt).await
}

修正後

fn main() {
start();
}

pub async fn select_employees(db: &DbConn) -> Result<Optionemployees::Model, DbErr> {
let selected: Optionemployees::Model = Employees::find_by_id(1).one(db).await?;
Ok(selected)
}

pub async fn start() {
let db:DatabaseConnection = establish_connection().await.expect("connection error!");
let select_res:Optionemployees::Model = select_employees(&db).await.expect("database select error!");
println!("{}", select_res.unwrap().id);
}

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

上記のエラーメッセージと同様ですが画像のようにエラーが出ています。

補足

・db_connection.rsを参考にしたサイト
https://zenn.dev/collabostyle/articles/0641d73f776d80

Answer 1

expected DatabaseConnectionと書かれているので、 let db:DatabaseConnection としているのですが解消できません。

このエラーは、db変数の型を推論せずに明示的にDatabaseConnectionと記載しているからかなと思います。
db変数はDatabaseConnection型を期待している(expected)けど、check_connection関数の戻り値はResult<(), DbErr>型のためmismatched typesですね。

参考にしているサイトのソースコードだとGitHubの次のページの137行目あたりを見ると良いかなと思います。
sea-orm-postgresql-demo/src/lib.rs at main · codemountains/sea-orm-postgresql-demo · GitHub

check_connection関数はデータベースに接続できるかチェックするためのもののようですので、
コネクション(DatabaseConnection)を取得するのはestablish_connection関数の方みたいですね。

let db = establish_connection().await.expect("connection error!");

追記です。

コメントありがとうございます。
ビルドエラーが解消されたようで良かったです。

実行時のエラーも出なかったためDBの接続、selectまで実施されているようなのですがDBの値がコンソールに出力されませんでした。

警告なども出力されなかったでしょうか？
おそらく非同期のため実際の評価はされないままmainが終了してしまっているためかなと思いました。
下の例は"hello, world!"が出力されないです。

async fn hello_world() {
    println!("hello, world!");
}
fn main() {
    hello_world(); // Nothing is printed
    // warning: unused implementer of `Future` that must be used
    // = note: futures do nothing unless you `.await` or poll them
    // = note: `#[warn(unused_must_use)]` on by default
}

下の例は"hello, world!"が出力されます。

use futures::executor::block_on;
async fn hello_world() {
    println!("hello, world!");
}
fn main() {
    let future = hello_world(); // Nothing is printed
    block_on(future); // `future` is run and "hello, world!" is printed
}

修正後の該当のソースコード(main.rs)の場合だとstartをblock_onで囲めば良いことになるかなと思いました。

SeaORMの公式のドキュメントには下に引用したような記載があります。
非同期ランタイムはblock_onではなく、サードパーティのtokioが良く使われている印象でした。(Rustの非同期を検索すると上の方に出てくる)

Third, there are multiple async runtimes in Rust. async-std and tokio are the two most widely used. SeaORM's underlying driver, SQLx, supports both.
(機械翻訳)第3に、Rustには複数の非同期ランタイムがあります。async-stdとtokioが最も広く使われている2つです。SeaORMの基本ドライバであるSQLxは両方をサポートしています。

Async Programming | SeaORM 🐚 An async & dynamic ORM for Rust

下のリンク先なども参照してみると良いかなと思います。

async/.await Primer - Asynchronous Programming in Rust
Async Programming | SeaORM 🐚 An async & dynamic ORM for Rust

---

見当違いなことを書いてしまっていましたらごめんなさい・・
DBにはデータは登録されている前提で良いですよね？

Comments

[shoco]

返信ありがとうございます。
確かResult型はexpect()で推論すること忘れていました。
expect()を付与したらエラーが解消しました。
（awaitはasync内でしか使えないため別メソッドとして切り出しました）

さらに質問してしまい恐縮ですが、
エラーが解消され、DBをから値を出力するためプログラムを実行してみました。
実行時のエラーも出なかったためDBの接続、selectまで実施されているようなのですがDBの値がコンソールに出力されませんでした。

修正後のsrc/main.rsの記載したので値の取り出し方法をお教えいただけないでしょうか？

[shoco]

追記ありがとうございます。
すみません。原因は自分がずっと`cargo run`と間違えて`cargo build`をしていたしていたせいでした。
`cargo run`でうまく出力されました。