From 477469c6a2698c5d9b6f0dc074e10130c50d5e88 Mon Sep 17 00:00:00 2001
From: Jeremiah Buddenhagen <git@bitspill.net>
Date: Tue, 26 Feb 2019 15:53:33 -0800
Subject: [PATCH] add artifact field cardinality api endpoint

---
 httpapi/artifacts.go | 40 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/httpapi/artifacts.go b/httpapi/artifacts.go
index 82510c2..454e3a3 100644
--- a/httpapi/artifacts.go
+++ b/httpapi/artifacts.go
@@ -1,10 +1,13 @@
 package httpapi
 
 import (
+	"context"
 	"net/http"
 	"net/url"
 
+	"github.com/bitspill/oip/datastore"
 	"github.com/gorilla/mux"
+	"github.com/pkg/errors"
 	"gopkg.in/olivere/elastic.v6"
 )
 
@@ -13,7 +16,7 @@ func init() {
 	rootRouter.HandleFunc("/artifact/get/latest", handleLatest)
 	rootRouter.HandleFunc("/artifact/get/{id:[a-f0-9]+}", handleGet)
 	rootRouter.HandleFunc("/artifact/search", handleArtifactSearch).Queries("q", "{query}")
-	rootRouter.HandleFunc("/artifact/search", handleArtifactSearch).Queries("q", "{query}")
+	rootRouter.HandleFunc("/artifact/cardinality", handleCardinality).Queries("f", "{field:[a-zA-Z\\.]+}")
 }
 
 var (
@@ -101,3 +104,38 @@ func handleGet(w http.ResponseWriter, r *http.Request) {
 
 	RespondSearch(w, searchService)
 }
+
+func handleCardinality(w http.ResponseWriter, r *http.Request) {
+	var opts = mux.Vars(r)
+
+	query := elastic.NewBoolQuery().Must(
+		elastic.NewTermQuery("meta.deactivated", false),
+	)
+
+	s, err := datastore.Client().
+		Search(datastore.Index("oip042_artifact")).
+		Size(0).
+		Query(query).
+		Aggregation(
+			"cardinality",
+			elastic.NewCardinalityAggregation().
+				Field(opts["field"]),
+		).
+		Do(context.TODO())
+
+	if err != nil {
+		RespondESError(w, err)
+		return
+	}
+
+	agg, ok := s.Aggregations.Cardinality("cardinality")
+	if !ok {
+		RespondESError(w, errors.New("cardinality not found"))
+		return
+	}
+
+	RespondJSON(w, http.StatusOK, map[string]interface{}{
+		"c": agg.Value,
+		"f": opts["field"],
+	})
+}