← Back to branch summary

~ubuntu-branches/debian/sid/golang-github-blevesearch-bleve/sid

« back to all changes in this revision

Viewing changes to http/doc_index.go

  • Committer: Package Import Robot
  • Author(s): Michael Lustfield
  • Date: 2017-03-30 16:06:03 UTC
  • Revision ID: package-import@ubuntu.com-20170330160603-0oogmb960l7918jx
Tags: upstream-0.5.0+git20170324.202.4702785f
Import upstream version 0.5.0+git20170324.202.4702785f

Show diffs side-by-side

added added

removed removed

Lines of Context:
http/doc_index.go
 
1
//  Copyright (c) 2014 Couchbase, Inc.
 
2
//
 
3
// Licensed under the Apache License, Version 2.0 (the "License");
 
4
// you may not use this file except in compliance with the License.
 
5
// You may obtain a copy of the License at
 
6
//
 
7
//              http://www.apache.org/licenses/LICENSE-2.0
 
8
//
 
9
// Unless required by applicable law or agreed to in writing, software
 
10
// distributed under the License is distributed on an "AS IS" BASIS,
 
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
12
// See the License for the specific language governing permissions and
 
13
// limitations under the License.
 
14
 
 
15
package http
 
16
 
 
17
import (
 
18
        "encoding/json"
 
19
        "fmt"
 
20
        "io/ioutil"
 
21
        "net/http"
 
22
)
 
23
 
 
24
type DocIndexHandler struct {
 
25
        defaultIndexName string
 
26
        IndexNameLookup  varLookupFunc
 
27
        DocIDLookup      varLookupFunc
 
28
}
 
29
 
 
30
func NewDocIndexHandler(defaultIndexName string) *DocIndexHandler {
 
31
        return &DocIndexHandler{
 
32
                defaultIndexName: defaultIndexName,
 
33
        }
 
34
}
 
35
 
 
36
func (h *DocIndexHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 
37
 
 
38
        // find the index to operate on
 
39
        var indexName string
 
40
        if h.IndexNameLookup != nil {
 
41
                indexName = h.IndexNameLookup(req)
 
42
        }
 
43
        if indexName == "" {
 
44
                indexName = h.defaultIndexName
 
45
        }
 
46
        index := IndexByName(indexName)
 
47
        if index == nil {
 
48
                showError(w, req, fmt.Sprintf("no such index '%s'", indexName), 404)
 
49
                return
 
50
        }
 
51
 
 
52
        // find the doc id
 
53
        var docID string
 
54
        if h.DocIDLookup != nil {
 
55
                docID = h.DocIDLookup(req)
 
56
        }
 
57
        if docID == "" {
 
58
                showError(w, req, "document id cannot be empty", 400)
 
59
                return
 
60
        }
 
61
 
 
62
        // read the request body
 
63
        requestBody, err := ioutil.ReadAll(req.Body)
 
64
        if err != nil {
 
65
                showError(w, req, fmt.Sprintf("error reading request body: %v", err), 400)
 
66
                return
 
67
        }
 
68
 
 
69
        // parse request body as json
 
70
        var doc interface{}
 
71
        err = json.Unmarshal(requestBody, &doc)
 
72
        if err != nil {
 
73
                showError(w, req, fmt.Sprintf("error parsing request body as JSON: %v", err), 400)
 
74
                return
 
75
        }
 
76
 
 
77
        err = index.Index(docID, doc)
 
78
        if err != nil {
 
79
                showError(w, req, fmt.Sprintf("error indexing document '%s': %v", docID, err), 500)
 
80
                return
 
81
        }
 
82
 
 
83
        rv := struct {
 
84
                Status string `json:"status"`
 
85
        }{
 
86
                Status: "ok",
 
87
        }
 
88
        mustEncode(w, rv)
 
89
}