RustとAWS Lambdaで定期的に価格チェックしてメール送信する

use reqwest::StatusCode;
use scraper::{Html, Selector};
mod utils;
#[tokio::main]
async fn main() {
    let client = utils::get_client();
    let url = "https://finance.yahoo.com";
    let result = client.get(url).send().await.unwrap();
    // HTMLを取得
    let raw_html = match result.status() {
        StatusCode::OK => result.text().await.unwrap(),
        _ => panic!("Something went wrong"),
    };
    let document = Html::parse_document(&raw_html);
    // 取得対象の要素
    let article_selector = Selector::parse("a.js-content-viewer").unwrap();
    // 取得！
    for element in document.select(&article_selector) {
        let inner = element.inner_html().to_string();
        let href = match element.value().attr("href") {
            Some(target_url) => target_url,
            _ => "no url found",
        };
        println!("Title: {}", &inner);
        println!("Link: {}", &href);
    }
}

use lambda_runtime::{service_fn, LambdaEvent, Error};
use serde_json::{json, Value};

#[tokio::main]
async fn main() -> Result<(), Error> {
    let func = service_fn(func);
    lambda_runtime::run(func).await?;
    Ok(())
}

// このメソッドをスクレイピングとメール送信のメソッドを呼び出すように修正します。
// これはサンプルコードそのままなので、一旦、 Hello World の表示を確認しても良いかもです。
async fn func(event: LambdaEvent<Value>) -> Result<Value, Error> {
    let (event, _context) = event.into_parts();
    let first_name = event["firstName"].as_str().unwrap_or("world");

    Ok(json!({ "message": format!("Hello, {}!", first_name) }))
}

Cannot start a runtime from within a runtime. This happens because a function (like `block_on`) attempted to block the current thread while the thread is being used to drive asynchronous tasks.'

async fn func(_event: LambdaEvent<Value>) -> Result<Value, Error> {
    tokio::task::spawn_blocking(|| {
        // 非同期処理はこの中で実行する
        let price_and_rates = utils::get_price_and_rates(); // スクレイピング
        utils::send_email(&price_and_rates); // メール送信
    })
    .await
    .expect("Blocking task panicked");
    Ok(json!({ "message": "success" }))
}

# マルチステージビルドで rust-musl-builder
FROM ekidd/rust-musl-builder:1.57.0 AS builder # versionには注意が必要
# build
USER rust
ADD --chown=rust:rust . ./
RUN cargo build --release

# とりあえず、コンテナ内でビルドするために用意したもの
version: '3.7'
services:
  builder:
    build:
      context: .
      target: builder
    image: builder:latest
    volumes:
      - ./:/home/rust/src/ # lambda.zipをコンテナ内で作成した時にホスト側でファイル取得できるようにする
      - builder-cargo-cache:/usr/local/cargo/registry
    tty: true # コンテナにattachしてビルドする用
    stdin_open: true # コンテナにattachしてビルドする用
  ...

volumes:
  builder-cargo-cache:

# x86_64-unknown-linux-musl 向けにクロスコンパイルする
cargo build --release --target x86_64-unknown-linux-musl

# lambda.zip を作成する
cp ./target/x86_64-unknown-linux-musl/release/${PROJECT_NAME} ./bootstrap
zip lambda.zip bootstrap

# bootstrap があるディレクトリでコマンド実行
docker run -it --rm -v $(pwd):/var/task:ro,delegated -e DOCKER_LAMBDA_USE_STDIN=1 -e AWS_LAMBDA_FUNCTION_MEMORY_SIZE=128 -e RUST_LOG=info lambci/lambda:provided
# コマンド実行後、引数の待ち状態になるので引数を入力して Ctrl+D を押下
# 引数不要なら {} でOK

# 実行結果例
START RequestId: 3387c069-4e4e-12d9-86aa-182c38367bf4 Version: $LATEST
END RequestId: 3387c069-4e4e-12d9-86aa-182c38367bf4
REPORT RequestId: 3387c069-4e4e-12d9-86aa-182c38367bf4	Init Duration: 4524.22 ms	Duration: 3.58 ms	Billed Duration: 4 ms	Memory Size: 128 MB	Max Memory Used: 10 MB

{"message":"success"}

# 環境変数が未設定だとエラーが発生
thread 'tokio-runtime-worker' panicked at 'called Result::unwrap() on an Err value: EnvVar(NotPresent)', src/utils.rs:54:59

# 日本時間 8-22時に動かす設定
cron(0 23,0-13 * * ? *)

# デフォルトのタイムアウト3秒で発生するエラー
"errorMessage": "2018-04-26T08:40:15.398Z 6f404961-492d-11e8-9047-f3c59f4ef90f Task timed out after 3.00 seconds"