datafusion-server 0.21.0

Web server library for session-based queries using Arrow and other large datasets as data sources.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

// http - Creates and register HTTP server
// Sasaki, Naoki <nsasaki@sal.co.jp> October 16, 2022
//

use std::sync::Arc;

use axum::http::{header, Method};
use std::net::SocketAddr;

use crate::context::session_manager::SessionManager;
use crate::server::routes;
use crate::settings::Settings;

pub async fn create_server<S: SessionManager>(
    session_mgr: Arc<tokio::sync::Mutex<S>>,
) -> Result<
    (
        axum::serve::Serve<tokio::net::TcpListener, axum::Router, axum::Router>,
        SocketAddr,
    ),
    anyhow::Error,
> {
    let app = routes::register::<S>(&session_mgr).layer(
        tower_http::cors::CorsLayer::new()
            .allow_headers(vec![
                header::ACCEPT,
                header::ACCEPT_LANGUAGE,
                header::AUTHORIZATION,
                header::CONTENT_LANGUAGE,
                header::CONTENT_TYPE,
            ])
            .allow_methods(vec![
                Method::GET,
                Method::POST,
                Method::PUT,
                Method::DELETE,
                Method::OPTIONS,
            ])
            .allow_origin(tower_http::cors::Any)
            .allow_credentials(false),
    );

    let sock_addr = format!(
        "{}:{}",
        Settings::global().server.address,
        Settings::global().server.port,
    )
    .parse::<SocketAddr>()?;

    let listener = tokio::net::TcpListener::bind(sock_addr).await?;

    Ok((axum::serve(listener, app), sock_addr))
}