90 lines
3.0 KiB
Go
90 lines
3.0 KiB
Go
package profilesvc
|
|
|
|
import (
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
"github.com/go-kit/kit/log"
|
|
)
|
|
|
|
// Middleware describes a service (as opposed to endpoint) middleware.
|
|
type Middleware func(Service) Service
|
|
|
|
func LoggingMiddleware(logger log.Logger) Middleware {
|
|
return func(next Service) Service {
|
|
return &loggingMiddleware{
|
|
next: next,
|
|
logger: logger,
|
|
}
|
|
}
|
|
}
|
|
|
|
type loggingMiddleware struct {
|
|
next Service
|
|
logger log.Logger
|
|
}
|
|
|
|
func (mw loggingMiddleware) PostProfile(ctx context.Context, p Profile) (err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "PostProfile", "id", p.ID, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.PostProfile(ctx, p)
|
|
}
|
|
|
|
func (mw loggingMiddleware) GetProfile(ctx context.Context, id string) (p Profile, err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "GetProfile", "id", id, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.GetProfile(ctx, id)
|
|
}
|
|
|
|
func (mw loggingMiddleware) PutProfile(ctx context.Context, id string, p Profile) (err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "PutProfile", "id", id, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.PutProfile(ctx, id, p)
|
|
}
|
|
|
|
func (mw loggingMiddleware) PatchProfile(ctx context.Context, id string, p Profile) (err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "PatchProfile", "id", id, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.PatchProfile(ctx, id, p)
|
|
}
|
|
|
|
func (mw loggingMiddleware) DeleteProfile(ctx context.Context, id string) (err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "DeleteProfile", "id", id, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.DeleteProfile(ctx, id)
|
|
}
|
|
|
|
func (mw loggingMiddleware) GetAddresses(ctx context.Context, profileID string) (addresses []Address, err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "GetAddresses", "profileID", profileID, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.GetAddresses(ctx, profileID)
|
|
}
|
|
|
|
func (mw loggingMiddleware) GetAddress(ctx context.Context, profileID string, addressID string) (a Address, err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "GetAddress", "profileID", profileID, "addressID", addressID, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.GetAddress(ctx, profileID, addressID)
|
|
}
|
|
|
|
func (mw loggingMiddleware) PostAddress(ctx context.Context, profileID string, a Address) (err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "PostAddress", "profileID", profileID, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.PostAddress(ctx, profileID, a)
|
|
}
|
|
|
|
func (mw loggingMiddleware) DeleteAddress(ctx context.Context, profileID string, addressID string) (err error) {
|
|
defer func(begin time.Time) {
|
|
mw.logger.Log("method", "DeleteAddress", "profileID", profileID, "addressID", addressID, "took", time.Since(begin), "err", err)
|
|
}(time.Now())
|
|
return mw.next.DeleteAddress(ctx, profileID, addressID)
|
|
}
|