Como criar um serviço de pesquisa em Java

1
http://service_host:8080/search?q=purple%20rain&limit=5&skip=0&project=title,genres,plot,year&search=title,plot&filter=genres:Music

1
{
2
 "request": {
3
   "q": "purple rain",
4
   "skip": 0,
5
   "limit": 5,
6
   "search": "title,plot",
7
   "project": "title,genres,plot,year",
8
   "filter": [
9
     "genres:Music"
10
   ]
11
 },
12
 "docs": [
13
   {
14
     "plot": "A young musician, tormented by an abusive situation at home, must contend with a rival singer, a burgeoning romance and his own dissatisfied band as his star begins to rise.",
15
     "genres": [
16
       "Drama",
17
       "Music",
18
       "Musical"
19
     ],
20
     "title": "Purple Rain",
21
     "year": 1984
22
   },
23
   {
24
     "plot": "Graffiti Bridge is the unofficial sequel to Purple Rain. In this movie, The Kid and Morris Day are still competitors and each runs a club of his own. They make a bet about who writes the ...",
25
     "genres": [
26
       "Drama",
27
       "Music",
28
       "Musical"
29
     ],
30
     "title": "Graffiti Bridge",
31
     "year": 1990
32
   }
33
 ],
34
 "meta": [
35
   {
36
     "count": {
37
       "total": 2
38
     }
39
   }
40
 ]
41
}

1
{
2
 "analyzer": "lucene.english",
3
 "searchAnalyzer": "lucene.english",
4
 "mappings": {
5
   "dynamic": true,
6
   "fields": {
7
     "cast": [
8
       {
9
         "type": "token"
10
       },
11
       {
12
         "type": "string"
13
       }
14
     ],
15
     "genres": [
16
       {
17
         "type": "token"
18
       },
19
       {
20
         "type": "string"
21
       }
22
     ]
23
   }
24
 }
25
}

1
public class SearchServlet extends HttpServlet {
2
  private MongoCollection<Document> collection;
3
  private String indexName;
4
5
  private Logger logger;
6
7
  // ...
8
  @Override
9
  protected void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException {
10
    String q = request.getParameter("q");
11
    String searchFieldsValue = request.getParameter("search");
12
    String limitValue = request.getParameter("limit");
13
    String skipValue = request.getParameter("skip");
14
    String projectFieldsValue = request.getParameter("project");
15
    String debugValue = request.getParameter("debug");
16
    String[] filters = request.getParameterMap().get("filter");
17
18
    // ...
19
  }
20
}

1
  @Override
2
  public void init(ServletConfig config) throws ServletException {
3
    super.init(config);
4
5
    logger = Logger.getLogger(config.getServletName());
6
7
    String uri = System.getenv("ATLAS_URI");
8
    if (uri == null) {
9
      throw new ServletException("ATLAS_URI must be specified");
10
    }
11
12
    String databaseName = config.getInitParameter("database");
13
    String collectionName = config.getInitParameter("collection");
14
    indexName = config.getInitParameter("index");
15
16
    MongoClient mongo_client = MongoClients.create(uri);
17
    MongoDatabase database = mongo_client.getDatabase(databaseName);
18
    collection = database.getCollection(collectionName);
19
20
    logger.info("Servlet " + config.getServletName() + " initialized: " + databaseName + " / " + collectionName + " / " + indexName);
21
  }

1
<web-app>
2
   <servlet>
3
       <servlet-name>SearchServlet</servlet-name>
4
       <servlet-class>com.mongodb.atlas.SearchServlet</servlet-class>
5
       <load-on-startup>1</load-on-startup>
6
       <!--
7
         The connection string must be defined in the
8
         `ATLAS_URI` environment variable
9
       -->
10
       <init-param>
11
           <param-name>database</param-name>
12
           <param-value>sample_mflix</param-value>
13
       </init-param>
14
       <init-param>
15
           <param-name>collection</param-name>
16
           <param-value>movies</param-value>
17
       </init-param>
18
       <init-param>
19
           <param-name>index</param-name>
20
           <param-value>movies_index</param-value>
21
       </init-param>
22
   </servlet>
23
24
   <servlet-mapping>
25
       <servlet-name>SearchServlet</servlet-name>
26
       <url-pattern>/search</url-pattern>
