RabbitMQ를 사용하여 Golang에서 분산 작업 대기열을 구현하기 위한 성능 조정 기술

go get github.com/streadway/amqp

package main

import (
    "fmt"
    "log"

    "github.com/streadway/amqp"
)

func failOnError(err error, msg string) {
    if err != nil {
        log.Fatalf("%s: %s", msg, err)
    }
}

func main() {
    conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
    failOnError(err, "Failed to connect to RabbitMQ")
    defer conn.Close()

    ch, err := conn.Channel()
    failOnError(err, "Failed to open a channel")
    defer ch.Close()

    // ...
}

func main() {
    // ...
    q, err := ch.QueueDeclare(
        "task_queue", // 队列名称
        true,         // durable
        false,        // delete when unused
        false,        // exclusive
        false,        // no-wait
        nil,          // arguments
    )
    failOnError(err, "Failed to declare a queue")

    body := "task body"
    err = ch.Publish(
        "",         // exchange
        q.Name,     // routing key
        false,      // mandatory
        false,      // immediate
        amqp.Publishing{
            Delay:      0,
            ContentType: "text/plain",
            Body:        []byte(body),
        })
    failOnError(err, "Failed to publish a message")

    // ...
}

func main() {
    // ...
    msgs, err := ch.Consume(
        q.Name, // queue
        "",     // consumer
        false,  // auto-ack
        false,  // exclusive
        false,  // no-local
        false,  // no-wait
        nil,    // args
    )
    failOnError(err, "Failed to register a consumer")

    forever := make(chan bool)

    go func() {
        for d := range msgs {
            log.Printf("Received a message: %s", d.Body)
            // 处理任务的逻辑
            d.Ack(false)
        }
    }()

    log.Printf(" [*] Waiting for messages. To exit press CTRL+C")
    <-forever
    
    // ...
}

err = ch.Qos(
    1,     // prefetch count
    0,     // prefetch size
    false, // global
)
failOnError(err, "Failed to set QoS")

for i := 0; i < 10; i++ {
    go func() {
        for d := range msgs {
            log.Printf("Received a message: %s", d.Body)
            // 处理任务的逻辑
            d.Ack(false)
        }
    }()
}