From 43dd2864777f2061469b4b9de3f9924343387d13 Mon Sep 17 00:00:00 2001 From: andy-stark-redis <164213578+andy-stark-redis@users.noreply.github.com> Date: Thu, 26 Sep 2024 14:13:22 +0100 Subject: [PATCH] DOC-4244 added range query examples (#3965) Co-authored-by: M Sazzadul Hoque <7600764+sazzad16@users.noreply.github.com> --- .../io/redis/examples/QueryRangeExample.java | 331 ++++++++++++++++++ 1 file changed, 331 insertions(+) create mode 100644 src/test/java/io/redis/examples/QueryRangeExample.java diff --git a/src/test/java/io/redis/examples/QueryRangeExample.java b/src/test/java/io/redis/examples/QueryRangeExample.java new file mode 100644 index 0000000000..5e4edf4d49 --- /dev/null +++ b/src/test/java/io/redis/examples/QueryRangeExample.java @@ -0,0 +1,331 @@ +// EXAMPLE: query_range +// REMOVE_START +package io.redis.examples; + +import org.junit.Assert; +import org.junit.Test; + +// REMOVE_END +// HIDE_START +import java.util.List; +import redis.clients.jedis.UnifiedJedis; +import redis.clients.jedis.search.*; +import redis.clients.jedis.search.schemafields.*; +import redis.clients.jedis.exceptions.JedisDataException; +import redis.clients.jedis.json.Path2; +import redis.clients.jedis.args.SortingOrder; +// HIDE_END + + + +// HIDE_START +public class QueryRangeExample { + @Test + public void run() { + UnifiedJedis jedis = new UnifiedJedis("redis://localhost:6379"); + + //REMOVE_START + // Clear any keys here before using them in tests. + try { jedis.ftDropIndex("idx:bicycle"); } catch (JedisDataException j) {} + //REMOVE_END + + SchemaField[] schema = { + TextField.of("$.brand").as("brand"), + TextField.of("$.model").as("model"), + TextField.of("$.description").as("description"), + NumericField.of("$.price").as("price"), + TagField.of("$.condition").as("condition") + }; + + jedis.ftCreate("idx:bicycle", + FTCreateParams.createParams() + .on(IndexDataType.JSON) + .addPrefix("bicycle:"), + schema + ); + + String[] bicycleJsons = new String[] { + " {" + + " \"pickup_zone\": \"POLYGON((-74.0610 40.7578, -73.9510 40.7578, -73.9510 40.6678, " + + "-74.0610 40.6678, -74.0610 40.7578))\"," + + " \"store_location\": \"-74.0060,40.7128\"," + + " \"brand\": \"Velorim\"," + + " \"model\": \"Jigger\"," + + " \"price\": 270," + + " \"description\": \"Small and powerful, the Jigger is the best ride for the smallest of tikes! " + + "This is the tiniest kids’ pedal bike on the market available without a coaster brake, the Jigger " + + "is the vehicle of choice for the rare tenacious little rider raring to go.\"," + + " \"condition\": \"new\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((-118.2887 34.0972, -118.1987 34.0972, -118.1987 33.9872, " + + "-118.2887 33.9872, -118.2887 34.0972))\"," + + " \"store_location\": \"-118.2437,34.0522\"," + + " \"brand\": \"Bicyk\"," + + " \"model\": \"Hillcraft\"," + + " \"price\": 1200," + + " \"description\": \"Kids want to ride with as little weight as possible. Especially " + + "on an incline! They may be at the age when a 27.5'' wheel bike is just too clumsy coming " + + "off a 24'' bike. The Hillcraft 26 is just the solution they need!\"," + + " \"condition\": \"used\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((-87.6848 41.9331, -87.5748 41.9331, -87.5748 41.8231, " + + "-87.6848 41.8231, -87.6848 41.9331))\"," + + " \"store_location\": \"-87.6298,41.8781\"," + + " \"brand\": \"Nord\"," + + " \"model\": \"Chook air 5\"," + + " \"price\": 815," + + " \"description\": \"The Chook Air 5 gives kids aged six years and older a durable " + + "and uberlight mountain bike for their first experience on tracks and easy cruising through " + + "forests and fields. The lower top tube makes it easy to mount and dismount in any " + + "situation, giving your kids greater safety on the trails.\"," + + " \"condition\": \"used\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((-80.2433 25.8067, -80.1333 25.8067, -80.1333 25.6967, " + + "-80.2433 25.6967, -80.2433 25.8067))\"," + + " \"store_location\": \"-80.1918,25.7617\"," + + " \"brand\": \"Eva\"," + + " \"model\": \"Eva 291\"," + + " \"price\": 3400," + + " \"description\": \"The sister company to Nord, Eva launched in 2005 as the first " + + "and only women-dedicated bicycle brand. Designed by women for women, allEva bikes " + + "are optimized for the feminine physique using analytics from a body metrics database. " + + "If you like 29ers, try the Eva 291. It’s a brand new bike for 2022.. This " + + "full-suspension, cross-country ride has been designed for velocity. The 291 has " + + "100mm of front and rear travel, a superlight aluminum frame and fast-rolling " + + "29-inch wheels. Yippee!\"," + + " \"condition\": \"used\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((-122.4644 37.8199, -122.3544 37.8199, -122.3544 37.7099, " + + "-122.4644 37.7099, -122.4644 37.8199))\"," + + " \"store_location\": \"-122.4194,37.7749\"," + + " \"brand\": \"Noka Bikes\"," + + " \"model\": \"Kahuna\"," + + " \"price\": 3200," + + " \"description\": \"Whether you want to try your hand at XC racing or are looking " + + "for a lively trail bike that's just as inspiring on the climbs as it is over rougher " + + "ground, the Wilder is one heck of a bike built specifically for short women. Both the " + + "frames and components have been tweaked to include a women’s saddle, different bars " + + "and unique colourway.\"," + + " \"condition\": \"used\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((-0.1778 51.5524, 0.0822 51.5524, 0.0822 51.4024, " + + "-0.1778 51.4024, -0.1778 51.5524))\"," + + " \"store_location\": \"-0.1278,51.5074\"," + + " \"brand\": \"Breakout\"," + + " \"model\": \"XBN 2.1 Alloy\"," + + " \"price\": 810," + + " \"description\": \"The XBN 2.1 Alloy is our entry-level road bike – but that’s " + + "not to say that it’s a basic machine. With an internal weld aluminium frame, a full " + + "carbon fork, and the slick-shifting Claris gears from Shimano’s, this is a bike which " + + "doesn’t break the bank and delivers craved performance.\"," + + " \"condition\": \"new\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((2.1767 48.9016, 2.5267 48.9016, 2.5267 48.5516, " + + "2.1767 48.5516, 2.1767 48.9016))\"," + + " \"store_location\": \"2.3522,48.8566\"," + + " \"brand\": \"ScramBikes\"," + + " \"model\": \"WattBike\"," + + " \"price\": 2300," + + " \"description\": \"The WattBike is the best e-bike for people who still " + + "feel young at heart. It has a Bafang 1000W mid-drive system and a 48V 17.5AH " + + "Samsung Lithium-Ion battery, allowing you to ride for more than 60 miles on one " + + "charge. It’s great for tackling hilly terrain or if you just fancy a more " + + "leisurely ride. With three working modes, you can choose between E-bike, " + + "assisted bicycle, and normal bike modes.\"," + + " \"condition\": \"new\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((13.3260 52.5700, 13.6550 52.5700, 13.6550 52.2700, " + + "13.3260 52.2700, 13.3260 52.5700))\"," + + " \"store_location\": \"13.4050,52.5200\"," + + " \"brand\": \"Peaknetic\"," + + " \"model\": \"Secto\"," + + " \"price\": 430," + + " \"description\": \"If you struggle with stiff fingers or a kinked neck or " + + "back after a few minutes on the road, this lightweight, aluminum bike alleviates " + + "those issues and allows you to enjoy the ride. From the ergonomic grips to the " + + "lumbar-supporting seat position, the Roll Low-Entry offers incredible comfort. " + + "The rear-inclined seat tube facilitates stability by allowing you to put a foot " + + "on the ground to balance at a stop, and the low step-over frame makes it " + + "accessible for all ability and mobility levels. The saddle is very soft, with " + + "a wide back to support your hip joints and a cutout in the center to redistribute " + + "that pressure. Rim brakes deliver satisfactory braking control, and the wide tires " + + "provide a smooth, stable ride on paved roads and gravel. Rack and fender mounts " + + "facilitate setting up the Roll Low-Entry as your preferred commuter, and the " + + "BMX-like handlebar offers space for mounting a flashlight, bell, or phone holder.\"," + + " \"condition\": \"new\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((1.9450 41.4301, 2.4018 41.4301, 2.4018 41.1987, " + + "1.9450 41.1987, 1.9450 41.4301))\"," + + " \"store_location\": \"2.1734, 41.3851\"," + + " \"brand\": \"nHill\"," + + " \"model\": \"Summit\"," + + " \"price\": 1200," + + " \"description\": \"This budget mountain bike from nHill performs well both " + + "on bike paths and on the trail. The fork with 100mm of travel absorbs rough " + + "terrain. Fat Kenda Booster tires give you grip in corners and on wet trails. " + + "The Shimano Tourney drivetrain offered enough gears for finding a comfortable " + + "pace to ride uphill, and the Tektro hydraulic disc brakes break smoothly. " + + "Whether you want an affordable bike that you can take to work, but also take " + + "trail in mountains on the weekends or you’re just after a stable, comfortable " + + "ride for the bike path, the Summit gives a good value for money.\"," + + " \"condition\": \"new\"" + + " }", + + " {" + + " \"pickup_zone\": \"POLYGON((12.4464 42.1028, 12.5464 42.1028, " + + "12.5464 41.7028, 12.4464 41.7028, 12.4464 42.1028))\"," + + " \"store_location\": \"12.4964,41.9028\"," + + " \"model\": \"ThrillCycle\"," + + " \"brand\": \"BikeShind\"," + + " \"price\": 815," + + " \"description\": \"An artsy, retro-inspired bicycle that’s as " + + "functional as it is pretty: The ThrillCycle steel frame offers a smooth ride. " + + "A 9-speed drivetrain has enough gears for coasting in the city, but we wouldn’t " + + "suggest taking it to the mountains. Fenders protect you from mud, and a rear " + + "basket lets you transport groceries, flowers and books. The ThrillCycle comes " + + "with a limited lifetime warranty, so this little guy will last you long " + + "past graduation.\"," + + " \"condition\": \"refurbished\"" + + " }" + }; + + for (int i = 0; i < bicycleJsons.length; i++) { + jedis.jsonSet("bicycle:" + i, Path2.ROOT_PATH, bicycleJsons[i]); + } +// HIDE_END + + + // STEP_START range1 + SearchResult res1 = jedis.ftSearch("idx:bicycle", "@price:[500 1000]"); + System.out.println(res1.getTotalResults()); // >>> 3 + + List docs1 = res1.getDocuments(); + + for (int i = 0; i < docs1.size(); i++) { + System.out.println(docs1.get(i).getId()); + } + // >>> bicycle:2 + // >>> bicycle:5 + // >>> bicycle:9 + // STEP_END + + // Tests for 'range1' step. + // REMOVE_START + Assert.assertEquals(3, res1.getTotalResults()); + Assert.assertEquals("bicycle:2", docs1.get(0).getId()); + Assert.assertEquals("bicycle:5", docs1.get(1).getId()); + Assert.assertEquals("bicycle:9", docs1.get(2).getId()); + // REMOVE_END + + + // STEP_START range2 + SearchResult res2 = jedis.ftSearch("idx:bicycle", + "*", + FTSearchParams.searchParams() + .filter("price", 500, 1000) + ); + System.out.println(res2.getTotalResults()); // >>> 3 + + List docs2 = res2.getDocuments(); + + for (int i = 0; i < docs2.size(); i++) { + System.out.println(docs2.get(i).getId()); + } + // >>> bicycle:2 + // >>> bicycle:5 + // >>> bicycle:9 + // STEP_END + + // Tests for 'range2' step. + // REMOVE_START + Assert.assertEquals(3, res2.getTotalResults()); + Assert.assertEquals("bicycle:2", docs2.get(0).getId()); + Assert.assertEquals("bicycle:5", docs2.get(1).getId()); + Assert.assertEquals("bicycle:9", docs2.get(2).getId()); + // REMOVE_END + + + // STEP_START range3 + SearchResult res3 = jedis.ftSearch("idx:bicycle", + "*", + FTSearchParams.searchParams() + .filter("price", 1000, true, Double.POSITIVE_INFINITY, false) + ); + System.out.println(res3.getTotalResults()); // >>> 5 + + List docs3 = res3.getDocuments(); + + for (int i = 0; i < docs3.size(); i++) { + System.out.println(docs3.get(i).getId()); + } + // >>> bicycle:1 + // >>> bicycle:4 + // >>> bicycle:6 + // >>> bicycle:3 + // >>> bicycle:8 + // STEP_END + + // Tests for 'range3' step. + // REMOVE_START + Assert.assertEquals(5, res3.getTotalResults()); + Assert.assertEquals("bicycle:1", docs3.get(0).getId()); + Assert.assertEquals("bicycle:4", docs3.get(1).getId()); + Assert.assertEquals("bicycle:6", docs3.get(2).getId()); + Assert.assertEquals("bicycle:3", docs3.get(3).getId()); + Assert.assertEquals("bicycle:8", docs3.get(4).getId()); + // REMOVE_END + + + // STEP_START range4 + SearchResult res4 = jedis.ftSearch("idx:bicycle", + "@price:[-inf 2000]", + FTSearchParams.searchParams() + .sortBy("price", SortingOrder.ASC) + .limit(0, 5) + ); + System.out.println(res4.getTotalResults()); // >>> 7 + + List docs4 = res4.getDocuments(); + + for (int i = 0; i < docs4.size(); i++) { + System.out.println(docs4.get(i).getId()); + } + // >>> bicycle:0 + // >>> bicycle:7 + // >>> bicycle:5 + // >>> bicycle:2 + // >>> bicycle:9 + // STEP_END + + // Tests for 'range4' step. + // REMOVE_START + Assert.assertEquals(7, res4.getTotalResults()); + Assert.assertEquals("bicycle:0", docs4.get(0).getId()); + Assert.assertEquals("bicycle:7", docs4.get(1).getId()); + Assert.assertEquals("bicycle:5", docs4.get(2).getId()); + Assert.assertEquals("bicycle:2", docs4.get(3).getId()); + Assert.assertEquals("bicycle:9", docs4.get(4).getId()); + // REMOVE_END + +// HIDE_START + jedis.close(); + } +} +// HIDE_END +