27
   </servlet-mapping>
28
</web-app>

1
[
2
 {
3
   "$search": {
4
     "compound": {
5
       "must": [
6
         {
7
           "text": {
8
             "query": "purple rain",
9
             "path": [
10
               "title",
11
               "plot"
12
             ]
13
           }
14
         }
15
       ],
16
       "filter": [
17
         {
18
           "equals": {
19
             "path": "genres",
20
             "value": "Music"
21
           }
22
         }
23
       ]
24
     },
25
     "index": "movies_index",
26
     "count": {
27
       "type": "total"
28
     }
29
   }
30
 },
31
 {
32
   "$facet": {
33
     "docs": [
34
       {
35
         "$skip": 0
36
       },
37
       {
38
         "$limit": 5
39
       },
40
       {
41
         "$project": {
42
           "title": 1,
43
           "genres": 1,
44
           "plot": 1,
45
           "year": 1,
46
           "_id": 0,
47
         }
48
       }
49
     ],
50
     "meta": [
51
       {
52
         "$replaceWith": "$$SEARCH_META"
53
       },
54
       {
55
         "$limit": 1
56
       }
57
     ]
58
   }
59
 }
60
]

1
    // $search
2
    List<SearchPath> searchPath = new ArrayList<>();
3
    for (String search_field : searchFields) {
4
      searchPath.add(SearchPath.fieldPath(search_field));
5
    }
6
7
    CompoundSearchOperator operator = SearchOperator.compound()
8
        .must(List.of(SearchOperator.text(searchPath, List.of(q))));
9
    if (filterOperators.size() > 0)
10
      operator = operator.filter(filterOperators);
11
12
    Bson searchStage = Aggregates.search(
13
        operator,
14
        SearchOptions.searchOptions()
15
            .option("scoreDetails", debug)
16
            .index(indexName)
17
            .count(SearchCount.total())
18
    );

1
    List<String> projectFields = new ArrayList<>();
2
    if (projectFieldsValue != null) {
3
      projectFields.addAll(List.of(projectFieldsValue.split(",")));
4
    }
5
6
    boolean include_id = false;
7
    if (projectFields.contains("_id")) {
8
      include_id = true;
9
      projectFields.remove("_id");
10
    }
11
12
    boolean includeScore = false;
13
    if (projectFields.contains("_score")) {
14
      includeScore = true;
15
      projectFields.remove("_score");
16
    }
17
18
    // ...
19
20
    // $project
21
    List<Bson> projections = new ArrayList<>();
22
    if (projectFieldsValue != null) {
23
      // Don't add _id inclusion or exclusion if no `project` parameter specified
24
      projections.add(Projections.include(projectFields));
25
      if (include_id) {
26
        projections.add(Projections.include("_id"));
27
      } else {
28
        projections.add(Projections.excludeId());
29
      }
30
    }
31
    if (debug) {
32
      projections.add(Projections.meta("_scoreDetails", "searchScoreDetails"));
33
    }
34
    if (includeScore) {
35
      projections.add(Projections.metaSearchScore("_score"));
36
    }

1
    AggregateIterable<Document> aggregationResults = collection.aggregate(List.of(
2
        searchStage,
3
        facetStage
4
    ));
5
6
    Document responseDoc = new Document();
7
    responseDoc.put("request", new Document()
8
        .append("q", q)
9
        .append("skip", skip)
10
        .append("limit", limit)
11
        .append("search", searchFieldsValue)
12
        .append("project", projectFieldsValue)
13
        .append("filter", filters==null ? Collections.EMPTY_LIST : List.of(filters)));
14
15
    if (debug) {
16
      responseDoc.put("debug", aggregationResults.explain().toBsonDocument());
17
    }
18
19
    // When using $facet stage, only one "document" is returned,
20
    // containing the keys specified above: "docs" and "meta"
21
    Document results = aggregationResults.first();
22
    if (results != null) {
23
      for (String s : results.keySet()) {
24
        responseDoc.put(s,results.get(s));
25
      }
26
    }
27
28
    response.setContentType("text/json");
29
    PrintWriter writer = response.getWriter();
30
    writer.println(responseDoc.toJson());
31
    writer.close();
32
33
    logger.info(request.getServletPath() + "?" + request.getQueryString());

1
./gradlew buildProduct