File: stream_reader_writer.rs

package info (click to toggle)
rust-geojson 0.24.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,436 kB
  • sloc: makefile: 2
file content (45 lines) | stat: -rw-r--r-- 1,249 bytes parent folder | download
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
use geojson::{de::deserialize_geometry, ser::serialize_geometry, FeatureReader, FeatureWriter};

use serde::{Deserialize, Serialize};
use std::error::Error;
use std::fs::File;
use std::io::{BufReader, BufWriter};

#[cfg(not(feature = "geo-types"))]
fn main() -> Result<(), Box<dyn Error>> {
    panic!("this example requires geo-types")
}

#[cfg(feature = "geo-types")]
fn main() -> Result<(), Box<dyn Error>> {
    #[derive(Serialize, Deserialize)]
    struct Country {
        #[serde(
            serialize_with = "serialize_geometry",
            deserialize_with = "deserialize_geometry"
        )]
        geometry: geo_types::Geometry,
        name: String,
    }

    let reader = {
        let file_reader = BufReader::new(File::open("tests/fixtures/countries.geojson")?);
        FeatureReader::from_reader(file_reader)
    };

    let mut writer = {
        let file_writer = BufWriter::new(File::create("example-output-countries.geojson")?);
        FeatureWriter::from_writer(file_writer)
    };

    let mut country_count = 0;
    for country in reader.deserialize::<Country>()? {
        let country = country?;
        country_count += 1;

        writer.serialize(&country)?;
    }

    assert_eq!(country_count, 180);
    Ok(())
}