Skip to content

Commit

Permalink
debug
Browse files Browse the repository at this point in the history
  • Loading branch information
daemonfire300 committed Jan 21, 2022
1 parent cc02bb7 commit 8c87074
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions grpc/middleware.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,11 @@
package grpc

import (
"bytes"
"context"
"encoding/base64"
"encoding/gob"
"fmt"
"strings"

"github.com/pace/bricks/pkg/tracking/utm"
Expand All @@ -20,7 +23,11 @@ func ContextWithUTMFromMetadata(parentCtx context.Context, md metadata.MD) conte
return parentCtx
}
var utmData utm.UTMData
if err := gob.NewDecoder(strings.NewReader(dataSlice[0])).Decode(&utmData); err != nil {
res, err := base64.StdEncoding.DecodeString(dataSlice[0])
if err != nil {
panic(fmt.Sprintf("woooooop wooooop: %s", err))
}
if err := gob.NewDecoder(bytes.NewBuffer(res)).Decode(&utmData); err != nil {
return parentCtx
}
return utm.ContextWithUTMData(parentCtx, utmData)
Expand All @@ -35,5 +42,5 @@ func EncodeContextWithUTMData(parentCtx context.Context) context.Context {
if err := gob.NewEncoder(&w).Encode(utmData); err != nil {
return parentCtx
}
return metadata.AppendToOutgoingContext(parentCtx, utmMetadataKey, w.String())
return metadata.AppendToOutgoingContext(parentCtx, utmMetadataKey, base64.StdEncoding.EncodeToString([]byte(w.String())))
}

0 comments on commit 8c87074

Please sign in to comment.