Usar o Atlas Search do Java

1
[
2
  {
3
    $match: {
4
      cast: {
5
        $in: ["Keanu Reeves"],
6
      },
7
      genres: {
8
        $all: ["Drama", "Romance"],
9
      },
10
    },
11
  }
12
]

1
[
2
  {
3
    "$search": {
4
      "compound": {
5
        "filter": [
6
          {
7
            "compound": {
8
              "must": [
9
                {
10
                  "text": {
11
                    "query": "Drama",
12
                    "path": "genres"
13
                  }
14
                },
15
                {
16
                  "text": {
17
                    "query": "Romance",
18
                    "path": "genres"
19
                  }
20
                }
21
              ]
22
            }
23
          }
24
        ],
25
        "must": [
26
          {
27
            "phrase": {
28
              "query": "keanu reeves",
29
              "path": {
30
                "value": "cast"
31
              }
32
            }
33
          }
34
        ]
35
      },
36
      "scoreDetails": true
37
    }
38
  },
39
  {
40
    "$project": {
41
      "_id": 0,
42
      "title": 1,
43
      "cast": 1,
44
      "genres": 1,
45
      "score": {
46
        "$meta": "searchScore"
47
      },
48
      "scoreDetails": {
49
        "$meta": "searchScoreDetails"
50
      }
51
    }
52
  },
53
  {
54
    "$limit": 10
55
  }
56
]

1
$ java --version
2
openjdk 17.0.7 2023-04-18
3
OpenJDK Runtime Environment Homebrew (build 17.0.7+0)
4
OpenJDK 64-Bit Server VM Homebrew (build 17.0.7+0, mixed mode, sharing)

1
git clone https://github.com/mongodb-developer/getting-started-search-java
2
cd getting-started-search-java

1
ATLAS_URI="<<insert your connection string here>>" ./gradlew run  

1
Sweet November
2
  Cast: [Keanu Reeves, Charlize Theron, Jason Isaacs, Greg Germann]
3
  Genres: [Drama, Romance]
4
  Score:6.011996746063232
5
6
Something's Gotta Give
7
  Cast: [Jack Nicholson, Diane Keaton, Keanu Reeves, Frances McDormand]
8
  Genres: [Comedy, Drama, Romance]
9
  Score:6.011996746063232
10
11
A Walk in the Clouds
12
  Cast: [Keanu Reeves, Aitana Sènchez-Gijèn, Anthony Quinn, Giancarlo Giannini]
13
  Genres: [Drama, Romance]
14
  Score:5.7239227294921875
15
16
The Lake House
17
  Cast: [Keanu Reeves, Sandra Bullock, Christopher Plummer, Ebon Moss-Bachrach]
18
  Genres: [Drama, Fantasy, Romance]
19
  Score:5.7239227294921875

1
plugins {
2
  id 'java'
3
  id 'application'
4
}
5
6
group 'com.mongodb.atlas'
7
version '1.0-SNAPSHOT'
8
9
repositories {
10
  mavenCentral()
11
}
12
13
dependencies {
14
  implementation 'org.mongodb:mongodb-driver-sync:4.10.1'
15
  implementation 'org.apache.logging.log4j:log4j-slf4j-impl:2.17.1'
16
}
17
18
application {
19
  mainClass = 'com.mongodb.atlas.FirstSearchExample'
20
}

1
// Set ATLAS_URI in your environment
2
String uri = System.getenv("ATLAS_URI");
3
if (uri == null) {
4
  throw new Exception("ATLAS_URI must be specified");
5
}
6
7
MongoClient mongoClient = MongoClients.create(uri);
8
MongoDatabase database = mongoClient.getDatabase("sample_mflix");
9
MongoCollection<Document> collection = database.getCollection("movies");

1
AggregateIterable<Document> aggregationResults = collection.aggregate(Arrays.asList(
2
      searchStage,
3
      project(fields(excludeId(),
4
              include("title", "cast", "genres"),
5
              metaSearchScore("score"),
6
              meta("scoreDetails", "searchScoreDetails"))),
7
      limit(10)));

1
[
2
  {
3
    "$search": { ... }
4
  },
5
  {
6
    "$project": {
7
      "_id": 0,
8
      "title": 1,
9
      "cast": 1,
10
      "genres": 1,
11
      "score": {
12
        "$meta": "searchScore"
13
      },
14
      "scoreDetails": {
15
        "$meta": "searchScoreDetails"
16
      }
17
    }
18
  },
19
  {
20
    "$limit": 10
21
  }
22
]

1
Bson searchStage = search(
2
      compound()
3
              .filter(List.of(genresClause))
4
              .must(List.of(SearchOperator.of(searchQuery))),
5
      searchOptions().option("scoreDetails", true)
6
);

1
"$search": {
2
  "compound": {
3
    "filter": [ . . . ],
4
    "must": [ . . . ]
5
  },
6
  "scoreDetails": true
7
}

1
SearchOperator genresClause = SearchOperator.compound()
2
      .must(Arrays.asList(
3
              SearchOperator.text(fieldPath("genres"),"Drama"),
4
              SearchOperator.text(fieldPath("genres"), "Romance")
5
      ));

1
"compound": {
2
  "must": [
3
    {
4
      "text": {
5
        "query": "Drama",
6
        "path": "genres"
7
      }
8
    },
9
    {
10
      "text": {
11
        "query": "Romance",
12
        "path": "genres"
13
      }
14
    }
15
  ]
16
}

1
Document searchQuery = new Document("phrase",
2
    new Document("query", "keanu reeves")
3
    .append("path", "cast"));

1
aggregationResults.forEach(doc -> {
2
  System.out.println(doc.get("title"));
3
  System.out.println("  Cast: " + doc.get("cast"));
4
  System.out.println("  Genres: " + doc.get("genres"));
5
  System.out.println("  Score:" + doc.get("score"));
6
  // printScoreDetails(2, doc.toBsonDocument().getDocument("scoreDetails"));
7
  System.out.println("");
8
});

1
System.out.println("Explain:");
2
System.out.println(format(aggregationResults.explain().toBsonDocument()));