From 72fb979e2ca7d353922a9647506faf96e8086bb6 Mon Sep 17 00:00:00 2001
From: Yichao Yang <yichao@temporal.io>
Date: Tue, 8 Mar 2022 10:17:55 -0800
Subject: [PATCH] Match history and proto task category definition (#2586)

---
 service/history/tasks/category.go | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/service/history/tasks/category.go b/service/history/tasks/category.go
index fffa4bc301f..e6fcc9155cb 100644
--- a/service/history/tasks/category.go
+++ b/service/history/tasks/category.go
@@ -28,6 +28,8 @@ import (
 	"fmt"
 	"strconv"
 	"sync"
+
+	enumsspb "go.temporal.io/server/api/enums/v1"
 )
 
 type (
@@ -41,11 +43,11 @@ type (
 )
 
 const (
-	CategoryIDUnspecified int32 = iota
-	CategoryIDTransfer
-	CategoryIDTimer
-	CategoryIDVisibility
-	CategoryIDReplication
+	CategoryIDUnspecified = int32(enumsspb.TASK_CATEGORY_UNSPECIFIED)
+	CategoryIDTransfer    = int32(enumsspb.TASK_CATEGORY_TRANSFER)
+	CategoryIDTimer       = int32(enumsspb.TASK_CATEGORY_TIMER)
+	CategoryIDReplication = int32(enumsspb.TASK_CATEGORY_REPLICATION)
+	CategoryIDVisibility  = int32(enumsspb.TASK_CATEGORY_VISIBILITY)
 )
 
 const (
@@ -57,8 +59,8 @@ const (
 const (
 	CategoryNameTransfer    = "transfer"
 	CategoryNameTimer       = "timer"
-	CategoryNameVisibility  = "visibility"
 	CategoryNameReplication = "replication"
+	CategoryNameVisibility  = "visibility"
 )
 
 var (
@@ -74,17 +76,17 @@ var (
 		name:  CategoryNameTimer,
 	}
 
-	CategoryVisibility = Category{
-		id:    CategoryIDVisibility,
-		cType: CategoryTypeImmediate,
-		name:  CategoryNameVisibility,
-	}
-
 	CategoryReplication = Category{
 		id:    CategoryIDReplication,
 		cType: CategoryTypeImmediate,
 		name:  CategoryNameReplication,
 	}
+
+	CategoryVisibility = Category{
+		id:    CategoryIDVisibility,
+		cType: CategoryTypeImmediate,
+		name:  CategoryNameVisibility,
+	}
 )
 
 var (