在进程内优雅管理多个服务

后端

2023-10-14 07:14:14

在 go-zero 社区里，经常会有同学问，把 API gateway 和 RPC service 放在同一个进程内可不可以？怎么弄？有时也会有同学把对外服务和消费队列放在一个进程内。

这篇文章就来给大家详细讲解一下，如何在 go-zero 中优雅地在同一个进程内管理多个服务，同时兼顾性能和可靠性。

为什么要在同一个进程内管理多个服务？

在同一个进程内管理多个服务有以下好处：

降低资源占用： 多个服务共享同一个进程，可以避免创建多个进程带来的资源开销。
简化部署： 一个进程只用部署一次，简化了部署流程。
提升性能： 在同一进程内通信，可以避免跨进程通信带来的性能损耗。

如何在 go-zero 中实现？

go-zero 提供了优雅管理多个服务的特性，可以轻松实现多个服务在同一进程内的运行。

具体实现步骤如下：

创建服务管理器： 使用 NewServiceManager 函数创建服务管理器。
注册服务： 使用 Register 方法注册服务到服务管理器。
启动服务： 使用 Start 方法启动服务管理器。

下面是一个示例代码：

package main

import (
	"context"
	"fmt"

	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/service"
)

func main() {
	sm := service.NewServiceManager()
	sm.Register("foo", newFooService())
	sm.Register("bar", newBarService())
	sm.Start()
	
	<-context.Background().Done() // 阻塞主进程
}

type fooService struct{}

func newFooService() service.Service {
	return &fooService{}
}

func (s *fooService) Start() error {
	logx.Info("foo service started")
	return nil
}

func (s *fooService) Stop() error {
	logx.Info("foo service stopped")
	return nil}

type barService struct{}

func newBarService() service.Service {
	return &barService{}
}

func (s *barService) Start() error {
	logx.Info("bar service started")
	return nil
}

func (s *barService) Stop() error {
	logx.Info("bar service stopped")
	return nil}