about summary refs log tree commit diff
path: root/users/flokli/archeology/parse_bucket_logs.rs
blob: 1fbba0506821185ab9114eaf15423933f0436b24 (plain) (blame)
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
use std::env;
use std::process::Command;
use std::process::ExitCode;

fn main() -> ExitCode {
    let args: Vec<String> = env::args().collect();
    if args.len() != 3 {
        eprintln!("needs two args, input s3 url (glob) and output pq file");
        return ExitCode::FAILURE;
    }

    let input_files = &args[1];
    let output_file = &args[2];

    let mut cmd = Command::new("clickhouse-local");
    cmd.arg("--progress")
        .arg("-q")
        .arg(format!(r#"SELECT
        key,
        toInt64(nullif(http_status, '-')) AS http_status,
        toInt64(nullif(object_size_str, '-')) AS object_size,
        toInt64(nullif(bytes_sent_str, '-')) AS bytes_sent,
        nullif(user_agent, '-') AS user_agent,
        operation,
        nullif(requester, '-') AS requester,
        parseDateTime(timestamp_str, '%d/%b/%Y:%k:%i:%s %z') AS timestamp
    FROM s3(
        '{}',
        'Regexp',
        'owner String , bucket String, timestamp_str String, remote_ip String, requester LowCardinality(String), request_id String, operation LowCardinality(String), key String, request_uri String, http_status String, error_code String, bytes_sent_str String, object_size_str String, total_time String, turn_around_time String, referer String, user_agent String, version_id String, host_id String, signature_version String, cipher_suite String, authentication_type String, host_header String, tls_version String, access_point_arn String, acl_required String'
    )
    SETTINGS format_regexp = '(\\S+) (\\S+) \\[(.*)\\] (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) ((?:\\S+ \\S+ \\S+)|\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+) (\\S+).*'
    INTO OUTFILE '{}' COMPRESSION 'zstd' FORMAT Parquet"#, input_files, output_file));

    cmd.status().expect("clickhouse-local failed");

    ExitCode::SUCCESS
}