优化机器人推送逻辑

controllers/sms.go

@@ -56,7 +56,7 @@ func (this *SmsController) SmsSend() {
 			return
 		}
 		//判断记录是否存在
-		if count <= 0 && !strings.Contains(v.Body, "测试") {
+		if count <= 0 {
 			var tm time.Time
 			if v.Date != "" {
 				smsDateInt64, _ := strconv.ParseInt(v.Date, 10, 64)
@@ -65,7 +65,11 @@ func (this *SmsController) SmsSend() {
 			record := new(models.SmsRecord)
 			record.CreateTime = time.Now()
 			record.ModifyTime = time.Now()
-			record.IsSend = 0
+			if strings.Contains(v.Body, "测试") {
+				record.IsSend = 1
+			} else {
+				record.IsSend = 0
+			}
 			record.SmsDateTime = tm
 			record.Mobile = v.Address
 			record.SmsDate = v.Date

models/sms_record.go

@@ -17,17 +17,17 @@ type SmsRecord struct {
 	UnionCode   string
 }
 
-//新增记录
+// 新增记录
 func AddSmsRecord(item *SmsRecord) (newId int64, err error) {
 	o := orm.NewOrm()
 	newId, err = o.Insert(item)
 	return
 }
 
-//批量新增记录
+// 批量新增记录
 func AddSmsRecordMulti(items []*SmsRecord) (newId int64, err error) {
 	o := orm.NewOrm()
-	newId, err = o.InsertMulti(1, items)
+	newId, err = o.InsertMulti(len(items), items)
 	return
 }
 
@@ -61,16 +61,16 @@ type SmsReq struct {
 	Sign    string
 }
 
-func ModifySmsRecordUnicode(unionCode string,id int) (err error) {
+func ModifySmsRecordUnicode(unionCode string, id int) (err error) {
 	o := orm.NewOrm()
 	sql := `UPDATE sms_record SET union_code=? WHERE id=? `
 	_, err = o.Raw(sql, unionCode, id).Exec()
 	return
 }
 
-func GetSmsRecordList()(list []*SmsRecord,err error)  {
+func GetSmsRecordList() (list []*SmsRecord, err error) {
 	o := orm.NewOrm()
 	sql := `SELECT * FROM sms_record WHERE union_code='' `
 	_, err = o.Raw(sql).QueryRows(&list)
 	return
-}
+}