golang 配置文件的方式实现gRPC接口映射-Go服务器开发

在golang中，可以使用配置文件的方式实现gRPC接口映射。这种方法将服务的配置信息存储在一个单独的文件中，可以方便地进行修改和管理。

以下是一个使用配置文件方式实现gRPC接口映射的示例代码：

config.yaml:

grpc:
  port: 8080

myService:
  interceptor: MyInterceptor

main.go:

package main

import (
	"context"
	"fmt"
	"log"
	"net"

	"google.golang.org/grpc"
	"gopkg.in/yaml.v2"
)

type Config struct {
	GRPC      GRPCConfig      `yaml:"grpc"`
	MyService MyServiceConfig `yaml:"myService"`
}

type GRPCConfig struct {
	Port int `yaml:"port"`
}

type MyServiceConfig struct {
	Interceptor string `yaml:"interceptor"`
}

type myService struct{}

func (s *myService) SayHello(ctx context.Context, req *HelloRequest) (*HelloResponse, error) {
	return &HelloResponse{Message: fmt.Sprintf("Hello %s", req.Name)}, nil
}

func main() {
	config := loadConfig()
	port := fmt.Sprintf(":%d", config.GRPC.Port)

	lis, err := net.Listen("tcp", port)
	if err != nil {
		log.Fatalf("failed to listen: %v", err)
	}
	srv := grpc.NewServer(grpc.UnaryInterceptor(getInterceptor(config)))

	RegisterMyServiceServer(srv, &myService{})

	if err := srv.Serve(lis); err != nil {
		log.Fatalf("failed to serve: %v", err)
	}
}

func loadConfig() Config {
	var config Config
	file, err := ioutil.ReadFile("./config.yaml")
	if err != nil {
		log.Fatalf("failed to read config file: %v", err)
	}
	err = yaml.Unmarshal(file, &config)
	if err != nil {
		log.Fatalf("failed to unmarshal config file: %v", err)
	}
	return config
}

func getInterceptor(config Config) grpc.UnaryServerInterceptor {
	interceptor := MyInterceptor
	if config.MyService.Interceptor != "" {
		interceptor = getFuncByName(config.MyService.Interceptor).(grpc.UnaryServerInterceptor)
	}
	return interceptor
}

func MyInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
	log.Printf("Received request: %+v\n", req)
	resp, err := handler(ctx, req)
	log.Printf("Sent response: %+v\n", resp)
	return resp, err
}

func getFuncByName(name string) interface{} {
	switch name {
	case "MyInterceptor":
		return MyInterceptor
	default:
		panic(fmt.Sprintf("unknown function name: %s", name))
	}
}

在上面的示例中，我们首先定义了一个Config结构体来存储配置信息，并将其从yaml文件中读取并解析。然后，我们使用该配置信息来启动grpc服务。

在getInterceptor函数中，我们根据配置信息获取拦截器函数，并将其作为UnaryInterceptor选项传递给grpc服务对象。如果未指定拦截器函数，则默认使用MyInterceptor函数。

通过配置文件方式实现gRPC接口映射可以方便地进行修改和管理，并且支持多种不同的配置选项。但是需要注意的是，在解析yaml文件时需要安装相应的第三方库，例如上面示例中使用的”gopkg.in/yaml.v2”库。