95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package v2
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
etcdErr "github.com/coreos/etcd/error"
|
|
"github.com/coreos/etcd/log"
|
|
"github.com/coreos/etcd/store"
|
|
"github.com/coreos/go-raft"
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
func GetHandler(w http.ResponseWriter, req *http.Request, s Server) error {
|
|
var err error
|
|
var event *store.Event
|
|
events := make([]*store.Event, 0)
|
|
|
|
vars := mux.Vars(req)
|
|
key := "/" + vars["key"]
|
|
|
|
// Help client to redirect the request to the current leader
|
|
if req.FormValue("consistent") == "true" && s.State() != raft.Leader {
|
|
leader := s.Leader()
|
|
hostname, _ := s.PeerURL(leader)
|
|
url := hostname + req.URL.Path
|
|
log.Debugf("Redirect consistent get to %s", url)
|
|
http.Redirect(w, req, url, http.StatusTemporaryRedirect)
|
|
return nil
|
|
}
|
|
|
|
recursive := (req.FormValue("recursive") == "true")
|
|
sorted := (req.FormValue("sorted") == "true")
|
|
|
|
if req.FormValue("wait") == "true" { // watch
|
|
// Create a command to watch from a given index (default 0).
|
|
var sinceIndex uint64 = 0
|
|
|
|
waitIndex := req.FormValue("waitIndex")
|
|
if waitIndex != "" {
|
|
sinceIndex, err = strconv.ParseUint(string(req.FormValue("waitIndex")), 10, 64)
|
|
if err != nil {
|
|
return etcdErr.NewError(etcdErr.EcodeIndexNaN, "Watch From Index", store.UndefIndex, store.UndefTerm)
|
|
}
|
|
}
|
|
|
|
// Start the watcher on the store.
|
|
eventChan, err := s.Store().Watch(key, recursive, sinceIndex, s.CommitIndex(), s.Term())
|
|
if err != nil {
|
|
return etcdErr.NewError(500, key, store.UndefIndex, store.UndefTerm)
|
|
}
|
|
|
|
cn, _ := w.(http.CloseNotifier)
|
|
closeChan := cn.CloseNotify()
|
|
|
|
eventLoop:
|
|
for {
|
|
select {
|
|
case <-closeChan:
|
|
return nil
|
|
case event = <-eventChan:
|
|
// for events other than expire, just one event for one watch
|
|
// for expire event, we might have a stream of events
|
|
// we use a nil item to terminate the expire event stream
|
|
if event != nil && event.Action == store.Expire {
|
|
events = append(events, event)
|
|
} else {
|
|
events = append(events, event)
|
|
break eventLoop
|
|
}
|
|
}
|
|
}
|
|
|
|
} else { //get
|
|
// Retrieve the key from the store.
|
|
event, err = s.Store().Get(key, recursive, sorted, s.CommitIndex(), s.Term())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
var b []byte
|
|
|
|
w.Header().Add("X-Etcd-Index", fmt.Sprint(events[0].Index))
|
|
w.Header().Add("X-Etcd-Term", fmt.Sprint(events[0].Term))
|
|
w.WriteHeader(http.StatusOK)
|
|
b, _ = json.Marshal(events)
|
|
|
|
w.Write(b)
|
|
|
|
return nil
|
|
}
|