From 0c7eb94fb95ba66d64038e52c1d7ee1d36b69f0e Mon Sep 17 00:00:00 2001
From: ajatprabha <ajat.prabha.leo@gmail.com>
Date: Sat, 18 Sep 2021 17:55:59 +0530
Subject: [PATCH] add PayloadStringer interface

---
 conversion_helpers.go | 44 ++++++++++++++++++++++++++++---------------
 1 file changed, 29 insertions(+), 15 deletions(-)

diff --git a/conversion_helpers.go b/conversion_helpers.go
index d786843..c872409 100644
--- a/conversion_helpers.go
+++ b/conversion_helpers.go
@@ -15,6 +15,27 @@ import (
 //   - conversion function from an external type to an internal type
 // ****************************************************************************
 
+type PayloadStringer interface {
+	String([]byte) string
+}
+
+type PayloadStringerFunc func([]byte) string
+
+func (f PayloadStringerFunc) String(b []byte) string {
+	return f(b)
+}
+
+var payloadStringer PayloadStringer = PayloadStringerFunc(func(payload []byte) string {
+	if !isPrintable(payload) {
+		return "non-printable bytes"
+	}
+	return string(payload)
+})
+
+func SetPayloadStringer(stringer PayloadStringer) {
+	payloadStringer = stringer
+}
+
 type QueueStateSnapshot struct {
 	// Name of the queue.
 	Queue string `json:"queue"`
@@ -129,7 +150,7 @@ func toTaskInfo(info *asynq.TaskInfo) *TaskInfo {
 		ID:            info.ID,
 		Queue:         info.Queue,
 		Type:          info.Type,
-		Payload:       toPrintablePayload(info.Payload),
+		Payload:       payloadStringer.String(info.Payload),
 		State:         info.State.String(),
 		MaxRetry:      info.MaxRetry,
 		Retried:       info.Retried,
@@ -172,7 +193,7 @@ func toActiveTask(t *asynq.TaskInfo) *ActiveTask {
 	base := &BaseTask{
 		ID:        t.ID,
 		Type:      t.Type,
-		Payload:   toPrintablePayload(t.Payload),
+		Payload:   payloadStringer.String(t.Payload),
 		Queue:     t.Queue,
 		MaxRetry:  t.MaxRetry,
 		Retried:   t.Retried,
@@ -198,7 +219,7 @@ func toPendingTask(t *asynq.TaskInfo) *PendingTask {
 	base := &BaseTask{
 		ID:        t.ID,
 		Type:      t.Type,
-		Payload:   toPrintablePayload(t.Payload),
+		Payload:   payloadStringer.String(t.Payload),
 		Queue:     t.Queue,
 		MaxRetry:  t.MaxRetry,
 		Retried:   t.Retried,
@@ -242,18 +263,11 @@ func isPrintable(data []byte) bool {
 	return !isAllSpace
 }
 
-func toPrintablePayload(payload []byte) string {
-	if !isPrintable(payload) {
-		return "non-printable bytes"
-	}
-	return string(payload)
-}
-
 func toScheduledTask(t *asynq.TaskInfo) *ScheduledTask {
 	base := &BaseTask{
 		ID:        t.ID,
 		Type:      t.Type,
-		Payload:   toPrintablePayload(t.Payload),
+		Payload:   payloadStringer.String(t.Payload),
 		Queue:     t.Queue,
 		MaxRetry:  t.MaxRetry,
 		Retried:   t.Retried,
@@ -282,7 +296,7 @@ func toRetryTask(t *asynq.TaskInfo) *RetryTask {
 	base := &BaseTask{
 		ID:        t.ID,
 		Type:      t.Type,
-		Payload:   toPrintablePayload(t.Payload),
+		Payload:   payloadStringer.String(t.Payload),
 		Queue:     t.Queue,
 		MaxRetry:  t.MaxRetry,
 		Retried:   t.Retried,
@@ -311,7 +325,7 @@ func toArchivedTask(t *asynq.TaskInfo) *ArchivedTask {
 	base := &BaseTask{
 		ID:        t.ID,
 		Type:      t.Type,
-		Payload:   toPrintablePayload(t.Payload),
+		Payload:   payloadStringer.String(t.Payload),
 		Queue:     t.Queue,
 		MaxRetry:  t.MaxRetry,
 		Retried:   t.Retried,
@@ -355,7 +369,7 @@ func toSchedulerEntry(e *asynq.SchedulerEntry) *SchedulerEntry {
 		ID:            e.ID,
 		Spec:          e.Spec,
 		TaskType:      e.Task.Type(),
-		TaskPayload:   toPrintablePayload(e.Task.Payload()),
+		TaskPayload:   payloadStringer.String(e.Task.Payload()),
 		Opts:          opts,
 		NextEnqueueAt: e.Next.Format(time.RFC3339),
 		PrevEnqueueAt: prev,
@@ -437,7 +451,7 @@ func toWorkerInfo(info *asynq.WorkerInfo) *WorkerInfo {
 		TaskID:     info.TaskID,
 		Queue:      info.Queue,
 		TaskType:   info.TaskType,
-		TakPayload: toPrintablePayload(info.TaskPayload),
+		TakPayload: payloadStringer.String(info.TaskPayload),
 		Started:    info.Started.Format(time.RFC3339),
 	}
 }