Merge pull request #91 from dongxuny/master

Add pointer related codes

go.mod

@@ -7,7 +7,7 @@ require (
 	github.com/gin-gonic/gin v1.8.1
 	github.com/golang-jwt/jwt/v4 v4.4.2
 	github.com/prometheus/client_golang v1.13.0
-	github.com/rookie-ninja/rk-entry/v2 v2.2.14
+	github.com/rookie-ninja/rk-entry/v2 v2.2.15
 	github.com/rookie-ninja/rk-logger v1.2.13
 	github.com/rookie-ninja/rk-query v1.2.14
 	github.com/rs/xid v1.3.0

go.sum

@@ -323,8 +323,8 @@ github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFR
 github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=
 github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
 github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
-github.com/rookie-ninja/rk-entry/v2 v2.2.14 h1:PHzofuw52zfskrgJKC+WIMt1LtoBosrukjWHZ/DKsTE=
-github.com/rookie-ninja/rk-entry/v2 v2.2.14/go.mod h1:P/Fd6Oyvvx0ITbEU2lzO3KkQ9miWVwd84aPaC0LMD0o=
+github.com/rookie-ninja/rk-entry/v2 v2.2.15 h1:K5Ww1wCjOcAvK11pBA8qM7FpbHyFE5i+YQBAIYxPrZc=
+github.com/rookie-ninja/rk-entry/v2 v2.2.15/go.mod h1:P/Fd6Oyvvx0ITbEU2lzO3KkQ9miWVwd84aPaC0LMD0o=
 github.com/rookie-ninja/rk-logger v1.2.13 h1:ERxeNZUmszlY4xehHcJRXECPtbjYIXzN8yRIyYyLGsg=
 github.com/rookie-ninja/rk-logger v1.2.13/go.mod h1:0ZiGn1KsHKOmCv+FHMH7k40DWYSJcj5yIR3EYcjlnLs=
 github.com/rookie-ninja/rk-query v1.2.14 h1:aYNyMXixpsEYRfEOz9Npt5QG3A6BQlo9vKjYc78x7bc=

middleware/context/context.go

@@ -24,6 +24,7 @@ import (
 var (
 	noopTracerProvider = trace.NewNoopTracerProvider()
 	noopEvent          = rkquery.NewEventFactory().CreateEventNoop()
+	pointerCreator     rkcursor.PointerCreator
 )
 
 // GetIncomingHeaders extract call-scoped incoming headers
@@ -52,12 +53,23 @@ func SetHeaderToClient(ctx *gin.Context, key, value string) {
 	header.Set(key, value)
 }
 
+// SetPointerCreator override  rkcursor.PointerCreator
+func SetPointerCreator(creator rkcursor.PointerCreator) {
+	pointerCreator = creator
+}
+
 // GetCursor create rkcursor.Cursor instance
 func GetCursor(ctx *gin.Context) *rkcursor.Cursor {
-	return rkcursor.NewCursor(
+	res := rkcursor.NewCursor(
 		rkcursor.WithLogger(GetLogger(ctx)),
 		rkcursor.WithEvent(GetEvent(ctx)),
 		rkcursor.WithEntryNameAndType(GetEntryName(ctx), "GinEntry"))
+
+	if pointerCreator != nil {
+		res.Creator = pointerCreator
+	}
+
+	return res
 }
 
 // GetEvent extract takes the call-scoped EventData from middleware.

middleware/context/context_test.go

@@ -8,6 +8,7 @@ package rkginctx
 import (
 	"github.com/gin-gonic/gin"
 	"github.com/golang-jwt/jwt/v4"
+	rkcursor "github.com/rookie-ninja/rk-entry/v2/cursor"
 	"github.com/rookie-ninja/rk-entry/v2/middleware"
 	"github.com/rookie-ninja/rk-logger"
 	"github.com/rookie-ninja/rk-query"
@@ -270,6 +271,35 @@ func TestGetCsrfToken(t *testing.T) {
 	assert.Equal(t, "value", GetCsrfToken(ctx))
 }
 
+func TestSetPointerCreator(t *testing.T) {
+	assert.Nil(t, pointerCreator)
+
+	SetPointerCreator(createFakePointer)
+
+	assert.NotNil(t, pointerCreator)
+}
+
+func createFakePointer(p *rkcursor.CursorPayload) rkcursor.Pointer {
+	return &fakePointer{}
+}
+
+type fakePointer struct{}
+
+func (f fakePointer) PrintError(err error) {
+	//TODO implement me
+	panic("implement me")
+}
+
+func (f fakePointer) ObserveError(err error) error {
+	//TODO implement me
+	panic("implement me")
+}
+
+func (f fakePointer) Release() {
+	//TODO implement me
+	panic("implement me")
+}
+
 func assertNotPanic(t *testing.T) {
 	if r := recover(); r != nil {
 		// Expect panic to be called with non nil error