pg2parquet 0.2.2

Command line tool for exporting PostgreSQL tables or queries into Parquet files
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

use postgres::{self, Row, types::Kind};
use uuid::Uuid;

pub fn identify_row(row: &Row) -> String {

	// first row with simple data type
	for (i, column) in row.columns().iter().enumerate() {
		let t = column.type_();
		if t.kind().clone() == Kind::Simple {
			match t.name() {
				"text" => {
					match row.get::<usize, Option<String>>(i) {
						Some(v) =>
							return format!("{}={:?}", column.name(), v),
						None => ()
					}
				},
				"oid" => {
					match row.get::<usize, Option<u32>>(i) {
						Some(v) =>
							return format!("{}={:?}", column.name(), v),
						None => ()
					}
				},
				"int4" => {
					match row.get::<usize, Option<i32>>(i) {
						Some(v) =>
							return format!("{}={:?}", column.name(), v),
						None => ()
					}
				},
				"int8" => {
					match row.get::<usize, Option<i64>>(i) {
						Some(v) =>
							return format!("{}={:?}", column.name(), v),
						None => ()
					}
				},
				"uuid" => {
					match row.get::<usize, Option<Uuid>>(i) {
						Some(v) =>
							return format!("{}={:?}", column.name(), v),
						None => ()
					}
				},
				_ => ()
			}
		}
	}
	"Row ¯\\_(ツ)_/¯".to_owned()
}