# Rustでシングルスレッドサーバを作ってみる

以下に404 NOT FOUNDの実装を書いていきます。

# 1. 404 NOT FOUNDのページを作成する

HTMLファイルに404 NOT FOUNDの旨を書けば良いです。

一度作ったindex.htmlを真似して書いてみます。

<!-- public/404.html -->
<!DOCTYPE html>
<html lang="en">
  <head>
    <meta charset="utf-8">
  </head>
  <body>
    <h1>404 NOT FOUND...</h1>
    <p>I do not want to work...</p>
  </body>
</html>

こんな感じで良いでしょう。

# 2. 404 NOT FOUNDのページを設定する

src/main.rsに追記すれば良いです。設定したルーティング設定に引っかからなかったときに404.htmlを返せば良いので, そのように設定します。

// src/main.rs

// ~~ 略 ~~

fn handle_connection(mut stream: TcpStream) {
  // TcpStreamインスタンスのデータを読み取るためのバッファを用意
  let mut buffer = [0;512];
  stream.read(&mut buffer).unwrap();
  println!("Request: {}", String::from_utf8_lossy(&buffer[..]));

  let get = b"GET / HTTP/1.1\r\n";

  if buffer.starts_with(get) {
    let mut file = File::open("public/index.html").unwrap();
    let mut contents = String::new();
    file.read_to_string(&mut contents).unwrap();

    let response = format!("HTTP/1.1 200 OK\r\n\r\n{}", contents);
    stream.write(response.as_bytes()).unwrap();
    // streamに完全にwriteされるように, flush()を呼び出しておく
    stream.flush().unwrap();
+ } else {
+   let mut file = File::open("public/404.html").unwrap();
+   let mut contents = String::new();
+   file.read_to_string(&mut contents).unwrap();
+
+   let response = format!("HTTP/1.1 404 NOT FOUND\r\n\r\n{}", contents);
+   stream.write(response.as_bytes()).unwrap();
+   // streamに完全にwriteされるように, flush()を呼び出しておく
+   stream.flush().unwrap();  
  }
}

以上です。これでhttp://127.0.0.1:3000/以外のURLでリクエストを飛ばすと, public/404.htmlの内容がレスポンスとして返されることが確認できると思います。

# さらにリファクタリングしてみる

上のコードだとコードとして重複している部分が複数あります。そこで, 重複している部分をまとめてみようと思います。

よくコードを見ると, 異なっている部分はステータスコードとレスポンスとして返すファイルの場所のみであることがわかります。したがって, その2つを変数としてレスポンスを返してみます。

コードは次の通りです。

src/main.rs

// ~~ 略 ~~

fn handle_connection(mut stream: TcpStream) {
  // TcpStreamインスタンスのデータを読み取るためのバッファを用意
  let mut buffer = [0;512];
  stream.read(&mut buffer).unwrap();
  println!("Request: {}", String::from_utf8_lossy(&buffer[..]));

  let get = b"GET / HTTP/1.1\r\n";

  let (status_line, resource_path) = if buffer.starts_with(get) {
    ("HTTP/1.1 200 OK\r\n\r\n", "public/index.html")
  } else {
    ("HTTP/1.1 404 NOT FOUND\r\n\r\n", "public/404.html")
  };

  let mut file = File::open(resource_path).unwrap();
  let mut contents = String::new();
  file.read_to_string(&mut contents).unwrap();
  
  let response = format!("{}{}", status_line, contents);
  stream.write(response.as_bytes()).unwrap();
  // streamに完全にwriteされるように, flush()を呼び出しておく
  stream.flush().unwrap();
}

お疲れ様でした。