Search APIの「Faceted Search」を使ってみた

// import com.google.appengine.api.search.Facet;

IndexSpec indexSpec = IndexSpec.newBuilder().setName("FacetedSearch").build();
Index index = SearchServiceFactory.getSearchService().getIndex(indexSpec);

Builder builder = Document.newBuilder().setId("1").setLocale(Locale.ENGLISH);
Field.Builder textFieldBuilder = Field.newBuilder();
textFieldBuilder.setName("name").setText("テスト太郎1");
builder.addField(textFieldBuilder);
Field.Builder atomFieldBuilder = Field.newBuilder();
atomFieldBuilder.setName("bloodType").setAtom("A");
builder.addField(atomFieldBuilder);
Field.Builder numFieldBuilder = Field.newBuilder();
numFieldBuilder.setName("age").setNumber(1);
builder.addField(numFieldBuilder);

// ↓↓↓↓↓↓ここからFaceted Searchの登録処理↓↓↓↓↓↓
// Faceted Searchする項目と値を登録する。Faceted Searchとして登録できる型はATOM型とNUMBER型の2種類だけです。
builder.addFacet(Facet.withAtom("bloodType", "A")); // ATOM型として、項目名「bloodType」と値「A」を設定
builder.addFacet(Facet.withNumber("age", Double.valueOf(1))); // NUMBER型として、項目名「age」と値「1」を設定
// ↑↑↑↑↑↑ここまでFaceted Searchの登録処理↑↑↑↑↑↑

index.put(builder);

// import com.google.appengine.api.search.FacetRange;
// import com.google.appengine.api.search.FacetRequest;
// import com.google.appengine.api.search.Query.Builder;

Builder queryBuilder = Query.newBuilder();

// ATOM型の値をFaceted Searchする場合は、以下のように指定します。
// 下記では「bloodType」というFacetの名前に対して、A,B,O,ABがそれぞれ何件あるのか集計を実行します。
queryBuilder.addReturnFacet(FacetRequest.newBuilder()
    .setName("bloodType").addValueConstraint("A")
    .addValueConstraint("B").addValueConstraint("O")
    .addValueConstraint("AB"));

// 数値型の値をFaceted Searchする場合は、以下のように指定します。
// FacetRange.withEndは指定した引数の値未満が対象。
// FacetRange.withStartは指定した引数の値以上が対象。
// FacetRange.withStartEndは第一引数の値以上、第二引数の値未満が対象。
// 下記では「age」というFacetの名前に対して、集計を実行します。
queryBuilder.addReturnFacet(FacetRequest.newBuilder().setName("age")
    .addRange(FacetRange.withEnd(10.0))// 10歳未満の件数をカウント
    .addRange(FacetRange.withStartEnd(10.0, 20.0))// 10歳以上20歳未満の件数をカウント
    .addRange(FacetRange.withStartEnd(20.0, 30.0))// 20歳以上30歳未満の件数をカウント
    .addRange(FacetRange.withStartEnd(40.0, 50.0))// 40歳以上50歳未満の件数をカウント
    .addRange(FacetRange.withStart(50.0)));// 50歳以上の件数をカウント

// 検索実行
IndexSpec indexSpec = IndexSpec.newBuilder().setName("FacetedSearch").build();
Index index = SearchServiceFactory.getSearchService().getIndex(indexSpec);
Results<ScoredDocument> results = index.search(queryBuilder.build(""));

// 検索の実行結果から、getFacetsでFaceted Searchの結果をFacet単位で取得することができます。
Collection<FacetResult> facets = results.getFacets();
System.out.println("total count=" + results.getNumberFound()); // 検索結果の合計件数を出力
// Faceted Searchの結果を出力
for (FacetResult facet : facets) {
    // Facetの名前を出力
    // FacetRequest.newBuilder().setNameに設定した値が出力されます。
    System.out.println("FacetResult name=" + facet.getName());
    List<FacetResultValue> values = facet.getValues();
    for (FacetResultValue value : values) {
        // FacetResultValue.getLabelでは、
        // FacetRequest.newBuilder().addValueConstraint、
        // もしくは、FacetRequest.newBuilder().addRangeで設定した集計条件が出力されます。
        // FacetResultValue.getCountでは、集計件数が出力されます。
        System.out.println("FacetResultValue label=" + value.getLabel() + "/ count=" + value.getCount());
    }
}

total count=50
FacetResult name=age
FacetResultValue label=[-Infinity,10.0)/ count=9
FacetResultValue label=[10.0,20.0)/ count=10
FacetResultValue label=[20.0,30.0)/ count=10
FacetResultValue label=[30.0,40.0)/ count=10
FacetResultValue label=[40.0,50.0)/ count=10
FacetResultValue label=[50.0,Infinity)/ count=1
FacetResult name=bloodType
FacetResultValue label=AB/ count=15
FacetResultValue label=A/ count=14
FacetResultValue label=O/ count=11
FacetResultValue label=B/ count=10