fix:优化获取列表添加ctype字段 用以区分微信和whatsapp

crontask/compute_statistic.go

@@ -107,6 +107,7 @@ func (l *CronTask) computeStatistic() {
 			friendCountInt, _ = l.svcCtx.DB.Contact.Query().Where(
 				contact.Type(1),
 				contact.WxWxid(wxinfo.Wxid),
+				contact.Ctype(1),
 			).Count(l.ctx)
 			orgFriendCountInt += friendCountInt
 			allHourFriendCountInt += friendCountInt
@@ -115,6 +116,7 @@ func (l *CronTask) computeStatistic() {
 			groupCountInt, _ = l.svcCtx.DB.Contact.Query().Where(
 				contact.Type(2),
 				contact.WxWxid(wxinfo.Wxid),
+				contact.Ctype(1),
 			).Count(l.ctx)
 			orgGroupCountInt += groupCountInt
 			allHourGroupCountInt += groupCountInt
@@ -188,7 +190,10 @@ func (l *CronTask) computeStatistic() {
 		}
 
 		LabelsCount := []custom_types.LabelDist{}
-		err := l.svcCtx.DB.LabelRelationship.Query().Where(labelrelationship.OrganizationIDEQ(orgID)).GroupBy(labelrelationship.FieldLabelID).Aggregate(ent.Count()).Scan(l.ctx, &LabelsCount)
+		err := l.svcCtx.DB.LabelRelationship.Query().Where(
+			labelrelationship.OrganizationIDEQ(orgID),
+			labelrelationship.Ctype(1),
+		).GroupBy(labelrelationship.FieldLabelID).Aggregate(ent.Count()).Scan(l.ctx, &LabelsCount)
 		l.Errorf("save hour data error:%v \n", err)
 		LabelsCountSet[orgID] = LabelsCount
 

crontask/send_msg.go

@@ -58,7 +58,11 @@ func (l *CronTask) sendMsg() {
 			var err error
 			if allContact && allGroup {
 				// 获取 contact 表中  wx_wxid 等于 req.Fromwxid 的 type 为1或2的数据
-				userList, err = l.svcCtx.DB.Contact.Query().Where(contact.WxWxid(batch.Fromwxid), contact.TypeIn(1, 2)).All(l.ctx)
+				userList, err = l.svcCtx.DB.Contact.Query().Where(
+					contact.WxWxid(batch.Fromwxid),
+					contact.TypeIn(1, 2),
+					contact.Ctype(1),
+				).All(l.ctx)
 				if err != nil {
 					l.Logger.Errorf("userlist err: %v", err)
 					continue
@@ -66,7 +70,11 @@ func (l *CronTask) sendMsg() {
 			} else {
 				if allContact { // 所有联系人
 					// 获取 contact 表中  wx_wxid 等于 req.Fromwxid 的 type 为1的数据
-					userList, err = l.svcCtx.DB.Contact.Query().Where(contact.WxWxid(batch.Fromwxid), contact.TypeEQ(1)).All(l.ctx)
+					userList, err = l.svcCtx.DB.Contact.Query().Where(
+						contact.WxWxid(batch.Fromwxid),
+						contact.TypeEQ(1),
+						contact.Ctype(1),
+					).All(l.ctx)
 					if err != nil {
 						l.Logger.Errorf("userList err: %v", err)
 						continue
@@ -81,7 +89,11 @@ func (l *CronTask) sendMsg() {
 
 				if allGroup { //所有群
 					// 获取 contact 表中  wx_wxid 等于 req.Fromwxid 的 type 为2的数据
-					groupList, err = l.svcCtx.DB.Contact.Query().Where(contact.WxWxid(batch.Fromwxid), contact.TypeEQ(2)).All(l.ctx)
+					groupList, err = l.svcCtx.DB.Contact.Query().Where(
+						contact.WxWxid(batch.Fromwxid),
+						contact.TypeEQ(2),
+						contact.Ctype(1),
+					).All(l.ctx)
 					if err != nil {
 						l.Logger.Errorf("groupList err: %v", err)
 						continue
@@ -125,6 +137,7 @@ func (l *CronTask) sendMsg() {
 							SetMsgtype(int32(msgItem.Type)).
 							SetNotNilMsg(&msgItem.Content).
 							SetStatus(0).
+							SetCtype(1).
 							SetNotNilBatchNo(&batch.BatchNo)
 
 						msgs = append(msgs, msgRow)
@@ -168,7 +181,11 @@ func (l *CronTask) sendMsg() {
 		}
 
 		// 获取当前批次的所有待发送消息
-		msglist, err := l.svcCtx.DB.Msg.Query().Where(msg.BatchNoEQ(batch.BatchNo), msg.StatusEQ(0)).All(l.ctx)
+		msglist, err := l.svcCtx.DB.Msg.Query().Where(
+			msg.BatchNoEQ(batch.BatchNo),
+			msg.StatusEQ(0),
+			msg.Ctype(1),
+		).All(l.ctx)
 		if err != nil {
 			l.Logger.Errorf("msglist err: %v", err)
 			continue
@@ -292,7 +309,12 @@ func (l *CronTask) getContactList(labels []uint64, fromWxId string, stype int) (
 
 	if len(contact_ids) > 0 {
 		// 获取 contact 表中 wx_wxid 等于 req.Fromwxid 并且 id 等于 contact_ids 并且 type 为1或2 的数据
-		userList, err = l.svcCtx.DB.Contact.Query().Where(contact.WxWxid(fromWxId), contact.IDIn(contact_ids...), contact.TypeEQ(stype)).All(l.ctx)
+		userList, err = l.svcCtx.DB.Contact.Query().Where(
+			contact.WxWxid(fromWxId),
+			contact.IDIn(contact_ids...),
+			contact.TypeEQ(stype),
+			contact.Ctype(1),
+		).All(l.ctx)
 		if err != nil {
 			return nil, dberrorhandler.DefaultEntError(l.Logger, err, nil)
 		}

crontask/send_wx.go

@@ -100,7 +100,11 @@ func (l *CronTask) sendWx() {
 
 	// 从 Redis 中获取服务器信息(ip, port)
 	getContactInfo := func(botwxid string, wxid string) (*ent.Contact, error) {
-		contactInfo, _ := l.svcCtx.DB.Contact.Query().Where(contact.WxWxidEQ(botwxid), contact.WxidEQ(wxid)).First(l.ctx)
+		contactInfo, _ := l.svcCtx.DB.Contact.Query().Where(
+			contact.WxWxidEQ(botwxid),
+			contact.WxidEQ(wxid),
+			contact.Ctype(1),
+		).First(l.ctx)
 
 		return contactInfo, nil
 	}

crontask/send_wx_on_timeout.go

@@ -141,7 +141,11 @@ func (l *CronTask) sendWxOnTimeout() {
 								if message.Meta != nil {
 									meta.Filename = message.Meta.Filename
 								}
-								contactInfo, _ := l.svcCtx.DB.Contact.Query().Where(contact.WxWxidEQ(s.BotWxid), contact.WxidEQ(s.ContactWxid)).First(l.ctx)
+								contactInfo, _ := l.svcCtx.DB.Contact.Query().Where(
+									contact.WxWxidEQ(s.BotWxid),
+									contact.WxidEQ(s.ContactWxid),
+									contact.Ctype(1),
+								).First(l.ctx)
 								content := varReplace(message.Content, contactInfo)
 								_, _ = l.svcCtx.DB.MessageRecords.Create().
 									SetBotWxid(s.BotWxid).

internal/logic/label_relationship/set_whatsapp_contact_batch_label_logic.go

@@ -85,6 +85,7 @@ func (l *SetWhatsappContactBatchLabelLogic) SetWhatsappContactBatchLabel(req *ty
 					SetLabelID(id).
 					SetContactID(contactId).
 					SetOrganizationID(organizationId).
+					SetCtype(2).
 					Save(l.ctx)
 				//if err != nil {
 				//	return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)

internal/logic/label_relationship/set_whatsapp_contact_label_logic.go

@@ -85,6 +85,7 @@ func (l *SetWhatsappContactLabelLogic) SetWhatsappContactLabel(req *types.LabelR
 			SetLabelID(id).
 			SetContactID(*req.ContactId).
 			SetOrganizationID(organizationId).
+			SetCtype(2).
 			Save(l.ctx)
 		//if err != nil {
 		//	return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)