Redis geo 实战：“附近的人”实现，打造社交的新维度

import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType;
import javax.persistence.Id;


@Entity
public class User {


    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String username;
    private double latitude;
    private double longitude;


    // Getters and setters
}

import org.springframework.data.jpa.repository.JpaRepository;


public interface UserRepository extends JpaRepository<User, Long> {
}

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;


@Service
public class UserService {


    private final UserRepository userRepository;


    @Autowired
    public UserService(UserRepository userRepository) {
        this.userRepository = userRepository;
    }


    public User saveUser(User user) {
        return userRepository.save(user);
    }


    public List<User> findNearbyUsers(double latitude, double longitude, double radiusInMeters) {
        // 查询附近的用户逻辑
        // 这里简化处理，实际应用中需要更精确的距离计算
        List<User> allUsers = userRepository.findAll();
        return allUsers.stream()
                .filter(user -> isWithinRadius(user, latitude, longitude, radiusInMeters))
                .collect(Collectors.toList());
    }


    private boolean isWithinRadius(User user, double centerLatitude, double centerLongitude, double radiusInMeters) {
        double earthRadius = 6371; // Earth's radius in kilometers
        double dLat = Math.toRadians(user.getLatitude() - centerLatitude);
        double dLon = Math.toRadians(user.getLongitude() - centerLongitude);
        double a = Math.sin(dLat  2) * Math.sin(dLat  2) +
                   Math.cos(Math.toRadians(centerLatitude)) * Math.cos(Math.toRadians(user.getLatitude())) *
                   Math.sin(dLon  2) * Math.sin(dLon  2);
        double c = 2 * Math.atan2(Math.sqrt(a), Math.sqrt(1 - a));
        double distance = earthRadius * c * 1000; // Convert to meters
        return distance <= radiusInMeters;
    }
}

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;


import java.util.List;


@RestController
@RequestMapping("/api/users")
public class UserController {


    private final UserService userService;


    @Autowired
    public UserController(UserService userService) {
        this.userService = userService;
    }


    @PostMapping
    public User addUser(@RequestBody User user) {
        return userService.saveUser(user);
    }


    @GetMapping("/nearby")
    public List<User> getNearbyUsers(@RequestParam double latitude, @RequestParam double longitude, @RequestParam double radiusInMeters) {
        return userService.findNearbyUsers(latitude, longitude, radiusInMeters);
    }
}

# 假设用户的 ID 是 12345，经度是 -122.4194，纬度是 37.7749（旧金山的位置）
GEOADD geolocation user1 -122.4194 37.7749

# 假设用户想知道他们周围 5 公里内的人
# 经度和纬度是用户的当前位置
GEORADIUS geolocation -122.4194 37.7749 5 km WITHDIST WITHCOORD

import redis


# 连接到 Redis
r = redis.Redis(host='localhost', port=6379, db=0)


def add_user_location(user_id, longitude, latitude):
    """
    添加用户的地理位置信息到 Redis。
    """
    r.geoadd('geolocation', longitude, latitude, user_id)


def find_nearby_users(user_longitude, user_latitude, radius):
    """
    查找给定半径内的用户。
    """
    nearby_users = r.georadius(
        'geolocation',
        user_longitude,
        user_latitude,
        radius,
        unit='km',
        withdist=True,
        withcoord=True
    )
    return nearby_users


# 示例：添加用户
add_user_location(user1, -122.4194, 37.7749)
add_user_location(user2, -122.4079, 37.7797)  # 另一个用户的坐标


# 示例：查找附近的人
nearby_users = find_nearby_users(-122.4194, 37.7749, 5)
for user in nearby_users:
    print(f"User ID: {user[0]}, Distance: {user[1]} km, Coordinates: ({user[2][0]}, {user[2][1]})")

<dependencies>
    <dependency>
        <groupId>redis.clients</groupId>
        <artifactId>jedis</artifactId>
        <version>3.10.0</version>
    </dependency>
</dependencies>

import redis.clients.jedis.Jedis;
import java.util.List;
import java.util.Map.Entry;


public class GeoSearchExample {


    private static final String GEOLOCATION_KEY = "geolocation";


    public static void main(String[] args) {
        // 连接到 Redis
        try (Jedis jedis = new Jedis("localhost", 6379)) {
            // 添加用户的地理位置信息到 Redis
            addUserLocation(jedis, user1, -122.4194, 37.7749);
            addUserLocation(jedis, user2, -122.4079, 37.7797);  // 另一个用户的坐标


            // 查找附近的人
            List<Entry<String, Double>> nearbyUsers = findNearbyUsers(jedis, -122.4194, 37.7749, 5);


            // 输出结果
            for (Entry<String, Double> user : nearbyUsers) {
                System.out.println("User ID: " + user.getKey() + ", Distance: " + user.getValue() + " km");
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    public static void addUserLocation(Jedis jedis, long userId, double longitude, double latitude) {
        // 添加用户的地理位置信息到 Redis。
        jedis.geoadd(GEOLOCATION_KEY, longitude, latitude, String.valueOf(userId));
    }


    public static List<Entry<String, Double>> findNearbyUsers(Jedis jedis, double longitude, double latitude, double radius) {
        // 查找给定半径内的用户。
        return jedis.georadius(GEOLOCATION_KEY, longitude, latitude, radius, "km", "WITHDIST");
    }
}