dm/vain
1
0
Fork 0
forked from sm/vain
Code

simple in-memory storage working for POST/GET

Browse Source
This commit is contained in:
Stephen McQuay 2016-02-13 01:18:06 -08:00
parent c1b5f61b64
commit b0ebbf742b
4 changed files with 76 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cmd/ysvd/main.go
View File

@ -21,15 +21,18 @@ const usage = `ysvd
environment vars: environment vars:
YSV_PORT: tcp listen port YSV_PORT: tcp listen port
YSV_HOST: hostname to use
` `
type config struct { type config struct {
Port int Port int
Host string
} }
func main() { func main() {
c := &config{ c := &config{
Port: 4040, Port: 4040,
Host: "localhost",
} }
if err := envconfig.Process("ysv", c); err != nil { if err := envconfig.Process("ysv", c); err != nil {
fmt.Fprintf(os.Stderr, "problem processing environment: %v", err) fmt.Fprintf(os.Stderr, "problem processing environment: %v", err)
@ -50,7 +53,8 @@ func main() {
} }
log.Printf("serving at: http://%s:%d/", hostname, c.Port) log.Printf("serving at: http://%s:%d/", hostname, c.Port)
sm := http.NewServeMux() sm := http.NewServeMux()
vain.NewServer(sm) ms := vain.NewMemStore()
vain.NewServer(sm, ms, c.Host)
addr := fmt.Sprintf(":%d", c.Port) addr := fmt.Sprintf(":%d", c.Port)
if err := http.ListenAndServe(addr, sm); err != nil { if err := http.ListenAndServe(addr, sm); err != nil {
log.Printf("problem with http server: %v", err) log.Printf("problem with http server: %v", err)

3
doc.go
View File

@ -19,10 +19,9 @@ A sample json object:
{ {
"vcs": "mercurial", "vcs": "mercurial",
"path": "mcquay.me/vain",
"repo": "https://s.mcquay.me/sm/vain" "repo": "https://s.mcquay.me/sm/vain"
} }
"path" and "repo" are required; leaving off the "vcs" member defaults to "git". "repo" is required; leaving off the "vcs" member defaults to "git".
*/ */
package vain package vain

27
server.go
View File

@ -1,25 +1,48 @@
package vain package vain
import ( import (
"encoding/json"
"fmt" "fmt"
"net/http" "net/http"
"strings"
) )
type Server struct { type Server struct {
hostname string
storage *MemStore
} }
func (s *Server) ServeHTTP(w http.ResponseWriter, req *http.Request) { func (s *Server) ServeHTTP(w http.ResponseWriter, req *http.Request) {
switch req.Method { switch req.Method {
case "GET": case "GET":
fmt.Fprintf(w, "<!DOCTYPE html>\n<html><head>\n")
for _, p := range s.storage.All() {
fmt.Fprintf(w, "%s\n", p)
}
fmt.Fprintf(w, "</head>\n</html>\n")
case "POST": case "POST":
if req.URL.Path == "/" {
http.Error(w, fmt.Sprintf("invalid path %q", req.URL.Path), http.StatusBadRequest)
return
}
p := Package{}
if err := json.NewDecoder(req.Body).Decode(&p); err != nil {
http.Error(w, fmt.Sprintf("unable to parse json from body: %v", err), http.StatusInternalServerError)
return
}
p.Path = fmt.Sprintf("%s/%s", s.hostname, strings.Trim(req.URL.Path, "/"))
s.storage.Add(p)
case "PATCH": case "PATCH":
default: default:
http.Error(w, fmt.Sprintf("unsupported method %q; accepted: POST, GET, PATCH", req.Method), http.StatusMethodNotAllowed) http.Error(w, fmt.Sprintf("unsupported method %q; accepted: POST, GET, PATCH", req.Method), http.StatusMethodNotAllowed)
} }
} }
func NewServer(sm *http.ServeMux) *Server { func NewServer(sm *http.ServeMux, ms *MemStore, hostname string) *Server {
s := &Server{} s := &Server{
storage: ms,
hostname: hostname,
}
sm.Handle("/", s) sm.Handle("/", s)
return s return s
} }

45
storage.go Normal file
View File

@ -0,0 +1,45 @@
package vain
import (
"errors"
"sync"
)
type MemStore struct {
l sync.RWMutex
p map[string]Package
}
func NewMemStore() *MemStore {
return &MemStore{
p: make(map[string]Package),
}
}
func (ms MemStore) Add(p Package) error {
ms.l.Lock()
ms.p[p.Path] = p
ms.l.Unlock()
return nil
}
func (ms MemStore) Remove(path string) error {
ms.l.Lock()
delete(ms.p, path)
ms.l.Unlock()
return nil
}
func (ms MemStore) Save() error {
return errors.New("save is not implemented")
}
func (ms MemStore) All() []Package {
r := []Package{}
ms.l.RLock()
for _, p := range ms.p {
r = append(r, p)
}
ms.l.RUnlock()
return r
}