Difrex/zoorest
Difrex/zoorest
1
0
Fork 0
Code Issues 1 Pull Requests Releases 2 Wiki Activity

Protect methonds and rmr fix

Browse Source
This commit is contained in:
Denis Zheleztsov 2017-02-17 12:34:43 +03:00
parent f1618cde11
commit dd3c90c94d
2 changed files with 33 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
rest/api.go
View File

@ -28,6 +28,12 @@ type Get struct {
// LS ...
func (zk ZooNode) LS(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
e := strings.Join([]string{"Method", r.Method, "not alowed"}, " ")
w.WriteHeader(500)
w.Write([]byte(e))
return
}
vars := mux.Vars(r)
path := vars["path"]
@ -82,7 +88,7 @@ func (zk ZooNode) UP(w http.ResponseWriter, r *http.Request) {
} else if r.Method == "POST" {
go func() { ch <- zk.UpdateNode(path, content) }()
} else {
e := strings.Join([]string{r.Method, "not alowed"}, " ")
e := strings.Join([]string{"Method", r.Method, "not alowed"}, " ")
w.WriteHeader(500)
w.Write([]byte(e))
return
@ -97,8 +103,8 @@ func (zk ZooNode) UP(w http.ResponseWriter, r *http.Request) {
// RM ...
func (zk ZooNode) RM(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
e := strings.Join([]string{r.Method, "not alowed"}, " ")
if r.Method != "DELETE" {
e := strings.Join([]string{"Method", r.Method, "not alowed"}, " ")
w.WriteHeader(500)
w.Write([]byte(e))
return
@ -106,13 +112,33 @@ func (zk ZooNode) RM(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
path := vars["path"]
go func() { zk.RMR(path) }()
var rmPath string
rmPath = strings.Join([]string{zk.Path, path}, "")
if path == "/" {
e := "Skiping root path"
w.WriteHeader(500)
w.Write([]byte(e))
return
}
if strings.Contains(rmPath, "//") {
rmPath = strings.Replace(rmPath, "//", "/", 1)
}
go func() { zk.RMR(rmPath) }()
w.WriteHeader(200)
w.Write([]byte(rmPath))
}
// GET ...
func (zk ZooNode) GET(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
e := strings.Join([]string{"Method", r.Method, "not alowed"}, " ")
w.WriteHeader(500)
w.Write([]byte(e))
return
}
vars := mux.Vars(r)
path := vars["path"]

16
rest/zoo.go
View File

@ -95,25 +95,15 @@ func (z ZooNode) GetNode(path string) Get {
//RMR remove Zk node recursive
func (z ZooNode) RMR(path string) {
var rmPath string
rmPath = strings.Join([]string{z.Path, path}, "")
if path == "/" {
return
}
if strings.Contains(rmPath, "//") {
rmPath = strings.Replace(rmPath, "//", "/", 1)
}
log.Print("rm: ", rmPath)
c, _, err := z.Conn.Children(rmPath)
log.Print("rm: ", path)
c, _, err := z.Conn.Children(path)
if err != nil {
log.Print("[zk ERROR] ", err)
}
log.Print("[WARNING] Trying delete ", path)
if len(c) > 0 {
for _, child := range c {
childPath := strings.Join([]string{rmPath, child}, "/")
childPath := strings.Join([]string{path, child}, "/")
z.RMR(childPath)
}
}