gauc 0.8.1

Couchbase Rust Adapter / CLI
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

extern crate log;
extern crate env_logger;

extern crate clap;
extern crate ctrlc;
extern crate gauc;
extern crate iron;

use clap::{App, Arg};

use gauc::cli;
use gauc::web;
use gauc::client::Client;

use std::env;
use std::process::exit;
use std::sync::{Arc, Mutex};
use std::sync::atomic::{AtomicBool, Ordering};

const AUTHORS: &'static str = env!("CARGO_PKG_AUTHORS");
const DESCRIPTION: &'static str = env!("CARGO_PKG_DESCRIPTION");
const VERSION: &'static str = env!("CARGO_PKG_VERSION");

//const DEFAULT_HOST: &'static str = "localhost";
//const DEFAULT_BUCKET: &'static str = "default";
const DEFAULT_URI: &'static str = "couchbase://localhost/default";

/// Handler of Ctrl+C
fn install_ctrl_c_handler() {
    let running = Arc::new(AtomicBool::new(true));
    let r = Arc::clone(&running);
    let _ = ctrlc::set_handler(move || {
        println!("");
        println!("");

        println!("Received ctrl+c, exiting ...");
        r.store(false, Ordering::SeqCst);
        exit(0);
    });
}

/// Web Server Entrypoint
fn main() {
    install_ctrl_c_handler();


    let couchbase_uri = match env::var("COUCHBASE_URI") {
        Ok(uri) => uri,
        Err(_) => DEFAULT_URI.to_string()
    };

    // Specify program options
    let matches = App::new(DESCRIPTION)
        .version(VERSION)
        .author(AUTHORS)
        .arg(Arg::with_name("interactive")
            .help("Interactive mode")
            .short("i")
            .long("interactive")
        )
        .arg(Arg::with_name("rest")
            .help("Run REST Server")
            .short("r")
            .long("rest")
        )
        .arg(Arg::with_name("rest-port")
            .help("REST Port")
            .short("p")
            .long("rest-port")
            .default_value("5000")
        )
        .arg(Arg::with_name("url")
            .help("URI - connection string")
            .short("u")
            .long("uri")
            .default_value(&couchbase_uri[..])
        )
        .arg(Arg::with_name("verbose")
            .help("Verbose mode")
            .short("v")
            .long("verbose")
            .multiple(true)
        )
        .get_matches();

    match matches.occurrences_of("verbose") {
        1 => env::set_var("RUST_LOG", "warn"),
        2 => env::set_var("RUST_LOG", "info"),
        3 => env::set_var("RUST_LOG", "debug"),
        _ => {}
    }

    env_logger::init().unwrap();

    if matches.is_present("interactive") {
        if let Ok(mut client) = Client::connect(matches.value_of("url").unwrap(), None) {
            cli::main(&matches, &mut client);
        }
    }

    let port: u16 = matches.value_of("rest-port").unwrap().to_string().parse::<u16>().unwrap();
    if matches.is_present("rest") {
        if let Ok(client) = Client::connect(matches.value_of("url").unwrap(), None) {
            web::start_web(&Arc::new(Mutex::new(client)), port);
        }
    }
}