11#[ global_allocator]
22static ALLOC : snmalloc_rs:: SnMalloc = snmalloc_rs:: SnMalloc ;
33
4- use std:: io:: Write ;
5-
64use actix:: prelude:: * ;
75use actix_http:: error:: ErrorInternalServerError ;
86use actix_http:: { HttpService , KeepAlive } ;
@@ -11,12 +9,13 @@ use actix_web::dev::{AppConfig, Body, Server};
119use actix_web:: http:: { header:: CONTENT_TYPE , header:: SERVER , HeaderValue , StatusCode } ;
1210use actix_web:: { web, App , Error , HttpRequest , HttpResponse } ;
1311use bytes:: BytesMut ;
12+ use yarte:: ywrite_html;
1413
1514mod db_pg;
1615mod models;
1716mod utils;
1817use crate :: db_pg:: { PgConnection , RandomWorld , RandomWorlds , TellFortune , UpdateWorld } ;
19- use crate :: utils:: { FortunesYarteTemplate , Writer } ;
18+ use crate :: utils:: Writer ;
2019
2120async fn world_row ( db : web:: Data < Addr < PgConnection > > ) -> Result < HttpResponse , Error > {
2221 let res = db
@@ -99,13 +98,10 @@ async fn fortune(db: web::Data<Addr<PgConnection>>) -> Result<HttpResponse, Erro
9998 match res {
10099 Ok ( fortunes) => {
101100 let mut body = BytesMut :: with_capacity ( 2048 ) ;
102- let mut writer = Writer ( & mut body) ;
103- let _ = write ! ( writer, "{}" , FortunesYarteTemplate { fortunes } ) ;
101+ ywrite_html ! ( body, "{{> fortune }}" ) ;
104102
105- let mut res = HttpResponse :: with_body (
106- StatusCode :: OK ,
107- Body :: Bytes ( body. freeze ( ) . into ( ) ) ,
108- ) ;
103+ let mut res =
104+ HttpResponse :: with_body ( StatusCode :: OK , Body :: Bytes ( body. freeze ( ) ) ) ;
109105 res. headers_mut ( )
110106 . insert ( SERVER , HeaderValue :: from_static ( "Actix" ) ) ;
111107 res. headers_mut ( ) . insert (
0 commit comments