GeoCommandsTest.java
package redis.clients.jedis.commands.jedis;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.junit.jupiter.api.Assertions.fail;
import static redis.clients.jedis.util.AssertUtil.assertByteArrayListEquals;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedClass;
import org.junit.jupiter.params.provider.MethodSource;
import redis.clients.jedis.GeoCoordinate;
import redis.clients.jedis.RedisProtocol;
import redis.clients.jedis.args.GeoUnit;
import redis.clients.jedis.params.GeoSearchParam;
import redis.clients.jedis.resps.GeoRadiusResponse;
import redis.clients.jedis.params.GeoAddParams;
import redis.clients.jedis.params.GeoRadiusParam;
import redis.clients.jedis.params.GeoRadiusStoreParam;
import redis.clients.jedis.util.GeoCoordinateMatcher;
import redis.clients.jedis.util.SafeEncoder;
@ParameterizedClass
@MethodSource("redis.clients.jedis.commands.CommandsTestsParameters#respVersions")
public class GeoCommandsTest extends JedisCommandsTestBase {
final byte[] bfoo = { 0x01, 0x02, 0x03, 0x04 };
final byte[] bA = { 0x0A };
final byte[] bB = { 0x0B };
final byte[] bC = { 0x0C };
final byte[] bD = { 0x0D };
final byte[] bNotexist = { 0x0F };
private static final double EPSILON = 1e-5;
public GeoCommandsTest(RedisProtocol protocol) {
super(protocol);
}
@Test
public void geoadd() {
assertEquals(1, jedis.geoadd("foo", 1, 2, "a"));
assertEquals(0, jedis.geoadd("foo", 2, 3, "a"));
Map<String, GeoCoordinate> coordinateMap = new HashMap<>();
coordinateMap.put("a", new GeoCoordinate(3, 4));
coordinateMap.put("b", new GeoCoordinate(2, 3));
coordinateMap.put("c", new GeoCoordinate(3.314, 2.3241));
assertEquals(2, jedis.geoadd("foo", coordinateMap));
// binary
assertEquals(1, jedis.geoadd(bfoo, 1, 2, bA));
assertEquals(0, jedis.geoadd(bfoo, 2, 3, bA));
Map<byte[], GeoCoordinate> bcoordinateMap = new HashMap<>();
bcoordinateMap.put(bA, new GeoCoordinate(3, 4));
bcoordinateMap.put(bB, new GeoCoordinate(2, 3));
bcoordinateMap.put(bC, new GeoCoordinate(3.314, 2.3241));
assertEquals(2, jedis.geoadd(bfoo, bcoordinateMap));
}
@Test
public void geoaddWithParams() {
assertEquals(1, jedis.geoadd("foo", 1, 2, "a"));
Map<String, GeoCoordinate> coordinateMap = new HashMap<>();
coordinateMap.put("a", new GeoCoordinate(3, 4));
assertEquals(0, jedis.geoadd("foo", GeoAddParams.geoAddParams().nx(), coordinateMap));
assertEquals(1, jedis.geoadd("foo", GeoAddParams.geoAddParams().xx().ch(), coordinateMap));
coordinateMap.clear();
coordinateMap.put("b", new GeoCoordinate(6, 7));
// never add elements.
assertEquals(0, jedis.geoadd("foo", GeoAddParams.geoAddParams().xx(), coordinateMap));
assertEquals(1, jedis.geoadd("foo", GeoAddParams.geoAddParams().nx(), coordinateMap));
// binary
assertEquals(1, jedis.geoadd(bfoo, 1, 2, bA));
Map<byte[], GeoCoordinate> bcoordinateMap = new HashMap<>();
bcoordinateMap.put(bA, new GeoCoordinate(3, 4));
assertEquals(0, jedis.geoadd(bfoo, GeoAddParams.geoAddParams().nx(), bcoordinateMap));
assertEquals(1, jedis.geoadd(bfoo, GeoAddParams.geoAddParams().xx().ch(), bcoordinateMap));
bcoordinateMap.clear();
bcoordinateMap.put(bB, new GeoCoordinate(6, 7));
// never add elements.
assertEquals(0, jedis.geoadd(bfoo, GeoAddParams.geoAddParams().xx(), bcoordinateMap));
assertEquals(1, jedis.geoadd(bfoo, GeoAddParams.geoAddParams().nx(), bcoordinateMap));
}
@Test
public void geodist() {
prepareGeoData();
Double dist = jedis.geodist("foo", "a", "b");
assertEquals(157149, dist.intValue());
dist = jedis.geodist("foo", "a", "b", GeoUnit.KM);
assertEquals(157, dist.intValue());
dist = jedis.geodist("foo", "a", "b", GeoUnit.MI);
assertEquals(97, dist.intValue());
dist = jedis.geodist("foo", "a", "b", GeoUnit.FT);
assertEquals(515583, dist.intValue());
// binary
dist = jedis.geodist(bfoo, bA, bB);
assertEquals(157149, dist.intValue());
dist = jedis.geodist(bfoo, bA, bB, GeoUnit.KM);
assertEquals(157, dist.intValue());
dist = jedis.geodist(bfoo, bA, bB, GeoUnit.MI);
assertEquals(97, dist.intValue());
dist = jedis.geodist(bfoo, bA, bB, GeoUnit.FT);
assertEquals(515583, dist.intValue());
}
@Test
public void geohash() {
prepareGeoData();
List<String> hashes = jedis.geohash("foo", "a", "b", "notexist");
assertEquals(3, hashes.size());
assertEquals("s0dnu20t9j0", hashes.get(0));
assertEquals("s093jd0k720", hashes.get(1));
assertNull(hashes.get(2));
// binary
List<byte[]> bhashes = jedis.geohash(bfoo, bA, bB, bNotexist);
assertEquals(3, bhashes.size());
assertArrayEquals(SafeEncoder.encode("s0dnu20t9j0"), bhashes.get(0));
assertArrayEquals(SafeEncoder.encode("s093jd0k720"), bhashes.get(1));
assertNull(bhashes.get(2));
}
@Test
public void geopos() {
prepareGeoData();
List<GeoCoordinate> coordinates = jedis.geopos("foo", "a", "b", "notexist");
assertEquals(3, coordinates.size());
assertEquals(3.0, coordinates.get(0).getLongitude(), EPSILON);
assertEquals(4.0, coordinates.get(0).getLatitude(), EPSILON);
assertEquals(2.0, coordinates.get(1).getLongitude(), EPSILON);
assertEquals(3.0, coordinates.get(1).getLatitude(), EPSILON);
assertNull(coordinates.get(2));
List<GeoCoordinate> bcoordinates = jedis.geopos(bfoo, bA, bB, bNotexist);
assertEquals(3, bcoordinates.size());
assertEquals(3.0, bcoordinates.get(0).getLongitude(), EPSILON);
assertEquals(4.0, bcoordinates.get(0).getLatitude(), EPSILON);
assertEquals(2.0, bcoordinates.get(1).getLongitude(), EPSILON);
assertEquals(3.0, bcoordinates.get(1).getLatitude(), EPSILON);
assertNull(bcoordinates.get(2));
}
@Test
public void georadius() {
// prepare datas
Map<String, GeoCoordinate> coordinateMap = new HashMap<>();
coordinateMap.put("Palermo", new GeoCoordinate(13.361389, 38.115556));
coordinateMap.put("Catania", new GeoCoordinate(15.087269, 37.502669));
jedis.geoadd("Sicily", coordinateMap);
List<GeoRadiusResponse> members = jedis.georadius("Sicily", 15, 37, 200, GeoUnit.KM);
assertEquals(2, members.size());
// sort
members = jedis.georadius("Sicily", 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortDescending());
assertEquals(2, members.size());
assertEquals("Catania", members.get(1).getMemberByString());
assertEquals("Palermo", members.get(0).getMemberByString());
// sort, count 1
members = jedis.georadius("Sicily", 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortAscending().count(1));
assertEquals(1, members.size());
// sort, count 1, withdist, withcoord
members = jedis.georadius("Sicily", 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortAscending().count(1).withCoord().withDist().withHash());
assertEquals(1, members.size());
GeoRadiusResponse response = members.get(0);
assertEquals(56.4413, response.getDistance(), EPSILON);
assertEquals(15.087269, response.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.502669, response.getCoordinate().getLatitude(), EPSILON);
assertEquals(3479447370796909L, response.getRawScore());
// sort, count 1, with hash
members = jedis.georadius("Sicily", 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortAscending().count(1).withHash());
assertEquals(1, members.size());
response = members.get(0);
assertEquals(3479447370796909L, response.getRawScore());
// sort, count 1, any
members = jedis.georadius("Sicily", 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortDescending().count(1, true));
assertEquals(1, members.size());
response = members.get(0);
assertTrue(coordinateMap.containsKey(response.getMemberByString()));
}
@Test
public void georadiusStore() {
// prepare datas
Map<String, GeoCoordinate> coordinateMap = new HashMap<>();
coordinateMap.put("Palermo", new GeoCoordinate(13.361389, 38.115556));
coordinateMap.put("Catania", new GeoCoordinate(15.087269, 37.502669));
jedis.geoadd("Sicily", coordinateMap);
long size = jedis.georadiusStore("Sicily", 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam(),
GeoRadiusStoreParam.geoRadiusStoreParam().store("SicilyStore"));
assertEquals(2, size);
List<String> expected = new ArrayList<>();
expected.add("Palermo");
expected.add("Catania");
assertEquals(expected, jedis.zrange("SicilyStore", 0, -1));
}
@Test
public void georadiusReadonly() {
// prepare datas
Map<String, GeoCoordinate> coordinateMap = new HashMap<>();
coordinateMap.put("Palermo", new GeoCoordinate(13.361389, 38.115556));
coordinateMap.put("Catania", new GeoCoordinate(15.087269, 37.502669));
jedis.geoadd("Sicily", coordinateMap);
List<GeoRadiusResponse> members = jedis.georadiusReadonly("Sicily", 15, 37, 200, GeoUnit.KM);
assertEquals(2, members.size());
// sort
members = jedis.georadiusReadonly("Sicily", 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending());
assertEquals(2, members.size());
assertEquals("Catania", members.get(0).getMemberByString());
assertEquals("Palermo", members.get(1).getMemberByString());
// sort, count 1
members = jedis.georadiusReadonly("Sicily", 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending().count(1));
assertEquals(1, members.size());
// sort, count 1, withdist, withcoord
members = jedis.georadiusReadonly("Sicily", 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending().count(1).withCoord().withDist());
assertEquals(1, members.size());
GeoRadiusResponse response = members.get(0);
assertEquals(56.4413, response.getDistance(), EPSILON);
assertEquals(15.087269, response.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.502669, response.getCoordinate().getLatitude(), EPSILON);
}
@Test
public void georadiusBinary() {
// prepare datas
Map<byte[], GeoCoordinate> bcoordinateMap = new HashMap<>();
bcoordinateMap.put(bA, new GeoCoordinate(13.361389, 38.115556));
bcoordinateMap.put(bB, new GeoCoordinate(15.087269, 37.502669));
jedis.geoadd(bfoo, bcoordinateMap);
List<GeoRadiusResponse> members = jedis.georadius(bfoo, 15, 37, 200, GeoUnit.KM);
assertEquals(2, members.size());
// sort
members = jedis.georadius(bfoo, 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortAscending());
assertEquals(2, members.size());
assertArrayEquals(bB, members.get(0).getMember());
assertArrayEquals(bA, members.get(1).getMember());
// sort, count 1
members = jedis.georadius(bfoo, 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortAscending().count(1));
assertEquals(1, members.size());
// sort, count 1, withdist, withcoord
members = jedis.georadius(bfoo, 15, 37, 200, GeoUnit.KM, GeoRadiusParam.geoRadiusParam()
.sortAscending().count(1).withCoord().withDist());
assertEquals(1, members.size());
GeoRadiusResponse response = members.get(0);
assertEquals(56.4413, response.getDistance(), EPSILON);
assertEquals(15.087269, response.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.502669, response.getCoordinate().getLatitude(), EPSILON);
}
@Test
public void georadiusStoreBinary() {
// prepare datas
Map<byte[], GeoCoordinate> bcoordinateMap = new HashMap<>();
bcoordinateMap.put(bA, new GeoCoordinate(13.361389, 38.115556));
bcoordinateMap.put(bB, new GeoCoordinate(15.087269, 37.502669));
jedis.geoadd(bfoo, bcoordinateMap);
long size = jedis.georadiusStore(bfoo, 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam(),
GeoRadiusStoreParam.geoRadiusStoreParam().store("SicilyStore"));
assertEquals(2, size);
List<byte[]> bexpected = new ArrayList<>();
bexpected.add(bA);
bexpected.add(bB);
assertByteArrayListEquals(bexpected, jedis.zrange("SicilyStore".getBytes(), 0, -1));
}
@Test
public void georadiusReadonlyBinary() {
// prepare datas
Map<byte[], GeoCoordinate> bcoordinateMap = new HashMap<>();
bcoordinateMap.put(bA, new GeoCoordinate(13.361389, 38.115556));
bcoordinateMap.put(bB, new GeoCoordinate(15.087269, 37.502669));
jedis.geoadd(bfoo, bcoordinateMap);
List<GeoRadiusResponse> members = jedis.georadiusReadonly(bfoo, 15, 37, 200, GeoUnit.KM);
assertEquals(2, members.size());
// sort
members = jedis.georadiusReadonly(bfoo, 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending());
assertEquals(2, members.size());
assertArrayEquals(bB, members.get(0).getMember());
assertArrayEquals(bA, members.get(1).getMember());
// sort, count 1
members = jedis.georadiusReadonly(bfoo, 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending().count(1));
assertEquals(1, members.size());
// sort, count 1, withdist, withcoord
members = jedis.georadiusReadonly(bfoo, 15, 37, 200, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending().count(1).withCoord().withDist());
assertEquals(1, members.size());
GeoRadiusResponse response = members.get(0);
assertEquals(56.4413, response.getDistance(), EPSILON);
assertEquals(15.087269, response.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.502669, response.getCoordinate().getLatitude(), EPSILON);
}
@Test
public void georadiusByMember() {
jedis.geoadd("Sicily", 13.583333, 37.316667, "Agrigento");
jedis.geoadd("Sicily", 13.361389, 38.115556, "Palermo");
jedis.geoadd("Sicily", 15.087269, 37.502669, "Catania");
List<GeoRadiusResponse> members = jedis.georadiusByMember("Sicily", "Agrigento", 100,
GeoUnit.KM);
assertEquals(2, members.size());
members = jedis.georadiusByMember("Sicily", "Agrigento", 100, GeoUnit.KM, GeoRadiusParam
.geoRadiusParam().sortAscending());
assertEquals(2, members.size());
assertEquals("Agrigento", members.get(0).getMemberByString());
assertEquals("Palermo", members.get(1).getMemberByString());
members = jedis.georadiusByMember("Sicily", "Agrigento", 100, GeoUnit.KM, GeoRadiusParam
.geoRadiusParam().sortAscending().count(1).withCoord().withDist());
assertEquals(1, members.size());
GeoRadiusResponse member = members.get(0);
assertEquals("Agrigento", member.getMemberByString());
assertEquals(0, member.getDistance(), EPSILON);
assertEquals(13.583333, member.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.316667, member.getCoordinate().getLatitude(), EPSILON);
}
@Test
public void georadiusByMemberStore() {
jedis.geoadd("Sicily", 13.583333, 37.316667, "Agrigento");
jedis.geoadd("Sicily", 13.361389, 38.115556, "Palermo");
jedis.geoadd("Sicily", 15.087269, 37.502669, "Catania");
long size = jedis.georadiusByMemberStore("Sicily", "Agrigento", 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam(),
GeoRadiusStoreParam.geoRadiusStoreParam().store("SicilyStore"));
assertEquals(2, size);
List<String> expected = new ArrayList<>();
expected.add("Agrigento");
expected.add("Palermo");
assertEquals(expected, jedis.zrange("SicilyStore", 0, -1));
}
@Test
public void georadiusByMemberReadonly() {
jedis.geoadd("Sicily", 13.583333, 37.316667, "Agrigento");
jedis.geoadd("Sicily", 13.361389, 38.115556, "Palermo");
jedis.geoadd("Sicily", 15.087269, 37.502669, "Catania");
List<GeoRadiusResponse> members = jedis.georadiusByMemberReadonly("Sicily", "Agrigento", 100,
GeoUnit.KM);
assertEquals(2, members.size());
members = jedis.georadiusByMemberReadonly("Sicily", "Agrigento", 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending());
assertEquals(2, members.size());
assertEquals("Agrigento", members.get(0).getMemberByString());
assertEquals("Palermo", members.get(1).getMemberByString());
members = jedis.georadiusByMemberReadonly("Sicily", "Agrigento", 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending().count(1).withCoord().withDist());
assertEquals(1, members.size());
GeoRadiusResponse member = members.get(0);
assertEquals("Agrigento", member.getMemberByString());
assertEquals(0, member.getDistance(), EPSILON);
assertEquals(13.583333, member.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.316667, member.getCoordinate().getLatitude(), EPSILON);
}
@Test
public void georadiusByMemberBinary() {
jedis.geoadd(bfoo, 13.583333, 37.316667, bA);
jedis.geoadd(bfoo, 13.361389, 38.115556, bB);
jedis.geoadd(bfoo, 15.087269, 37.502669, bC);
List<GeoRadiusResponse> members = jedis.georadiusByMember(bfoo, bA, 100, GeoUnit.KM);
assertEquals(2, members.size());
members = jedis.georadiusByMember(bfoo, bA, 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending());
assertEquals(2, members.size());
assertArrayEquals(bA, members.get(0).getMember());
assertArrayEquals(bB, members.get(1).getMember());
members = jedis.georadiusByMember(bfoo, bA, 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending().count(1).withCoord().withDist());
assertEquals(1, members.size());
GeoRadiusResponse member = members.get(0);
assertArrayEquals(bA, member.getMember());
assertEquals(0, member.getDistance(), EPSILON);
assertEquals(13.583333, member.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.316667, member.getCoordinate().getLatitude(), EPSILON);
}
@Test
public void georadiusByMemberStoreBinary() {
jedis.geoadd(bfoo, 13.583333, 37.316667, bA);
jedis.geoadd(bfoo, 13.361389, 38.115556, bB);
jedis.geoadd(bfoo, 15.087269, 37.502669, bC);
assertEquals(2, jedis.georadiusByMemberStore(bfoo, bA, 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam(),
GeoRadiusStoreParam.geoRadiusStoreParam().store("SicilyStore")));
List<byte[]> bexpected = new ArrayList<>();
bexpected.add(bA);
bexpected.add(bB);
assertByteArrayListEquals(bexpected, jedis.zrange("SicilyStore".getBytes(), 0, -1));
}
@Test
public void georadiusByMemberReadonlyBinary() {
jedis.geoadd(bfoo, 13.583333, 37.316667, bA);
jedis.geoadd(bfoo, 13.361389, 38.115556, bB);
jedis.geoadd(bfoo, 15.087269, 37.502669, bC);
List<GeoRadiusResponse> members = jedis.georadiusByMemberReadonly(bfoo, bA, 100, GeoUnit.KM);
assertEquals(2, members.size());
members = jedis.georadiusByMemberReadonly(bfoo, bA, 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending());
assertEquals(2, members.size());
assertArrayEquals(bA, members.get(0).getMember());
assertArrayEquals(bB, members.get(1).getMember());
members = jedis.georadiusByMemberReadonly(bfoo, bA, 100, GeoUnit.KM,
GeoRadiusParam.geoRadiusParam().sortAscending().count(1).withCoord().withDist());
assertEquals(1, members.size());
GeoRadiusResponse member = members.get(0);
assertArrayEquals(bA, member.getMember());
assertEquals(0, member.getDistance(), EPSILON);
assertEquals(13.583333, member.getCoordinate().getLongitude(), EPSILON);
assertEquals(37.316667, member.getCoordinate().getLatitude(), EPSILON);
}
@Test
public void geosearch() {
jedis.geoadd("barcelona", 2.1909389952632d, 41.433791470673d, "place1");
jedis.geoadd("barcelona", 2.1873744593677d, 41.406342043777d, "place2");
jedis.geoadd("barcelona", 2.583333d, 41.316667d, "place3");
// FROMLONLAT and BYRADIUS
List<GeoRadiusResponse> members = jedis.geosearch("barcelona",
new GeoCoordinate(2.191d,41.433d), 1000, GeoUnit.M);
assertEquals(1, members.size());
assertEquals("place1", members.get(0).getMemberByString());
// using Params
members = jedis.geosearch("barcelona", new GeoSearchParam().byRadius(3000, GeoUnit.M)
.fromLonLat(2.191d,41.433d).desc());
assertEquals(2, members.size());
assertEquals("place2", members.get(0).getMemberByString());
// FROMMEMBER and BYRADIUS
members = jedis.geosearch("barcelona","place3", 100, GeoUnit.KM);
assertEquals(3, members.size());
// using Params
members = jedis.geosearch("barcelona", new GeoSearchParam().fromMember("place1")
.byRadius(100, GeoUnit.KM).withDist().withCoord().withHash().count(2));
assertEquals(2, members.size());
assertEquals("place1", members.get(0).getMemberByString());
GeoRadiusResponse res2 = members.get(1);
assertEquals("place2", res2.getMemberByString());
assertEquals(3.0674157, res2.getDistance(), 5);
assertEquals(new GeoCoordinate(2.187376320362091, 41.40634178640635), res2.getCoordinate());
// FROMMEMBER and BYBOX
members = jedis.geosearch("barcelona","place3", 100, 100, GeoUnit.KM);
assertEquals(3, members.size());
// using Params
members = jedis.geosearch("barcelona", new GeoSearchParam().fromMember("place3")
.byBox(100, 100, GeoUnit.KM).asc().count(1, true));
assertEquals(1, members.size());
// FROMLONLAT and BYBOX
members = jedis.geosearch("barcelona", new GeoCoordinate(2.191, 41.433),
1, 1, GeoUnit.KM);
assertEquals(1, members.size());
// using Params
members = jedis.geosearch("barcelona", new GeoSearchParam().byBox(1,1, GeoUnit.KM)
.fromLonLat(2.191, 41.433).withDist().withCoord());
assertEquals(1, members.size());
assertEquals("place1", members.get(0).getMemberByString());
assertEquals(0.0881, members.get(0).getDistance(), 10);
assertThat(members.get(0).getCoordinate(),
GeoCoordinateMatcher.atCoordinates(2.19093829393386841, 41.43379028184083523));
}
@Test
public void geosearchNegative() {
// combine byradius and bybox
try {
jedis.geosearch("barcelona", new GeoSearchParam()
.byRadius(3000, GeoUnit.M)
.byBox(300, 300, GeoUnit.M));
fail();
} catch (IllegalArgumentException ignored) { }
// without byradius and without bybox
try {
jedis.geosearch("barcelona", new GeoSearchParam().fromMember("foobar"));
fail();
} catch (IllegalArgumentException ignored) { }
// combine frommember and fromlonlat
try {
jedis.geosearch("barcelona", new GeoSearchParam()
.fromMember("foobar")
.fromLonLat(10,10));
fail();
} catch (IllegalArgumentException ignored) { }
// without frommember and without fromlonlat
try {
jedis.geosearch("barcelona", new GeoSearchParam().byRadius(10, GeoUnit.MI));
fail();
} catch (IllegalArgumentException ignored) { }
}
@Test
public void geosearchstore() {
jedis.geoadd("barcelona", 2.1909389952632d, 41.433791470673d, "place1");
jedis.geoadd("barcelona", 2.1873744593677d, 41.406342043777d, "place2");
jedis.geoadd("barcelona", 2.583333d, 41.316667d, "place3");
// FROMLONLAT and BYRADIUS
long members = jedis.geosearchStore("tel-aviv", "barcelona", new GeoCoordinate(2.191d,41.433d),
1000, GeoUnit.M);
assertEquals(1, members);
List<String> expected = new ArrayList<>();
expected.add("place1");
assertEquals(expected, jedis.zrange("tel-aviv", 0, -1));
members = jedis.geosearchStore("tel-aviv","barcelona", new GeoSearchParam()
.byRadius(3000, GeoUnit.M)
.fromLonLat(new GeoCoordinate(2.191d,41.433d)));
assertEquals(2, members);
assertEquals(2, members);
// FROMMEMBER and BYRADIUS
members = jedis.geosearchStore("tel-aviv", "barcelona","place3", 100, GeoUnit.KM);
assertEquals(3, members);
// FROMMEMBER and BYBOX
members = jedis.geosearchStore("tel-aviv","barcelona","place3", 100, 100, GeoUnit.KM);
assertEquals(3, members);
// FROMLONLAT and BYBOX
members = jedis.geosearchStore("tel-aviv","barcelona", new GeoCoordinate(2.191, 41.433),
1, 1, GeoUnit.KM);
assertEquals(1, members);
}
@Test
public void geosearchstoreWithdist() {
jedis.geoadd("barcelona", 2.1909389952632d, 41.433791470673d, "place1");
jedis.geoadd("barcelona", 2.1873744593677d, 41.406342043777d, "place2");
long members = jedis.geosearchStoreStoreDist("tel-aviv","barcelona", new GeoSearchParam().byRadius(3000, GeoUnit.M)
.fromLonLat(2.191d,41.433d));
assertEquals(2, members);
assertEquals(88.05060698409301, jedis.zscore("tel-aviv", "place1"), 5);
}
private void prepareGeoData() {
Map<String, GeoCoordinate> coordinateMap = new HashMap<>();
coordinateMap.put("a", new GeoCoordinate(3, 4));
coordinateMap.put("b", new GeoCoordinate(2, 3));
coordinateMap.put("c", new GeoCoordinate(3.314, 2.3241));
assertEquals(3, jedis.geoadd("foo", coordinateMap));
Map<byte[], GeoCoordinate> bcoordinateMap = new HashMap<>();
bcoordinateMap.put(bA, new GeoCoordinate(3, 4));
bcoordinateMap.put(bB, new GeoCoordinate(2, 3));
bcoordinateMap.put(bC, new GeoCoordinate(3.314, 2.3241));
assertEquals(3, jedis.geoadd(bfoo, bcoordinateMap));
}
}