go.temporal.io/server@v1.23.0/common/config/fx.go (about) 1 // The MIT License 2 // 3 // Copyright (c) 2020 Temporal Technologies Inc. All rights reserved. 4 // 5 // Copyright (c) 2020 Uber Technologies, Inc. 6 // 7 // Permission is hereby granted, free of charge, to any person obtaining a copy 8 // of this software and associated documentation files (the "Software"), to deal 9 // in the Software without restriction, including without limitation the rights 10 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 11 // copies of the Software, and to permit persons to whom the Software is 12 // furnished to do so, subject to the following conditions: 13 // 14 // The above copyright notice and this permission notice shall be included in 15 // all copies or substantial portions of the Software. 16 // 17 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 18 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 19 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 20 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 21 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 22 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 23 // THE SOFTWARE. 24 25 package config 26 27 import ( 28 "go.uber.org/fx" 29 30 "go.temporal.io/server/common/primitives" 31 ) 32 33 // ServicePortMap contains the gRPC ports for our services. 34 type ServicePortMap map[primitives.ServiceName]int 35 36 var Module = fx.Provide( 37 provideRPCConfig, 38 provideMembershipConfig, 39 provideServicePortMap, 40 ) 41 42 func provideRPCConfig(cfg *Config, svcName primitives.ServiceName) *RPC { 43 c := cfg.Services[string(svcName)].RPC 44 45 return &c 46 } 47 48 func provideMembershipConfig(cfg *Config) *Membership { 49 return &cfg.Global.Membership 50 } 51 52 func provideServicePortMap(cfg *Config) ServicePortMap { 53 servicePortMap := make(ServicePortMap) 54 for sn, sc := range cfg.Services { 55 servicePortMap[primitives.ServiceName(sn)] = sc.RPC.GRPCPort 56 } 57 58 return servicePortMap 59 }