use of com.baremaps.osm.domain.Relation in project baremaps by baremaps.
the class PostgresRelationRepository method put.
/**
* {@inheritDoc}
*/
@Override
public void put(List<Relation> values) throws RepositoryException {
if (values.isEmpty()) {
return;
}
try (Connection connection = dataSource.getConnection();
PreparedStatement statement = connection.prepareStatement(insert)) {
for (Relation value : values) {
statement.clearParameters();
setValue(statement, value);
statement.addBatch();
}
statement.executeBatch();
} catch (SQLException | JsonProcessingException e) {
throw new RepositoryException(e);
}
}
use of com.baremaps.osm.domain.Relation in project baremaps by baremaps.
the class OpenStreetMapBenchmark method entityStream.
@Benchmark
@BenchmarkMode(Mode.SingleShotTime)
@Warmup(iterations = 2)
@Measurement(iterations = 5)
public void entityStream() throws IOException {
AtomicLong nodes = new AtomicLong(0);
AtomicLong ways = new AtomicLong(0);
AtomicLong relations = new AtomicLong(0);
try (InputStream inputStream = new BufferedInputStream(Files.newInputStream(path))) {
new OsmPbfParser().entities(inputStream).forEach(new EntityConsumerAdapter() {
@Override
public void match(Node node) {
nodes.incrementAndGet();
}
@Override
public void match(Way way) {
ways.incrementAndGet();
}
@Override
public void match(Relation relation) {
relations.incrementAndGet();
}
});
}
}
use of com.baremaps.osm.domain.Relation in project baremaps by baremaps.
the class OpenStreetMapTest method process.
void process(Stream<Entity> stream, long headerCount, long boundCount, long nodeCount, long wayCount, long relationCount) {
AtomicLong headers = new AtomicLong(0);
AtomicLong bounds = new AtomicLong(0);
AtomicLong nodes = new AtomicLong(0);
AtomicLong ways = new AtomicLong(0);
AtomicLong relations = new AtomicLong(0);
stream.forEach(new EntityConsumer() {
@Override
public void match(Header header) {
assertNotNull(header);
assertEquals("osmium/1.8.0", header.getWritingProgram());
headers.incrementAndGet();
}
@Override
public void match(Bound bound) {
assertNotNull(bound);
assertEquals(43.75169, bound.getMaxLat(), 0.000001);
assertEquals(7.448637, bound.getMaxLon(), 0.000001);
assertEquals(43.72335, bound.getMinLat(), 0.000001);
assertEquals(7.409205, bound.getMinLon(), 0.000001);
bounds.incrementAndGet();
}
@Override
public void match(Node node) {
assertNotNull(node);
nodes.incrementAndGet();
}
@Override
public void match(Way way) {
assertNotNull(way);
ways.incrementAndGet();
}
@Override
public void match(Relation relation) {
assertNotNull(relation);
relations.incrementAndGet();
}
});
assertEquals(headerCount, headers.get());
assertEquals(boundCount, bounds.get());
assertEquals(nodeCount, nodes.get());
assertEquals(wayCount, ways.get());
assertEquals(relationCount, relations.get());
}
use of com.baremaps.osm.domain.Relation in project baremaps by baremaps.
the class RelationGeometryTest method handleRelation.
Geometry handleRelation(String file) throws IOException {
InputStream input = new GZIPInputStream(this.getClass().getResourceAsStream(file));
List<Entity> entities = new OsmXmlParser().entities(input).collect(Collectors.toList());
LongDataMap<Coordinate> coordinates = new MockLongDataMap<>(entities.stream().filter(e -> e instanceof Node).map(e -> (Node) e).collect(Collectors.toMap(n -> n.getId(), n -> new Coordinate(n.getLon(), n.getLat()))));
LongDataMap<List<Long>> references = new MockLongDataMap<>(entities.stream().filter(e -> e instanceof Way).map(e -> (Way) e).collect(Collectors.toMap(w -> w.getId(), w -> w.getNodes())));
Relation relation = entities.stream().filter(e -> e instanceof Relation).map(e -> (Relation) e).findFirst().get();
new CreateGeometryConsumer(coordinates, references).match(relation);
return relation.getGeometry();
}
use of com.baremaps.osm.domain.Relation in project baremaps by baremaps.
the class PostgresRelationRepository method copy.
/**
* {@inheritDoc}
*/
@Override
public void copy(List<Relation> values) throws RepositoryException {
if (values.isEmpty()) {
return;
}
try (Connection connection = dataSource.getConnection()) {
PGConnection pgConnection = connection.unwrap(PGConnection.class);
try (CopyWriter writer = new CopyWriter(new PGCopyOutputStream(pgConnection, copy))) {
writer.writeHeader();
for (Relation value : values) {
writer.startRow(10);
writer.writeLong(value.getId());
writer.writeInteger(value.getInfo().getVersion());
writer.writeInteger(value.getInfo().getUid());
writer.writeLocalDateTime(value.getInfo().getTimestamp());
writer.writeLong(value.getInfo().getChangeset());
writer.writeJsonb(toJson(value.getTags()));
writer.writeLongList(value.getMembers().stream().map(Member::getRef).collect(Collectors.toList()));
writer.writeIntegerList(value.getMembers().stream().map(Member::getType).map(MemberType::ordinal).collect(Collectors.toList()));
writer.writeStringList(value.getMembers().stream().map(Member::getRole).collect(Collectors.toList()));
writer.writeGeometry(value.getGeometry());
}
}
} catch (IOException | SQLException ex) {
throw new RepositoryException(ex);
}
}
Aggregations