-
Notifications
You must be signed in to change notification settings - Fork 0
/
message_parser.go
55 lines (47 loc) · 1.4 KB
/
message_parser.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
package middleware
import (
"context"
"encoding/json"
"net/http"
"github.com/antonio-muniz/gommon/pkg/respond"
"github.com/antonio-muniz/gommon/pkg/validator"
)
type ContextKey string
const Message ContextKey = "message"
type MessageType interface {
NewPointer() interface{}
Dereference(pointer interface{}) interface{}
}
type messageParser struct {
messageType MessageType
nextHandler http.Handler
}
func MessageParser(messageType MessageType, nextHandler http.Handler) http.Handler {
return messageParser{
messageType: messageType,
nextHandler: nextHandler,
}
}
func (m messageParser) ServeHTTP(response http.ResponseWriter, request *http.Request) {
messagePointer := m.messageType.NewPointer()
err := json.NewDecoder(request.Body).Decode(messagePointer)
if err != nil {
respond.MalformedRequest(response)
return
}
message := m.messageType.Dereference(messagePointer)
validator := validator.New(validator.ErrorFieldFromJSONTag())
validationResult, err := validator.Validate(message)
if err != nil {
respond.InternalServerError(response)
return
}
if validationResult.Invalid() {
respond.InvalidRequestParameters(response, validationResult)
return
}
originalContext := request.Context()
contextWithMessage := context.WithValue(originalContext, Message, message)
requestWithMessage := request.WithContext(contextWithMessage)
m.nextHandler.ServeHTTP(response, requestWithMessage)
}