/ common / trpc / init.go
init.go
 1  // Copyright (c) 2024-2026 Tencent Zhuque Lab. All rights reserved.
 2  //
 3  // Licensed under the Apache License, Version 2.0 (the "License");
 4  // you may not use this file except in compliance with the License.
 5  // You may obtain a copy of the License at
 6  //
 7  //     http://www.apache.org/licenses/LICENSE-2.0
 8  //
 9  // Unless required by applicable law or agreed to in writing, software
10  // distributed under the License is distributed on an "AS IS" BASIS,
11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  // See the License for the specific language governing permissions and
13  // limitations under the License.
14  //
15  // Requirement: Any integration or derivative work must explicitly attribute
16  // Tencent Zhuque Lab (https://github.com/Tencent/AI-Infra-Guard) in its
17  // documentation or user interface, as detailed in the NOTICE file.
18  
19  package trpc
20  
21  import (
22  	"trpc.group/trpc-go/trpc-go"
23  )
24  
25  // InitTrpc 初始化trpc-go
26  func InitTrpc(configPath string) error {
27  	// 加载全局配置
28  	err := trpc.LoadGlobalConfig(configPath)
29  	if err != nil {
30  		return err
31  	}
32  
33  	// 创建trpc server(这会加载插件、启动admin等)
34  	_ = trpc.NewServer()
35  
36  	return nil
37  }
38  
39  // GetTrpcConfig 获取trpc配置
40  func GetTrpcConfig() *trpc.Config {
41  	return trpc.GlobalConfig()
42  }