impl entities except for PATCH
This commit is contained in:
82
.view/port.go
Normal file
82
.view/port.go
Normal file
@@ -0,0 +1,82 @@
|
||||
package view
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"io/ioutil"
|
||||
"local/dndex/storage"
|
||||
"local/dndex/storage/entity"
|
||||
"net/http"
|
||||
|
||||
"github.com/buger/jsonparser"
|
||||
)
|
||||
|
||||
func port(g storage.RateLimitedGraph, w http.ResponseWriter, r *http.Request) error {
|
||||
switch r.Method {
|
||||
case http.MethodGet:
|
||||
return portGet(g, w, r)
|
||||
case http.MethodPost:
|
||||
return portPost(g, w, r)
|
||||
default:
|
||||
http.NotFound(w, r)
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
func portGet(g storage.RateLimitedGraph, w http.ResponseWriter, r *http.Request) error {
|
||||
namespace, err := getNamespace(r)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return nil
|
||||
}
|
||||
|
||||
ones, err := g.List(r.Context(), namespace)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return json.NewEncoder(w).Encode(map[string]interface{}{namespace: ones})
|
||||
}
|
||||
|
||||
func portPost(g storage.RateLimitedGraph, w http.ResponseWriter, r *http.Request) error {
|
||||
namespace, err := getNamespace(r)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return nil
|
||||
}
|
||||
|
||||
b, err := ioutil.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
inserted := 0
|
||||
|
||||
var errIn error
|
||||
_, err = jsonparser.ArrayEach(b, func(b []byte, _ jsonparser.ValueType, _ int, err error) {
|
||||
if err != nil {
|
||||
errIn = err
|
||||
return
|
||||
}
|
||||
|
||||
o := entity.One{}
|
||||
if err := json.Unmarshal(b, &o); err != nil {
|
||||
errIn = err
|
||||
return
|
||||
}
|
||||
|
||||
if err := g.Insert(r.Context(), namespace, o); err != nil {
|
||||
errIn = err
|
||||
return
|
||||
}
|
||||
|
||||
inserted += 1
|
||||
}, namespace)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if errIn != nil {
|
||||
return errIn
|
||||
}
|
||||
|
||||
return json.NewEncoder(w).Encode(map[string]int{namespace: inserted})
|
||||
}
|
||||
Reference in New Issue
Block a user