Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.springframework.data.redis.core.DefaultGeoOperations Maven / Gradle / Ivy
/*
* Copyright 2016-2022 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
*
* https://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 org.springframework.data.redis.core;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import org.springframework.data.geo.Circle;
import org.springframework.data.geo.Distance;
import org.springframework.data.geo.GeoResults;
import org.springframework.data.geo.Metric;
import org.springframework.data.geo.Point;
import org.springframework.data.redis.connection.RedisGeoCommands;
import org.springframework.data.redis.connection.RedisGeoCommands.GeoLocation;
import org.springframework.data.redis.connection.RedisGeoCommands.GeoRadiusCommandArgs;
import org.springframework.data.redis.domain.geo.GeoReference;
import org.springframework.data.redis.domain.geo.GeoReference.GeoMemberReference;
import org.springframework.data.redis.domain.geo.GeoShape;
/**
* Default implementation of {@link GeoOperations}.
*
* @author Ninad Divadkar
* @author Christoph Strobl
* @author Mark Paluch
* @since 1.8
*/
class DefaultGeoOperations extends AbstractOperations implements GeoOperations {
/**
* Creates new {@link DefaultGeoOperations}.
*
* @param template must not be {@literal null}.
*/
DefaultGeoOperations(RedisTemplate template) {
super(template);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#add(java.lang.Object, org.springframework.data.geo.Point, java.lang.Object)
*/
@Override
public Long add(K key, Point point, M member) {
byte[] rawKey = rawKey(key);
byte[] rawMember = rawValue(member);
return execute(connection -> connection.geoAdd(rawKey, point, rawMember));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#add(java.lang.Object, org.springframework.data.redis.connection.RedisGeoCommands.GeoLocation)
*/
@Override
public Long add(K key, GeoLocation location) {
return add(key, location.getPoint(), location.getName());
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#add(java.lang.Object, java.util.Map)
*/
@Override
public Long add(K key, Map memberCoordinateMap) {
byte[] rawKey = rawKey(key);
Map rawMemberCoordinateMap = new HashMap<>();
for (M member : memberCoordinateMap.keySet()) {
byte[] rawMember = rawValue(member);
rawMemberCoordinateMap.put(rawMember, memberCoordinateMap.get(member));
}
return execute(connection -> connection.geoAdd(rawKey, rawMemberCoordinateMap));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#add(java.lang.Object, java.lang.Iterable)
*/
@Override
public Long add(K key, Iterable> locations) {
Map memberCoordinateMap = new LinkedHashMap<>();
for (GeoLocation location : locations) {
memberCoordinateMap.put(location.getName(), location.getPoint());
}
return add(key, memberCoordinateMap);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#distance(java.lang.Object, java.lang.Object, java.lang.Object)
*/
@Override
public Distance distance(K key, M member1, M member2) {
byte[] rawKey = rawKey(key);
byte[] rawMember1 = rawValue(member1);
byte[] rawMember2 = rawValue(member2);
return execute(connection -> connection.geoDist(rawKey, rawMember1, rawMember2));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#distance(java.lang.Object, java.lang.Object, java.lang.Object, org.springframework.data.geo.Metric)
*/
@Override
public Distance distance(K key, M member1, M member2, Metric metric) {
byte[] rawKey = rawKey(key);
byte[] rawMember1 = rawValue(member1);
byte[] rawMember2 = rawValue(member2);
return execute(connection -> connection.geoDist(rawKey, rawMember1, rawMember2, metric));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#hash(java.lang.Object, java.lang.Object[])
*/
@Override
public List hash(K key, M... members) {
byte[] rawKey = rawKey(key);
byte[][] rawMembers = rawValues(members);
return execute(connection -> connection.geoHash(rawKey, rawMembers));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#position(java.lang.Object, java.lang.Object[])
*/
@Override
public List position(K key, M... members) {
byte[] rawKey = rawKey(key);
byte[][] rawMembers = rawValues(members);
return execute(connection -> connection.geoPos(rawKey, rawMembers));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#radius(java.lang.Object, org.springframework.data.geo.Circle)
*/
@Override
public GeoResults> radius(K key, Circle within) {
byte[] rawKey = rawKey(key);
GeoResults> raw = execute(connection -> connection.geoRadius(rawKey, within));
return deserializeGeoResults(raw);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#radius(java.lang.Object, org.springframework.data.geo.Circle, org.springframework.data.redis.core.GeoRadiusCommandArgs)
*/
@Override
public GeoResults> radius(K key, Circle within, GeoRadiusCommandArgs args) {
byte[] rawKey = rawKey(key);
GeoResults> raw = execute(connection -> connection.geoRadius(rawKey, within, args));
return deserializeGeoResults(raw);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#radius(java.lang.Object, java.lang.Object, double)
*/
@Override
public GeoResults> radius(K key, M member, double radius) {
byte[] rawKey = rawKey(key);
byte[] rawMember = rawValue(member);
GeoResults> raw = execute(
connection -> connection.geoRadiusByMember(rawKey, rawMember, radius));
return deserializeGeoResults(raw);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#radius(java.lang.Object, java.lang.Object, org.springframework.data.geo.Distance)
*/
@Override
public GeoResults> radius(K key, M member, Distance distance) {
byte[] rawKey = rawKey(key);
byte[] rawMember = rawValue(member);
GeoResults> raw = execute(
connection -> connection.geoRadiusByMember(rawKey, rawMember, distance));
return deserializeGeoResults(raw);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#radius(java.lang.Object, java.lang.Object, double, org.springframework.data.geo.Metric, org.springframework.data.redis.core.GeoRadiusCommandArgs)
*/
@Override
public GeoResults> radius(K key, M member, Distance distance, GeoRadiusCommandArgs param) {
byte[] rawKey = rawKey(key);
byte[] rawMember = rawValue(member);
GeoResults> raw = execute(
connection -> connection.geoRadiusByMember(rawKey, rawMember, distance, param));
return deserializeGeoResults(raw);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#remove(java.lang.Object, java.lang.Object[])
*/
@Override
public Long remove(K key, M... members) {
byte[] rawKey = rawKey(key);
byte[][] rawMembers = rawValues(members);
return execute(connection -> connection.zRem(rawKey, rawMembers));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#search(java.lang.Object, org.springframework.data.redis.connection.RedisGeoCommands.GeoReference, org.springframework.data.redis.connection.RedisGeoCommands.GeoShape, org.springframework.data.redis.connection.RedisGeoCommands.GeoSearchCommandArgs)
*/
@Override
public GeoResults> search(K key, GeoReference reference,
GeoShape geoPredicate, RedisGeoCommands.GeoSearchCommandArgs args) {
byte[] rawKey = rawKey(key);
GeoReference rawMember = getGeoReference(reference);
GeoResults> raw = execute(
connection -> connection.geoSearch(rawKey, rawMember, geoPredicate, args));
return deserializeGeoResults(raw);
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.GeoOperations#searchAndStore(java.lang.Object, java.lang.Object, org.springframework.data.redis.connection.RedisGeoCommands.GeoReference, org.springframework.data.redis.connection.RedisGeoCommands.GeoShape, org.springframework.data.redis.connection.RedisGeoCommands.GeoSearchStoreCommandArgs)
*/
@Override
public Long searchAndStore(K key, K destKey, GeoReference reference,
GeoShape geoPredicate, RedisGeoCommands.GeoSearchStoreCommandArgs args) {
byte[] rawKey = rawKey(key);
byte[] rawDestKey = rawKey(destKey);
GeoReference rawMember = getGeoReference(reference);
return execute(connection -> connection.geoSearchStore(rawDestKey, rawKey, rawMember, geoPredicate, args));
}
@SuppressWarnings("unchecked")
private GeoReference getGeoReference(GeoReference reference) {
return reference instanceof GeoReference.GeoMemberReference
? GeoReference
.fromMember(rawValue(((GeoMemberReference) reference).getMember()))
: (GeoReference) reference;
}
}