108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/emicklei/go-restful"
|
||
|
|
)
|
||
|
|
|
||
|
|
// This example has the same service definition as restful-user-resource
|
||
|
|
// but uses a different router (CurlyRouter) that does not use regular expressions
|
||
|
|
//
|
||
|
|
// POST http://localhost:8080/users
|
||
|
|
// <User><Id>1</Id><Name>Melissa Raspberry</Name></User>
|
||
|
|
//
|
||
|
|
// GET http://localhost:8080/users/1
|
||
|
|
//
|
||
|
|
// PUT http://localhost:8080/users/1
|
||
|
|
// <User><Id>1</Id><Name>Melissa</Name></User>
|
||
|
|
//
|
||
|
|
// DELETE http://localhost:8080/users/1
|
||
|
|
//
|
||
|
|
|
||
|
|
type User struct {
|
||
|
|
Id, Name string
|
||
|
|
}
|
||
|
|
|
||
|
|
type UserResource struct {
|
||
|
|
// normally one would use DAO (data access object)
|
||
|
|
users map[string]User
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u UserResource) Register(container *restful.Container) {
|
||
|
|
ws := new(restful.WebService)
|
||
|
|
ws.
|
||
|
|
Path("/users").
|
||
|
|
Consumes(restful.MIME_XML, restful.MIME_JSON).
|
||
|
|
Produces(restful.MIME_JSON, restful.MIME_XML) // you can specify this per route as well
|
||
|
|
|
||
|
|
ws.Route(ws.GET("/{user-id}").To(u.findUser))
|
||
|
|
ws.Route(ws.POST("").To(u.updateUser))
|
||
|
|
ws.Route(ws.PUT("/{user-id}").To(u.createUser))
|
||
|
|
ws.Route(ws.DELETE("/{user-id}").To(u.removeUser))
|
||
|
|
|
||
|
|
container.Add(ws)
|
||
|
|
}
|
||
|
|
|
||
|
|
// GET http://localhost:8080/users/1
|
||
|
|
//
|
||
|
|
func (u UserResource) findUser(request *restful.Request, response *restful.Response) {
|
||
|
|
id := request.PathParameter("user-id")
|
||
|
|
usr := u.users[id]
|
||
|
|
if len(usr.Id) == 0 {
|
||
|
|
response.AddHeader("Content-Type", "text/plain")
|
||
|
|
response.WriteErrorString(http.StatusNotFound, "User could not be found.")
|
||
|
|
} else {
|
||
|
|
response.WriteEntity(usr)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// POST http://localhost:8080/users
|
||
|
|
// <User><Id>1</Id><Name>Melissa Raspberry</Name></User>
|
||
|
|
//
|
||
|
|
func (u *UserResource) updateUser(request *restful.Request, response *restful.Response) {
|
||
|
|
usr := new(User)
|
||
|
|
err := request.ReadEntity(&usr)
|
||
|
|
if err == nil {
|
||
|
|
u.users[usr.Id] = *usr
|
||
|
|
response.WriteEntity(usr)
|
||
|
|
} else {
|
||
|
|
response.AddHeader("Content-Type", "text/plain")
|
||
|
|
response.WriteErrorString(http.StatusInternalServerError, err.Error())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// PUT http://localhost:8080/users/1
|
||
|
|
// <User><Id>1</Id><Name>Melissa</Name></User>
|
||
|
|
//
|
||
|
|
func (u *UserResource) createUser(request *restful.Request, response *restful.Response) {
|
||
|
|
usr := User{Id: request.PathParameter("user-id")}
|
||
|
|
err := request.ReadEntity(&usr)
|
||
|
|
if err == nil {
|
||
|
|
u.users[usr.Id] = usr
|
||
|
|
response.WriteHeaderAndEntity(http.StatusCreated, usr)
|
||
|
|
} else {
|
||
|
|
response.AddHeader("Content-Type", "text/plain")
|
||
|
|
response.WriteErrorString(http.StatusInternalServerError, err.Error())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// DELETE http://localhost:8080/users/1
|
||
|
|
//
|
||
|
|
func (u *UserResource) removeUser(request *restful.Request, response *restful.Response) {
|
||
|
|
id := request.PathParameter("user-id")
|
||
|
|
delete(u.users, id)
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
wsContainer := restful.NewContainer()
|
||
|
|
wsContainer.Router(restful.CurlyRouter{})
|
||
|
|
u := UserResource{map[string]User{}}
|
||
|
|
u.Register(wsContainer)
|
||
|
|
|
||
|
|
log.Printf("start listening on localhost:8080")
|
||
|
|
server := &http.Server{Addr: ":8080", Handler: wsContainer}
|
||
|
|
log.Fatal(server.ListenAndServe())
|
||
|
|
}
|