forked from oapi-codegen/oapi-codegen
-
Notifications
You must be signed in to change notification settings - Fork 0
/
petstore.go
56 lines (45 loc) · 1.48 KB
/
petstore.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
46
47
48
49
50
51
52
53
54
55
56
// This is an example of implementing the Pet Store from the OpenAPI documentation
// found at:
// https://github.com/OAI/OpenAPI-Specification/blob/master/examples/v3.0/petstore.yaml
package main
import (
"flag"
"fmt"
"log"
"net/http"
"os"
"github.com/gin-gonic/gin"
"github.com/deepmap/oapi-codegen/examples/petstore-expanded/gin/api"
middleware "github.com/deepmap/oapi-codegen/pkg/gin-middleware"
)
func NewGinPetServer(petStore *api.PetStore, port int) *http.Server {
swagger, err := api.GetSwagger()
if err != nil {
fmt.Fprintf(os.Stderr, "Error loading swagger spec\n: %s", err)
os.Exit(1)
}
// Clear out the servers array in the swagger spec, that skips validating
// that server names match. We don't know how this thing will be run.
swagger.Servers = nil
// This is how you set up a basic gin router
r := gin.Default()
// Use our validation middleware to check all requests against the
// OpenAPI schema.
r.Use(middleware.OapiRequestValidator(swagger))
// We now register our petStore above as the handler for the interface
r = api.RegisterHandlers(r, petStore)
s := &http.Server{
Handler: r,
Addr: fmt.Sprintf("0.0.0.0:%d", port),
}
return s
}
func main() {
var port = flag.Int("port", 8080, "Port for test HTTP server")
flag.Parse()
// Create an instance of our handler which satisfies the generated interface
petStore := api.NewPetStore()
s := NewGinPetServer(petStore, *port)
// And we serve HTTP until the world ends.
log.Fatal(s.ListenAndServe())
}