实现用户注册与激活

本章将实现最终的功能：用户注册，并发送激活邮件。

handler

我们增加几个 handler，其它都过于简单这里不再赘述。重点来看一下注册时的 handler：

// src/handler.rs

pub async fn register(
    Extension(state): Extension<Arc<AppState>>,
    Form(frm): Form<form::RegisterForm>,
) -> Result<(StatusCode, HeaderMap, ())> {
    let cfg = &state.cfg;
    let active_code = model::user::ActiveCode {
        code: gen_active_code(&frm.email),
        email: frm.email,
        email_cfg: cfg.email.clone(),
    };

    let payload = serde_json::to_string(&active_code).map_err(Error::from)?;

    // 发送消息
    topic::send(
        &cfg.rabbitmq.dsn,
        &cfg.rabbitmq.exchange_name,
        &cfg.rabbitmq.queue_name,
        &cfg.rabbitmq.routing_key,
        &payload,
    )
    .await
    .map_err(Error::from)?;

    redirect("/active")
}

消息消费者（发送邮件验证码）

// src/main.rs

#[tokio::main]
async fn main() {
   // ...

    tokio::spawn(send_active_code(cfg.clone()));
	
    // ...
}

async fn send_active_code(cfg: Config) {
    topic::receive(
        &cfg.rabbitmq.dsn,
        &cfg.rabbitmq.exchange_name,
        &cfg.rabbitmq.queue_name,
        &cfg.rabbitmq.routing_key,
        "MAIL",
        move |delivery: DeliveryResult| async move {
            tracing::debug!("aaa");
            let delivery = match delivery {
                Ok(Some(delivery)) => delivery,
                Ok(None) => {
                    tracing::error!("None ");
                    return;
                }
                Err(err) => {
                    tracing::error!("Failed to consume queue message {}", err);
                    return;
                }
            };

            let message = String::from_utf8_lossy(&delivery.data);
            tracing::info!("Received a message: {}", message);

            let ac: model::user::ActiveCode = serde_json::from_str(&message).unwrap();
            let from = ac.email_cfg.username.clone();
            let resp = email::send(
                ac.email_cfg,
                model::email::Email {
                    from,
                    to: ac.email,
                    subject: format!("激活账号"),
                    body: format!("你的激活码是：{}", ac.code),
                },
            )
            .await
            .unwrap();

            tracing::info!("Send email: {:?}", resp);
            delivery.ack(BasicAckOptions::default()).await.unwrap();
        },
    )
    .await
    .unwrap();
}

本章代码位于03/注册与激活分支。