From 12111d21f1100e65f5148a244db7f2a7bcdede40 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8B=8F=E5=B0=B9=E5=B2=9A?= <770236076@qq.com>
Date: Wed, 19 Aug 2020 14:18:47 +0800
Subject: [PATCH] =?UTF-8?q?im=E8=81=8A=E5=A4=A9=E8=AE=B0=E5=BD=95=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 business/model/message.go  | 17 +++++++++++++----
 globals/beegodb/beegodb.go |  1 +
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/business/model/message.go b/business/model/message.go
index b0c0bbd0e..cbe5e65dc 100644
--- a/business/model/message.go
+++ b/business/model/message.go
@@ -51,10 +51,19 @@ func (*MessageStatus) TableIndex() [][]string {
 
 type ImMessageRecord struct {
 	ModelIDCULD
-	UserID        string `orm:"size(48);column(user_id)" json:"userID"` //消息发送方userID
-	Receive       string `json:"receive"`                               //接收方ID，可以是userID，也可以是群ID
-	Content       string `orm:"type(text)" json:"content"`              //消息内容
-	MessageType   int    `json:"messageType"`                           //消息类型，文字，图片
+	UserID        string `orm:"size(48);column(user_id)" json:"userID"`     //消息发送方userID
+	ReceiveID     string `orm:"size(48);column(receive_id)" json:"receive"` //接收方ID，可以是userID，也可以是群ID
+	Content       string `orm:"type(text)" json:"content"`                  //消息内容
+	MessageType   int    `json:"messageType"`                               //消息类型，文字，图片
 	StoreID       int    `orm:"column(store_id)" json:"storeID"`
 	VendorOrderID string `orm:"column(vendor_order_id);size(48)" json:"vendorOrderID"`
 }
+
+func (*ImMessageRecord) TableIndex() [][]string {
+	return [][]string{
+		[]string{"UserID"},
+		[]string{"ReceiveID"},
+		[]string{"StoreID"},
+		[]string{"VendorOrderID"},
+	}
+}
diff --git a/globals/beegodb/beegodb.go b/globals/beegodb/beegodb.go
index 96fd31ced..be52e1e91 100644
--- a/globals/beegodb/beegodb.go
+++ b/globals/beegodb/beegodb.go
@@ -27,6 +27,7 @@ func Init() {
 
 	orm.RegisterModel(new(model.Message))
 	orm.RegisterModel(new(model.MessageStatus))
+	orm.RegisterModel(new(model.ImMessageRecord))
 
 	orm.RegisterModel(&model.Place{})
 	orm.RegisterModel(&model.Store{}, &model.StoreSub{}, &model.StoreMap{}, &model.StoreCourierMap{}, &model.StoreCategoryMap{})