SpringBoot集成redis的LBS功能

下面的代碼實現了添加經緯度數據 和 搜索經緯度數據的功能：

import java.util.List;

import com.longge.goods.dto.BuildingDto;
import com.longge.goods.dto.BuildingLbsDto;

/**
 * LBS相關的服務
 * @author yangzhilong
 *
 */
public interface LbsService {
    /**
     * 新增或者修改樓宇的經緯度
     * @param buildDto
     */
    void saveBuildingLonAndLat(BuildingDto buildDto);
    
    /**
     * 查詢指定經緯度附近的樓宇
     * @param lon 經度
     * @param lat 緯度
     * @param limit 記錄數
     * @return
     */
    List<BuildingLbsDto> listNearbyBuilding(double lon, double lat, int limit);
    
}

 

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.Resource;

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.geo.Circle;
import org.springframework.data.geo.Distance;
import org.springframework.data.geo.GeoResults;
import org.springframework.data.geo.Metrics;
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.core.GeoOperations;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import com.alibaba.fastjson.JSONObject;
import com.longge.core.util.BeanMapper;
import com.longge.goods.api.BuildingService;
import com.longge.goods.api.LbsService;
import com.longge.goods.constants.BuildingConstant;
import com.longge.goods.dto.BuildingDto;
import com.longge.goods.dto.BuildingLbsDto;
import com.longge.goods.util.GoodsRedisKeyUtils;

import lombok.extern.slf4j.Slf4j;

@Service
@Slf4j
public class LbsServiceImpl implements LbsService {
    @Autowired
    private BuildingService buildingService;
    // 搜索范圍
    @Value("${lbs.distance:3}")
    private Double lbsDistance;
    @Resource
    private StringRedisTemplate redisTemplate;

    @Override
    public void saveBuildingLonAndLat(BuildingDto buildDto) {
        log.info("開始處理樓宇的經緯度數據,樓宇：{}", JSONObject.toJSONString(buildDto));
        GeoOperations<String, String> ops = redisTemplate.opsForGeo();
        String member = String.valueOf(buildDto.getId());
        // 查詢樓宇的geo數據是否存在
        List<String> list = ops.hash(GoodsRedisKeyUtils.getBuildingLbsKey(), member);
        if (!CollectionUtils.isEmpty(list)) {
            log.info("該樓宇數據存在，先進行刪除操作");
            // 存在，先刪除，后新增
            ops.remove(GoodsRedisKeyUtils.getBuildingLbsKey(), member);
        }
        ops.add(GoodsRedisKeyUtils.getBuildingLbsKey(),
                new RedisGeoCommands.GeoLocation<String>(member, new Point(buildDto.getLon(), buildDto.getLat())));
        log.info("樓宇經緯度信息處理完成");
    }

    @Override
    public List<BuildingLbsDto> listNearbyBuilding(double lon, double lat, int limit) {
        log.info("查詢LBS樓宇信息，經度：{}，緯度：{}，條數：{}", lon, lat, limit);
        GeoOperations<String, String> ops = redisTemplate.opsForGeo();
        
        // 獲取redis里全局的配置，實現參數動態化
        String redisLbsDistance = (String)RedisCache.hGet("config:global", "lbsDistance");
        if(StringUtils.isNotBlank(redisLbsDistance)) {
            lbsDistance = Double.valueOf(redisLbsDistance);
        }
        
        Point center = new Point(lon, lat);
        Distance radius = new Distance(lbsDistance, Metrics.KILOMETERS);
        Circle within = new Circle(center, radius);
        // order by 距離 limit 20 ，同時返回距離中心點的距離
        GeoRadiusCommandArgs args = GeoRadiusCommandArgs.newGeoRadiusArgs().includeDistance().limit(limit).sortAscending();
        
        GeoResults<GeoLocation<String>> result = ops.radius(GoodsRedisKeyUtils.getBuildingLbsKey(), within, args);
        log.info("查詢redis后的經緯度數據：{}", JSONObject.toJSONString(result));
        if(null != result) {
            Map<Long, Double> data = new HashMap<>();
            List<Long> buildings = new ArrayList<>();
            
            result.forEach(item -> {
                data.put(Long.valueOf(item.getContent().getName()), item.getDistance().getValue());
                buildings.add(Long.valueOf(item.getContent().getName()));
            });
            if(!CollectionUtils.isEmpty(buildings)) {
                List<BuildingDto> buildingList = buildingService.getBuildList(buildings);
                if(!CollectionUtils.isEmpty(buildingList)) {
                    List<BuildingLbsDto> resp = new ArrayList<>();
                    buildingList.stream().forEach(item -> {
                        BuildingLbsDto dto = BeanMapper.map(item, BuildingLbsDto.class);
                        if(BuildingConstant.BuildingStatusEnum.OPENED.getValue() == dto.getStatus()) {
                            dto.setDistance(data.get(item.getId()));
                            resp.add(dto);
                        }
                    });
                    log.info("查詢LBS樓宇信息的結果：{}", JSONObject.toJSONString(resp));
                    return resp;
                }
            }
        }
        log.warn("未查詢LBS樓宇信息");
        return null;
    }
}

import java.util.Date;

/**
 * 樓宇信息dto
 *
 */
@Data
public class BuildingDto {
    private Long id;

    private Short status;
    /**
     * 緯度
     */
    private Double lat;
    /**
     * 經度
     */
    private Double lon;
}

import java.io.Serializable;

import lombok.Getter;
import lombok.Setter;

/**
 * 包含距離定位點距離的樓宇dto
 * @author yangzhilong
 *
 */
@Getter
@Setter
public class BuildingLbsDto extends BuildingDto implements Serializable{
    /**
     * 
     */
    private static final long serialVersionUID = 3224249189169148012L;
    // 距離，單位：千米
    private Double distance;
}