Refactor health check

minhduc140583 · minhduc140583 · commit 2f9c87af9c09 · 2021-04-24T18:44:40.000+07:00
diff --git a/health_checker.go b/health_checker.go
@@ -11,11 +11,17 @@ type HealthChecker struct {
 	Service   string
 }
 
-func NewHealthChecker(client *sqs.SQS, queueName string) *HealthChecker {
-	return NewSQSHealthChecker(client, queueName, "sqs")
+func NewHealthChecker(client *sqs.SQS, queueName string, options...string) *HealthChecker {
+	var name string
+	if len(options) > 0 && len(options[0]) > 0 {
+		name = options[0]
+	} else {
+		name = "sqs"
+	}
+	return NewSQSHealthChecker(client, name, queueName)
 }
 func NewSQSHealthChecker(client *sqs.SQS, name string, queueName string) *HealthChecker {
-	return &HealthChecker{client, &queueName, name}
+	return &HealthChecker{Client: client, QueueName: &queueName, Service: name}
 }
 
 func (h *HealthChecker) Name() string {

Original file line number	Diff line number	Diff line change
`@@ -11,11 +11,17 @@ type HealthChecker struct {`
`11`	`11`	`Service string`
`12`	`12`	`}`
`13`	`13`
`14`		`-func NewHealthChecker(client sqs.SQS, queueName string) HealthChecker {`
`15`		`- return NewSQSHealthChecker(client, queueName, "sqs")`
	`14`	`+func NewHealthChecker(client sqs.SQS, queueName string, options...string) HealthChecker {`
	`15`	`+ var name string`
	`16`	`+ if len(options) > 0 && len(options[0]) > 0 {`
	`17`	`+ name = options[0]`
	`18`	`+ } else {`
	`19`	`+ name = "sqs"`
	`20`	`+ }`
	`21`	`+ return NewSQSHealthChecker(client, name, queueName)`
`16`	`22`	`}`
`17`	`23`	`func NewSQSHealthChecker(client sqs.SQS, name string, queueName string) HealthChecker {`
`18`		`- return &HealthChecker{client, &queueName, name}`
	`24`	`+ return &HealthChecker{Client: client, QueueName: &queueName, Service: name}`
`19`	`25`	`}`
`20`	`26`
`21`	`27`	`func (h *HealthChecker) Name() string {`