131 lines
3.8 KiB
Go
131 lines
3.8 KiB
Go
package index_management
|
|
|
|
import (
|
|
httprouter "infini.sh/framework/core/api/router"
|
|
"infini.sh/framework/core/elastic"
|
|
"infini.sh/framework/core/util"
|
|
"net/http"
|
|
log "src/github.com/cihub/seelog"
|
|
)
|
|
|
|
func (handler APIHandler) HandleGetMappingsAction(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
targetClusterID := ps.ByName("id")
|
|
client := elastic.GetClient(targetClusterID)
|
|
indexName := ps.ByName("index")
|
|
resBody := newResponseBody()
|
|
var copyAll = false
|
|
if indexName == "*" {
|
|
indexName = ""
|
|
copyAll = true
|
|
}
|
|
_, _, idxs, err := client.GetMapping(copyAll, indexName)
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
//if copyAll {
|
|
// for key, _ := range *idxs {
|
|
// if strings.HasPrefix(key, ".") || strings.HasPrefix(key, "infini-") {
|
|
// delete(*idxs, key)
|
|
// }
|
|
// }
|
|
//}
|
|
|
|
handler.WriteJSON(w, idxs, http.StatusOK)
|
|
}
|
|
|
|
func (handler APIHandler) HandleGetIndicesAction(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
targetClusterID := ps.ByName("id")
|
|
client := elastic.GetClient(targetClusterID)
|
|
catIndices, err := client.GetIndices("")
|
|
resBody := util.MapStr{}
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err.Error()
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
handler.WriteJSON(w, catIndices, http.StatusOK)
|
|
}
|
|
|
|
func (handler APIHandler) HandleGetSettingsAction(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
targetClusterID := ps.ByName("id")
|
|
client := elastic.GetClient(targetClusterID)
|
|
indexName := ps.ByName("index")
|
|
resBody := newResponseBody()
|
|
indexes, err := client.GetIndexSettings(indexName)
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err.Error()
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
handler.WriteJSON(w, indexes, http.StatusOK)
|
|
}
|
|
|
|
func (handler APIHandler) HandleUpdateSettingsAction(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
targetClusterID := ps.ByName("id")
|
|
client := elastic.GetClient(targetClusterID)
|
|
indexName := ps.ByName("index")
|
|
settings := map[string]interface{}{}
|
|
resBody := newResponseBody()
|
|
err := handler.DecodeJSON(req, &settings)
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
err = client.UpdateIndexSettings(indexName, settings)
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
resBody["result"] = "updated"
|
|
handler.WriteJSON(w, resBody, http.StatusCreated)
|
|
}
|
|
|
|
func (handler APIHandler) HandleDeleteIndexAction(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
targetClusterID := ps.ByName("id")
|
|
client := elastic.GetClient(targetClusterID)
|
|
indexName := ps.ByName("index")
|
|
resBody := newResponseBody()
|
|
err := client.DeleteIndex(indexName)
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
resBody["result"] = "deleted"
|
|
handler.WriteJSON(w, resBody, http.StatusOK)
|
|
}
|
|
|
|
func (handler APIHandler) HandleCreateIndexAction(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
targetClusterID := ps.ByName("id")
|
|
client := elastic.GetClient(targetClusterID)
|
|
indexName := ps.ByName("index")
|
|
resBody := newResponseBody()
|
|
config := map[string]interface{}{}
|
|
err := handler.DecodeJSON(req, &config)
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
err = client.CreateIndex(indexName, config)
|
|
if err != nil {
|
|
log.Error(err)
|
|
resBody["error"] = err
|
|
handler.WriteJSON(w, resBody, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
resBody["result"] = "created"
|
|
handler.WriteJSON(w, resBody, http.StatusCreated)
|
|
}
|