微服務之間經過RabbitMQ通訊

微服務之間經過RabbitMQ通訊

微服務之間是相互獨立的，不像單個工程同樣各個模塊之間能夠直接經過方法調用實現通訊，相互獨立的服務直接通常的通訊方式是使用 HTTP協議、rpc協議或者使用消息中間件如RabbitMQ``Kafka等

在這篇文章 使用Golang和MongoDB構建微服務 已經實現了一個微服務的應用，在文章中已經實現了各個服務直接的通訊，是使用的 HTTP的形式 ，那各個服務之間如何經過 RabbitMQ進行消息通訊呢，咱們如今要實現一個功能，就是一個用戶預訂電影票的接口，須要服務 User Service(port 8000) 和 服務 **Booking Service(port 8003)**之間通訊，用戶預訂以後，把預訂信息寫入到 booking的數據庫中

安裝 RabbitMQ

安裝 RabbitMQ 以前須要先安裝 Erlang 的環境 ,而後下載安裝RabbitMQ ,請選擇對應的版本，安裝完成以後，RabbitMQ在Windows上是做爲一個服務在後臺運行，關於 RabbitMQ 的接口如何使用，請參考官網的 教程，有各個主流語言的實現咱們使用的是Go版本,請下載對應的實現接口 go get github.com/streadway/amqp

對RabbitMQ的接口作一下簡單的封裝

• 定義一個接口

messaging/message.go

type IMessageClient interface {
	ConnectToBroker(connectionStr string) error
	PublishToQueue(data []byte, queueName string) error
	SubscribeToQueue(queueName string, handlerFunc func(amqp.Delivery)) error
	Close()
}

type MessageClient struct {
	conn *amqp.Connection
}
複製代碼

• 鏈接接口

func (m *MessageClient) ConnectToBroker(connectionStr string) error {
	if connectionStr == "" {
		panic("the connection str mustnt be null")
	}
	var err error
	m.conn, err = amqp.Dial(connectionStr)
	return err
}
複製代碼

• 發佈消息接口

func (m *MessageClient) PublishToQueue(body []byte, queueName string) error {
	if m.conn == nil {
		panic("before publish you must connect the RabbitMQ first")
	}

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

	q, err := ch.QueueDeclare(
		queueName,
		false,
		false,
		false,
		false,
		nil,
	)
	failOnError(err, "Failed to declare a queue")

	err = ch.Publish(
		"",
		q.Name,
		false,
		false,
		amqp.Publishing{
			ContentType: "application/json",
			Body:        body,
		},
	)
	failOnError(err, "Failed to publish a message")

	return nil
}
複製代碼

• 訂閱消息接口

func (m *MessageClient) SubscribeToQueue(queueName string, handlerFunc func(amqp.Delivery)) error {
	ch, err := m.conn.Channel()
	//defer ch.Close()
	failOnError(err, "Failed to open a channel")

	q, err := ch.QueueDeclare(
		queueName,
		false,
		false,
		false,
		false,
		nil,
	)
	failOnError(err, "Failed to declare a queue")

	msgs, err := ch.Consume(
		q.Name,
		"",
		true,
		false,
		false,
		false,
		nil,
	)

	failOnError(err, "Failed to register a consumer")
	go consumeLoop(msgs, handlerFunc)
	return nil
}
複製代碼

實現通訊

在 User Service中定義一個新的POST接口 /user/{name}/booking,實現用戶的預訂功能，預訂以後，經過RabbitMQ發佈一個消息給 Booking Service,Booking Service接收到消息以後，作相應的處理(寫入數據庫)

User Service

• 初始化 MessageClient

users/controllers/user.go

var client messaging.IMessageClient

func init() {
	client = &messaging.MessageClient{}
	err := client.ConnectToBroker("amqp://guest:guest@localhost:5672/")
	if err != nil {
		fmt.Println("connect to rabbitmq error", err)
	}
}
複製代碼

• 添加新的路由和實現

routes.go

register("POST", "/user/{name}/booking", controllers.NewBooking, nil)
複製代碼

users/controllers/user.go

func NewBooking(w http.ResponseWriter, r *http.Request) {
	params := mux.Vars(r)
	user_name := params["name"]
	defer r.Body.Close()

	var bookings models.Booking
	body, _ := ioutil.ReadAll(r.Body)
	err := json.Unmarshal(body, &bookings)
	if err != nil {
		fmt.Println("the format body error ", err)
	}
	fmt.Println("user name:", user_name, bookings)
	go notifyMsg(body)
}
複製代碼

• 用一個協程實現消息的發佈

func notifyMsg(body []byte) {
	err := client.PublishToQueue(body, "new_booking")
	if err != nil {
		fmt.Println("Failed to publis message", err)
	}
}
複製代碼

Booking Service

• 初始化MessageClient

var client messaging.IMessageClient

func initMessage() {
	client = &messaging.MessageClient{}
	err := client.ConnectToBroker("amqp://guest:guest@localhost:5672/")
	if err != nil {
		fmt.Println("Failed to connect to RabbitMQ", err)
	}

	err = client.SubscribeToQueue("new_booking", getBooking)
	if err != nil {
		fmt.Println("Failed to comsuer the msg", err)
	}
}
複製代碼

在 web服務以前啓動

func main() {

	initMessage()

	r := routes.NewRouter()
	http.ListenAndServe(":8003", r)

}
複製代碼

• 接收後的消息處理

func getBooking(delivery amqp.Delivery) {

  var booking models.Booking
	json.Unmarshal(delivery.Body, &booking)
  booking.Id = bson.NewObjectId().Hex()
	dao.Insert("Booking", "BookModel", booking)
	fmt.Println("the booking msg", booking)
}
複製代碼

驗證，須要啓動 User Service 和 Booking Service
使用 Postman 發送對應的數據

post 127.0.0.1:8000/user/kevin_woo/booking

{
	"name":"kevin_woo",
	"books":[
		{
			"date":"20180727",
			"movies":["5b4c45d49d5e3e33c4a5b97a"]
		},
		{
			"date":"20180810",
			"movies":["5b4c45ea9d5e3e33c4a5b97b"]
		}
	]
}
複製代碼

能夠看到數據庫已經有了一條新的預訂信息

說明，我這裏POST的數據就是booking數據庫中的結構，實際狀況須要對數據進行封裝處理，在POST數據時，沒有對數據進行驗證， 在實際開發過程當中須要對各個數據作相應的驗證，這裏主要是看一下 RabbitMQ的消息傳遞處理的過程

源碼 Github