消息队列

README.md

@@ -1,2 +1,30 @@
 # go-rabbitmq
 rabbitmq的消费者和生产者
+
+# 在项目中安装使用
+```shell
+go get -u github.com/MQEnergy/go-rabbitmq
+```
+
+# 测试
+## 1、安装依赖
+```shell
+go mod tidy
+```
+
+## 运行example
+配置 amqp
+消费者
+```shell
+go run examples/consumer.go
+```
+生产者
+```shell
+go run examples/producer.go
+```
+查看效果
+
+## 2、单元测试
+```shell
+go test
+```

examples/consumer.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"fmt"
+	go_rabbitmq "github.com/MQEnergy/go-rabbitmq"
+	"sync"
+	"time"
+)
+
+func main() {
+	config := &go_rabbitmq.Config{
+		User:     "root",
+		Password: "",
+		Host:     "",
+		Port:     "5672",
+		Vhost:    "",
+	}
+	mq := go_rabbitmq.New(config, "test", "", "", 0, 1)
+	time.Sleep(time.Second * 1)
+	amqphandler(mq, 3)
+}
+
+// amqphandler 消息队列处理
+func amqphandler(mq *go_rabbitmq.RabbitMQ, consumerNum int) error {
+	var wg sync.WaitGroup
+	cherrors := make(chan error)
+	wg.Add(consumerNum)
+	for i := 0; i < consumerNum; i++ {
+		fmt.Printf("正在开启消费者：第 %d 个\n", i+1)
+		go func() {
+			defer wg.Done()
+			deliveries, err := mq.Consume()
+			if err != nil {
+				cherrors <- err
+			}
+			for d := range deliveries {
+				// 消费者逻辑 to do
+				fmt.Printf("got %dbyte delivery: [%v] %s %q\n", len(d.Body), d.DeliveryTag, d.Exchange, d.Body)
+				d.Ack(false)
+			}
+		}()
+	}
+	select {
+	case err := <-cherrors:
+		close(cherrors)
+		fmt.Printf("Consumer failed: %s\n", err)
+		return err
+	}
+	wg.Wait()
+	return nil
+}

examples/producer.go

@@ -0,0 +1,26 @@
+package main
+
+import (
+	"fmt"
+	go_rabbitmq "github.com/MQEnergy/go-rabbitmq"
+	"time"
+)
+
+func main() {
+	config := &go_rabbitmq.Config{
+		User:     "root",
+		Password: "",
+		Host:     "",
+		Port:     "5672",
+		Vhost:    "",
+	}
+	mq := go_rabbitmq.New(config, "test", "", "", 0, 1)
+	// 需要等待一秒钟
+	time.Sleep(time.Second * 1)
+
+	data := []byte("{\"hello\":\"world " + time.Now().Format("2006-01-02 15:04:05") + "\"}")
+	if err := mq.Push(data); err != nil {
+		panic(err)
+	}
+	fmt.Println("Push succeeded!", string(data))
+}

go.mod

@@ -0,0 +1,5 @@
+module github.com/MQEnergy/go-rabbitmq
+
+go 1.18
+
+require github.com/streadway/amqp v1.0.0