Created Shard Manager Service

cmd/server/cadence/server.go

@@ -53,6 +53,7 @@ import (
 	"github.com/uber/cadence/service/frontend"
 	"github.com/uber/cadence/service/history"
 	"github.com/uber/cadence/service/matching"
+	"github.com/uber/cadence/service/shardmanager"
 	"github.com/uber/cadence/service/worker"
 )
 
@@ -322,6 +323,8 @@ func (s *server) startService() common.Daemon {
 		daemon, err = matching.NewService(&params)
 	case service.Worker:
 		daemon, err = worker.NewService(&params)
+	case service.ShardManager:
+		daemon, err = shardmanager.NewService(&params, resource.NewResourceFactory())
 	}
 	if err != nil {
 		params.Logger.Fatal("Fail to start "+s.name+" service ", tag.Error(err))

common/dynamicconfig/constants.go

@@ -1466,6 +1466,29 @@ const (
 	// Value type: Int
 	// Default value: 100
 	ESAnalyzerMinNumWorkflowsForAvg
+
+	// key for shard manager
+
+	// ShardManagerPersistenceMaxQPS is the max qps a shard manager host can query DB
+	// KeyName: shardManager.persistenceMaxQPS
+	// Value type: Int
+	// Default value: 3000
+	// Allowed filters: N/A
+	ShardManagerPersistenceMaxQPS
+	// ShardManagerPersistenceGlobalMaxQPS is the max qps matching cluster can query DB
+	// KeyName: shardManager.persistenceGlobalMaxQPS
+	// Value type: Int
+	// Default value: 0
+	// Allowed filters: N/A
+	ShardManagerPersistenceGlobalMaxQPS
+
+	// ShardManagerThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger
+	// KeyName: shardManager.throttledLogRPS
+	// Value type: Int
+	// Default value: 20
+	// Allowed filters: N/A
+	ShardManagerThrottledLogRPS
+
 	// Usage: VisibilityArchivalQueryMaxRangeInDays is the maximum number of days for a visibility archival query
 	// KeyName: N/A
 	// Default value: N/A
@@ -3921,6 +3944,21 @@ var IntKeys = map[IntKey]DynamicInt{
 		Description:  "ESAnalyzerMinNumWorkflowsForAvg controls how many workflows to have at least to rely on workflow run time avg per type",
 		DefaultValue: 100,
 	},
+	ShardManagerPersistenceMaxQPS: {
+		KeyName:      "shardManager.persistenceMaxQPS",
+		Description:  "ShardManagerPersistenceMaxQPS is the max qps shard manager host can query DB",
+		DefaultValue: 3000,
+	},
+	ShardManagerPersistenceGlobalMaxQPS: {
+		KeyName:      "shardManager.persistenceGlobalMaxQPS",
+		Description:  "ShardManagerPersistenceGlobalMaxQPS is the max qps shard manager cluster can query DB",
+		DefaultValue: 0,
+	},
+	ShardManagerThrottledLogRPS: {
+		KeyName:      "shardManager.throttledLogRPS",
+		Description:  "ShardManagerThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger",
+		DefaultValue: 20,
+	},
 	VisibilityArchivalQueryMaxRangeInDays: {
 		KeyName:      "frontend.visibilityArchivalQueryMaxRangeInDays",
 		Description:  "VisibilityArchivalQueryMaxRangeInDays is the maximum number of days for a visibility archival query",

common/metrics/defs.go

@@ -67,6 +67,7 @@ const (
 	History
 	Matching
 	Worker
+	ShardManager
 	NumServices
 )
 

common/resource/resource.go

@@ -18,6 +18,8 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 // THE SOFTWARE.
 
+//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination resource_mock.go -self_package github.com/uber/cadence/common/resource
+
 package resource
 
 import (
@@ -48,8 +50,16 @@ import (
 	"github.com/uber/cadence/common/persistence"
 	persistenceClient "github.com/uber/cadence/common/persistence/client"
 	"github.com/uber/cadence/common/quotas/global/rpc"
+	"github.com/uber/cadence/common/service"
 )
 
+type ResourceFactory interface {
+	NewResource(params *Params,
+		serviceName string,
+		serviceConfig *service.Config,
+	) (resource Resource, err error)
+}
+
 type (
 	// Resource is the interface which expose common resources
 	Resource interface {

common/resource/resourceImpl.go

@@ -63,6 +63,19 @@
 	"github.com/uber/cadence/common/service"
 )
 
+func NewResourceFactory() ResourceFactory {
+	return &resourceImplFactory{}
+}
+
+type resourceImplFactory struct{}
+
+func (*resourceImplFactory) NewResource(params *Params,
+	serviceName string,
+	serviceConfig *service.Config,
+) (resource Resource, err error) {
+	return New(params, serviceName, serviceConfig)
+}
+
 type (
 
 	// VisibilityManagerInitializer is the function each service should implement