Go-Zero를 사용하여 분산 작업 분배 및 스케줄링 구현

인터넷 비즈니스의 급속한 발전과 점차 증가하는 비즈니스 규모로 인해 단일 서버가 처리할 수 있는 데이터의 양은 수요를 충족시키기에는 턱없이 부족합니다. 높은 동시성, 고가용성, 고성능이라는 요구 사항을 충족하기 위해 시대가 요구하는 대로 분산 아키텍처가 등장했습니다.

분산 아키텍처에서 작업 분배 및 예약은 매우 중요한 구성 요소입니다. 작업 분배 및 스케줄링의 품질은 전체 시스템의 성능과 안정성에 직접적인 영향을 미칩니다. 여기서는 go-zero 프레임워크를 사용하여 분산 작업 분배 및 스케줄링을 구현하는 방법을 소개합니다.

1. 분산 작업 분배

작업 분배는 실행할 작업을 서로 다른 노드에 할당하는 것입니다. 분산 환경에서 작업 배포는 일반적으로 메시지 대기열을 통해 구현됩니다. 메시지 큐는 고가용성, 비동기식, 분리형 특성을 가지며 작업 분배 프로세스의 위험과 불확실성을 잘 해결할 수 있습니다.

go-zero는 Rabbitmq 및 kafka와 같은 메시지 대기열을 지원합니다. 여기서는 Rabbitmq를 예로 들어 이를 사용하여 분산 작업 분배를 달성하는 방법을 소개합니다.

1.1 Rabbitmq 설치

먼저 Rabbitmq를 설치해야 합니다. Rabbitmq 공식 홈페이지의 설명서를 참고하여 설치하시면 됩니다. 설치가 완료되면 새로운 가상호스트와 사용자를 생성하고 권한을 설정해야 합니다.

# 创建 vhost
sudo rabbitmqctl add_vhost vhost-test

# 创建用户
sudo rabbitmqctl add_user user-test passwd-test

# 设置用户权限
sudo rabbitmqctl set_permissions -p vhost-test user-test ".*" ".*" ".*"

1.2 Rabbitmq 구성

다음으로 구성 파일에 Rabbitmq 관련 구성을 추가해야 합니다.

[message]
    # 是否启用message
    enable = true
    # message类型，支持multi、nsq、kafka、rabbitmq
    type = "rabbitmq"
    # rabbitmq地址（IP:PORT）
    addr = "localhost:5672"
    # rabbitmq账号
    user = "user-test"
    # rabbitmq密码
    password = "passwd-test"
    # rabbitmq虚拟主机（默认值：/）
    virtualhost = "vhost-test"
    # 消息队列名称
    queue = "test-queue"

1.3 작업 보내기

go-zero에서는 메시지 대기열을 통해 분산 작업 배포를 달성할 수 있습니다. 메시지 대기열을 통해 메시지를 보낼 수 있으며, 메시지 소비자는 메시지 대기열에서 메시지를 검색하고 해당 작업을 수행합니다.

여기에서는 이메일 보내기를 예로 들어 작업을 보내는 방법을 소개합니다.

func sendMail(ctx context.Context, req *types.SendMailRequest) error {
    // 将任务转为消息发送到消息队列中
    return message.SendMessage(ctx, "test-queue", &types.SendMailRequest{
        Email: req.Email,
        Title: req.Title,
        Content: req.Content,
    })
}

이 방법에서는 이메일 작업을 메시지로 변환하고 SendMessage 함수를 통해 메시지를 메시지 대기열로 보냅니다.

2. 분산 작업 스케줄링

분산 작업 스케줄링은 여러 노드에 작업을 할당하고 스케줄링하는 것입니다. 분산 환경에서 작업 스케줄링은 일반적으로 cron과 같은 예약된 작업 시스템을 통해 수행됩니다.

go-zero 프레임워크는 작업 예약을 용이하게 할 수 있는 croexpr 패키지를 제공합니다. croexpr 패키지를 통해 cron 표현식을 구문 분석한 다음 해당 작업을 수행할 수 있습니다.

2.1 작업 추가

AddFunc, AddJob 및 기타 기능을 통해 예약된 작업 시스템에 작업을 추가할 수 있습니다. 예:

func startSchedule() {
    // 解析cron表达式，每天凌晨1点执行
    expr, err := cronexpr.Parse("0 0 1 * * *")
    if err != nil {
        log.Fatalf("failed to parse cron expression: %s", err.Error())
    }

    // 添加任务
    cron.Schedule(expr, cron.FuncJob(func() {
        // do something
    }))
}

이 예에서는 매일 오전 1시에 실행되는 cron 표현식을 구문 분석한 다음 예약된 작업에 작업 추가 FuncJob이 시스템에 추가됩니다.

2.2 작업 실행

예약된 작업 시스템은 작업에 해당하는 함수를 호출하여 작업을 실행합니다. 해당 처리 함수를 작성하여 작업을 처리할 수 있습니다. 예:

func handleMailTask() {
    // 监听消息队列
    message.ReceiveMessage(context.Background(),"test-queue", func(ctx context.Context, data []byte) error {
        var req types.SendMailRequest
        // 解析消息
        if err := json.Unmarshal(data, &req); err != nil {
            return err
        }

        // 发送邮件
        if err := sendMail(context.Background(), &req); err != nil {
            log.Printf("failed to send mail of %s: %s", req.Email, err.Error())
        }

        return nil
    })
}

이 처리 함수에서는 메시지 메시지 대기열을 수신하고 메시지를 얻은 다음 작업을 구문 분석합니다. 그런 다음 sendMail 함수를 호출하여 이메일을 보냅니다.

3. 요약

이 글에서는 Go-Zero 프레임워크를 사용하여 분산 작업 분배 및 스케줄링을 달성하는 방법을 소개합니다. 메시지 큐와 예약된 작업 시스템을 통해 작업 분배와 예약을 쉽게 실현하고 시스템의 성능과 가용성을 향상시킬 수 있습니다.

위 내용은 Go-Zero를 사용하여 분산 작업 분배 및 스케줄링 구현의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!