summaryrefslogtreecommitdiffstats
path: root/vespa-hadoop/src/test/pig/query.pig
diff options
context:
space:
mode:
Diffstat (limited to 'vespa-hadoop/src/test/pig/query.pig')
-rw-r--r--vespa-hadoop/src/test/pig/query.pig18
1 files changed, 18 insertions, 0 deletions
diff --git a/vespa-hadoop/src/test/pig/query.pig b/vespa-hadoop/src/test/pig/query.pig
new file mode 100644
index 00000000000..70f53a992e2
--- /dev/null
+++ b/vespa-hadoop/src/test/pig/query.pig
@@ -0,0 +1,18 @@
+-- REGISTER vespa-hadoop.jar -- Not needed in tests
+
+-- Define Vespa query for retrieving blog posts
+DEFINE BlogPostRecommendations
+ com.yahoo.vespa.hadoop.pig.VespaQuery(
+ 'query=$ENDPOINT/search?query=<userid>&hits=100',
+ 'schema=rank:int,id:chararray,relevance:double,fields/id:chararray,fields/content:chararray'
+ );
+
+-- Load data from a local file
+users = LOAD 'src/test/resources/user_ids.csv' AS (userid:chararray);
+users = FILTER users BY userid IS NOT null;
+
+-- Run a set of queries against Vespa
+recommendations = FOREACH users GENERATE userid, FLATTEN(BlogPostRecommendations(*));
+
+-- Output recommendations
+DUMP recommendations;