example.springdata.redis.commands.GeoOperationsTests.java Source code

Introduction

Here is the source code for example.springdata.redis.commands.GeoOperationsTests.java

Source

/*
 * Copyright 2017 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package example.springdata.redis.commands;

import static org.assertj.core.api.Assertions.*;

import example.springdata.redis.test.util.RequiresRedisServer;

import java.util.List;

import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.data.geo.Circle;
import org.springframework.data.geo.Distance;
import org.springframework.data.geo.GeoResults;
import org.springframework.data.geo.Point;
import org.springframework.data.redis.connection.RedisGeoCommands.DistanceUnit;
import org.springframework.data.redis.connection.RedisGeoCommands.GeoLocation;
import org.springframework.data.redis.core.GeoOperations;
import org.springframework.data.redis.core.RedisOperations;
import org.springframework.test.context.junit4.SpringRunner;

/**
 * Show usage of redis geo-index operations using Template API provided by {@link GeoOperations}.
 *
 * @author Mark Paluch
 */
@RunWith(SpringRunner.class)
@SpringBootApplication
public class GeoOperationsTests {

    // we only want to run this tests when redis is up an running
    public static @ClassRule RequiresRedisServer requiresServer = RequiresRedisServer.onLocalhost().atLeast("3.2");

    @Autowired
    RedisOperations<String, String> operations;
    GeoOperations<String, String> geoOperations;

    @Before
    public void before() {

        geoOperations = operations.opsForGeo();

        geoOperations.geoAdd("Sicily", new Point(13.361389, 38.115556), "Arigento");
        geoOperations.geoAdd("Sicily", new Point(15.087269, 37.502669), "Catania");
        geoOperations.geoAdd("Sicily", new Point(13.583333, 37.316667), "Palermo");
    }

    /**
     * Look up points using a geo-index member as reference.
     */
    @Test
    public void geoRadiusByMember() {

        GeoResults<GeoLocation<String>> byDistance = geoOperations.geoRadiusByMember("Sicily", "Palermo",
                new Distance(100, DistanceUnit.KILOMETERS));

        assertThat(byDistance).hasSize(2).extracting("content.name").contains("Arigento", "Palermo");

        GeoResults<GeoLocation<String>> greaterDistance = geoOperations.geoRadiusByMember("Sicily", "Palermo",
                new Distance(200, DistanceUnit.KILOMETERS));

        assertThat(greaterDistance).hasSize(3).extracting("content.name").contains("Arigento", "Catania",
                "Palermo");
    }

    /**
     * Lookup points within a circle around coordinates.
     */
    @Test
    public void geoRadius() {

        Circle circle = new Circle(new Point(13.583333, 37.316667), //
                new Distance(100, DistanceUnit.KILOMETERS));
        GeoResults<GeoLocation<String>> result = geoOperations.geoRadius("Sicily", circle);

        assertThat(result).hasSize(2).extracting("content.name").contains("Arigento", "Palermo");
    }

    /**
     * Calculate the distance between two geo-index members.
     */
    @Test
    public void geoDistance() {

        Distance distance = geoOperations.geoDist("Sicily", "Catania", "Palermo", DistanceUnit.KILOMETERS);

        assertThat(distance.getValue()).isBetween(130d, 140d);
    }

    /**
     * Return the geo-hash.
     */
    @Test
    public void geoHash() {

        List<String> geohashes = geoOperations.geoHash("Sicily", "Catania", "Palermo");

        assertThat(geohashes).hasSize(2).contains("sqdtr74hyu0", "sq9sm1716e0");
    }
}