feat(location): implement list by location

floriaaan · Aug 19, 2023 · f69b814 · f69b814
1 parent 20f4e3a
commit f69b814
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 7 deletions.
diff --git a/services/restaurant/package.json b/services/restaurant/package.json
@@ -21,6 +21,7 @@
 		"amqplib": "^0.10.3",
 		"dotenv": "^16.3.1",
 		"esbuild": "^0.18.15",
+		"geolib": "^3.3.4",
 		"nodemon": "^3.0.1",
 		"path": "^0.12.7",
 		"ts-node": "^10.9.1",

diff --git a/services/restaurant/pnpm-lock.yaml b/services/restaurant/pnpm-lock.yaml
diff --git a/services/restaurant/src/handler/list-by-location.ts b/services/restaurant/src/handler/list-by-location.ts
@@ -6,6 +6,23 @@ import {
   Restaurant,
   RestaurantList,
 } from "@restaurant/types/restaurant";
+import geolib from "geolib";
+
+interface RestaurantWithDistance extends Restaurant {
+  distance: number;
+}
+
+export const sortRestaurantsByDistance = (
+  restaurants: Restaurant[],
+  location: [number, number]
+): RestaurantWithDistance[] => {
+  return restaurants
+    .map((restaurant) => ({
+      ...restaurant,
+      distance: geolib.getDistance(location, restaurant.location),
+    }))
+    .sort((a, b) => a.distance - b.distance);
+};
 
 export const GetRestaurantsByLocation = async (
   { request }: Data<ByLocationInput>,
@@ -14,13 +31,11 @@ export const GetRestaurantsByLocation = async (
   try {
     const { location } = request;
 
-    const restaurants = (await prisma.restaurant.findMany({
-      where: {
-        location: {
-          equals: location,
-        },
-      },
-    })) as unknown as Restaurant[];
+    const all = (await prisma.restaurant.findMany()) as unknown as Restaurant[];
+
+    const restaurants = sortRestaurantsByDistance(all, location)
+      .map(({ distance, ...restaurant }) => restaurant as Restaurant)
+      .filter((_, i) => i < 10);
 
     callback(null, { restaurants });
   } catch (error) {