From a67403d733621c26b9b9dc35d7c4d89bbc38c0f5 Mon Sep 17 00:00:00 2001
From: yuyuanshifu <747342561@qq.com>
Date: Tue, 1 Sep 2020 17:34:27 +0800
Subject: [PATCH] replace redis with rabbitmq

---
 modules/setting/setting.go | 10 ++++++++--
 modules/worker/worker.go   |  5 +++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/modules/setting/setting.go b/modules/setting/setting.go
index de8598ba0..e933690cd 100755
--- a/modules/setting/setting.go
+++ b/modules/setting/setting.go
@@ -423,6 +423,9 @@ var (
 	Broker        string
 	DefaultQueue  string
 	ResultBackend string
+	Exchange      string
+	ExchangeType  string
+	BindingKey    string
 
 	//decompress config
 	DecompressAddress        string
@@ -1084,9 +1087,12 @@ func NewContext() {
 	}
 
 	sec = Cfg.Section("machinery")
-	Broker = sec.Key("BROKER").MustString("redis://localhost:6379")
+	Broker = sec.Key("BROKER").MustString("amqp://admin:pcladmin@127.0.0.1:5672/")
 	DefaultQueue = sec.Key("DEFAULT_QUEUE").MustString("DecompressTasksQueue")
-	ResultBackend = sec.Key("RESULT_BACKEND").MustString("redis://localhost:6379")
+	ResultBackend = sec.Key("RESULT_BACKEND").MustString("amqp://admin:pcladmin@127.0.0.1:5672/")
+	Exchange = sec.Key("EXCHANGE").MustString("decompress_exchange")
+	ExchangeType = sec.Key("EXCHANGE_TYPE").MustString("direct")
+	BindingKey = sec.Key("BINDING_KEY").MustString("decompress_task")
 
 	sec = Cfg.Section("decompress")
 	DecompressAddress = sec.Key("HOST").MustString("http://192.168.207.34:39987")
diff --git a/modules/worker/worker.go b/modules/worker/worker.go
index 37445c978..1dbfff3b3 100755
--- a/modules/worker/worker.go
+++ b/modules/worker/worker.go
@@ -15,6 +15,11 @@ func NewTaskCenter() {
 		Broker:        setting.Broker,
 		DefaultQueue:  setting.DefaultQueue,
 		ResultBackend: setting.ResultBackend,
+		AMQP:          &mchConf.AMQPConfig{
+			Exchange:     "machinery_exchange",
+			ExchangeType: "direct",
+			BindingKey:   "machinery_task",
+		},
 	}
 	tc, err := machinery.NewServer(cnf)
 	if err != nil {