-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.go
45 lines (37 loc) · 1.55 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package pezinventory
import (
"os"
"github.com/cloudfoundry-community/go-cfenv"
"github.com/codegangsta/negroni"
"github.com/gorilla/mux"
"github.com/pivotal-pez/cfmgo"
"github.com/unrolled/render"
)
var formatter *render.Render
//Formatter returns the address for a global response formatter
//realized in the `github.com/unrolled/render` package.
func Formatter() *render.Render {
if formatter == nil {
formatter = render.New(render.Options{
IndentJSON: true,
})
}
return formatter
}
//NewServer configures and returns a Server.
func NewServer(appEnv *cfenv.App) *negroni.Negroni {
inventoryServiceName := os.Getenv("INVENTORY_DB_NAME")
inventoryServiceURIName := os.Getenv("INVENTORY_DB_URI")
inventoryServiceURI := cfmgo.GetServiceBinding(inventoryServiceName, inventoryServiceURIName, appEnv)
inventoryCollection := cfmgo.Connect(cfmgo.NewCollectionDialer, inventoryServiceURI, InventoryCollectionName)
leaseCollection := cfmgo.Connect(cfmgo.NewCollectionDialer, inventoryServiceURI, LeaseCollectionName)
n := negroni.Classic()
mx := mux.NewRouter()
mx.HandleFunc("/v1/inventory", ListInventoryItemsHandler(inventoryCollection)).Methods("GET")
mx.HandleFunc("/v1/inventory", InsertInventoryItemHandler(inventoryCollection)).Methods("POST")
mx.HandleFunc("/v1/leases/{id}", FindLeaseByIDHandler(leaseCollection)).Methods("GET")
mx.HandleFunc("/v1/leases", FindLeasesHandler(leaseCollection)).Methods("GET")
mx.HandleFunc("/v1/leases", LeaseInventoryItemHandler(inventoryCollection, leaseCollection)).Methods("POST")
n.UseHandler(mx)
return n
}