Use the context interface rather than struct (#33)

fossas · Feb 7, 2024 · 3fea682 · 3fea682
1 parent 5eb2e6d
commit 3fea682
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/batch/middlware.go b/batch/middlware.go
@@ -9,7 +9,7 @@ import (
 )
 
 func (b *BatchSubsystem) pushMiddleware(ctx context.Context, next func() error) error {
-	mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Ctx)
+	mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Context)
 
 	if bid, ok := mh.Job().GetCustom("bid"); ok {
 		batchId, err := b.batchManager.getBatchIdFromInterface(bid)
@@ -33,7 +33,7 @@ func (b *BatchSubsystem) pushMiddleware(ctx context.Context, next func() error)
 
 func (b *BatchSubsystem) handleJobFinished(success bool) func(ctx context.Context, next func() error) error {
 	return func(ctx context.Context, next func() error) error {
-		mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Ctx)
+		mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Context)
 
 		if success {
 			// check if this is a success / complete job from batch

diff --git a/uniq/uniq.go b/uniq/uniq.go
@@ -73,7 +73,7 @@ func (u *UniqSubsystem) addMiddleware() {
 }
 
 func (u *UniqSubsystem) lockMiddleware(ctx context.Context, next func() error) error {
-	mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Ctx)
+	mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Context)
 
 	var uniqueFor float64
 	uniqueForValue, ok := mh.Job().GetCustom("unique_for")
@@ -129,7 +129,7 @@ func (u *UniqSubsystem) lockMiddleware(ctx context.Context, next func() error) e
 
 func (u *UniqSubsystem) releaseLockMiddleware(releaseAt string) func(context.Context, func() error) error {
 	return func(ctx context.Context, next func() error) error {
-		mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Ctx)
+		mh := ctx.Value(manager.MiddlewareHelperKey).(manager.Context)
 
 		if _, ok := mh.Job().GetCustom("unique_for"); !ok {
 			return next()