is_recommend

controllers/book/book.go

@@ -6,6 +6,7 @@ import (
 	"spaceofcheng/services/models/model"
 	"spaceofcheng/services/service/book"
 	"strings"
+	"time"
 
 	"github.com/tealeg/xlsx"
 )
@@ -289,12 +290,13 @@ func (c *BookController) GetBookById() {
 	c.ResponseJSON(list)
 }
 
-// type Kernel struct {
-// 	schedules map[string]ScheduleTask
-// }
+func (c *BookController) SendMessageToCustomer() time.Duration {
+	var timeNext time.Duration
+	return timeNext
+}
 
-// func NewKernel() *Kernel {
-// 	return &Kernel{
-// 		schedules: make(map[string]ScheduleTask),
-// 	}
+// func (c *BookController) ScheduleTask() {
+// 	kernel := utils.NewKernel()
+// 	kernel.SetTask("refresh-token", SendMessageToCustomer)
+// 	kernel.StartTask("refresh-token")
 // }

models/book/book.go

@@ -97,6 +97,7 @@ func (m *BookDAO) EditBook(book model.TaBook) (*model.TaBook, error) {
 		"publisher",
 		"price",
 		"book_description",
+		"is_recommend",
 	}
 	_, err := m.db.Cols(cols...).Where("book_id = ?", book.BookId).Update(book)
 	return &book, err
@@ -431,13 +432,20 @@ func (m *BookDAO) GetCustomerByCustomerInfo(customerInfo string) (*model.TaCusto
 
 }
 
+type TimeList struct {
+	model.TaBookBorrowRecord `xorm:"extends"`
+	BookName                 string
+}
+
 // GetAllOneWeekRecord 获取时间还剩一周过期的借阅记录
-func (m *BookDAO) GetAllOneWeekRecord() ([]model.TaBookBorrowRecord, error) {
-	var record []model.TaBookBorrowRecord
+func (m *BookDAO) GetAllOneWeekRecord() ([]TimeList, error) {
+	var record []TimeList
 	sql := `SELECT
-	a.* 
+	a.*,
+	b.book_name
 FROM
 	ta_book_borrow_record a 
+	inner join ta_book b on a.book_id = b.book_id
 WHERE
 	DATE_FORMAT( a.end_date, "%Y-%m-%d" ) = DATE_FORMAT( ( SELECT date_add( now( ), INTERVAL 1 WEEK ) ), "%Y-%m-%d" )
 	and a.borrow_status = '` + models.BORROW_TYPE_BORROWED + `'`
@@ -446,12 +454,14 @@ WHERE
 }
 
 // GetAllLateRecord 获取过期两天的借阅记录
-func (m *BookDAO) GetAllLateRecord() ([]model.TaBookBorrowRecord, error) {
-	var record []model.TaBookBorrowRecord
+func (m *BookDAO) GetAllLateRecord() ([]TimeList, error) {
+	var record []TimeList
 	sql := `SELECT
-	a.* 
+	a.*,
+	b.book_name
 FROM
 	ta_book_borrow_record a 
+	inner join ta_book b on a.book_id = b.book_id
 WHERE
 	DATE_FORMAT( now(), "%Y-%m-%d" ) = DATE_FORMAT( ( SELECT date_add( a.end_date, INTERVAL 2 DAY ) ), "%Y-%m-%d" )
 	and a.borrow_status = '` + models.BORROW_TYPE_LATE + `'`
@@ -470,3 +480,13 @@ func (m *BookDAO) DeleteBook(bookId string) error {
 	_, err := m.db.Cols(cols...).Where("book_id = ?", book.BookId).Update(book)
 	return err
 }
+
+func (m *BookDAO) GetUserMappingById(customerId string) (*model.TaUserMapping, error) {
+	var userMap []model.TaUserMapping
+	sql := `select * from ta_user_mapping where user_id = '` + customerId + `'`
+	err := m.db.Sql(sql).Find(&userMap)
+	if len(userMap) <= 0 {
+		return nil, err
+	}
+	return &userMap[0], err
+}

service/book/book.go

@@ -395,5 +395,64 @@ func (s *BookServ) DeleteBook(bookId string) error {
 		return errors.New("删除失败")
 	}
 	return nil
+}
 
+func (s *BookServ) SendMessageToCustomer() error {
+	weekList, err := s.dao.GetAllOneWeekRecord()
+	if err != nil {
+		utils.LogError("发送消息失败失败: " + err.Error())
+		return errors.New("发送消息失败失败")
+	}
+	lateList, err := s.dao.GetAllLateRecord()
+	if err != nil {
+		utils.LogError("发送消息失败失败: " + err.Error())
+		return errors.New("发送消息失败失败")
+	}
+	if len(weekList) > 0 {
+		for i := 0; i < len(weekList); i++ {
+			userMap, err := s.dao.GetUserMappingById(weekList[i].CustomerId)
+			if err != nil {
+				utils.LogError("发送消息失败失败: " + err.Error())
+				return errors.New("发送消息失败失败")
+			}
+			message := utils.Message{
+				To: utils.ClientID{ID: userMap.Openid},
+				Data: map[string]interface{}{
+					"orgid": weekList[i].OrgId,
+					"tplid": "",
+					"data": map[string]string{
+						"first":    "还书提醒",
+						"keyword1": weekList[i].BookName,
+						"keyword2": weekList[i].EndDate.Format("2006-01-02"),
+						// "remark":   fmt.Sprintf(remarkTpl, recored.ValidateEnd.Format("2006-01-02"), caseName),
+					},
+				},
+			}
+			go utils.SendWechat(message)
+		}
+	}
+	if len(lateList) > 0 {
+		for i := 0; i < len(lateList); i++ {
+			userMap, err := s.dao.GetUserMappingById(lateList[i].CustomerId)
+			if err != nil {
+				utils.LogError("发送消息失败失败: " + err.Error())
+				return errors.New("发送消息失败失败")
+			}
+			message := utils.Message{
+				To: utils.ClientID{ID: userMap.Openid},
+				Data: map[string]interface{}{
+					"orgid": lateList[i].OrgId,
+					"tplid": "",
+					"data": map[string]string{
+						"first":    "逾期提醒",
+						"keyword1": lateList[i].BookName,
+						"keyword2": lateList[i].EndDate.Format("2006-01-02"),
+						// "remark":   fmt.Sprintf(remarkTpl, recored.ValidateEnd.Format("2006-01-02"), caseName),
+					},
+				},
+			}
+			go utils.SendWechat(message)
+		}
+	}
+	return nil
 }

utils/core.go

@@ -0,0 +1,63 @@
+package utils
+
+import (
+	"time"
+)
+
+// API API 接口信息
+type API struct {
+	Method       string
+	URI          string
+	ResponseType string
+}
+
+// Kernel 默认核心类
+type Kernel struct {
+	schedules map[string]ScheduleTask
+}
+
+const (
+	// ResponseJSON 返回类型 json
+	ResponseJSON = "json"
+
+	// ResponseXML 返回类型 xml
+	ResponseXML = "xml"
+)
+
+// NewKernel 初始化
+func NewKernel() *Kernel {
+	return &Kernel{
+		schedules: make(map[string]ScheduleTask),
+	}
+}
+
+// SetScheduleTask 设置定时任务
+func (t *Kernel) SetScheduleTask(id string, schedule ScheduleTask) {
+	t.schedules[id] = schedule
+}
+
+// SetTask 设置任务
+func (t *Kernel) SetTask(id string, task Task) {
+	schedule, ok := t.schedules[id]
+	if !ok {
+		schedule = NewDefaultScheduleServer()
+		t.schedules[id] = schedule
+	}
+
+	taskServer, _ := schedule.(*DefaultScheduleServer)
+	taskServer.SetTask(task)
+}
+
+// StartTask 启动
+func (t *Kernel) StartTask(id string, delay ...time.Duration) {
+	if schedule, ok := t.schedules[id]; ok {
+		schedule.Start(delay...)
+	}
+}
+
+// StopTask 关闭任务
+func (t *Kernel) StopTask(id string) {
+	if schedule, ok := t.schedules[id]; ok {
+		schedule.Stop()
+	}
+}

utils/scheduleTask.go

@@ -0,0 +1,75 @@
+package utils
+
+import (
+	"time"
+)
+
+// ScheduleTask 定时任务
+type ScheduleTask interface {
+	// Start 启动服务
+	Start(delay ...time.Duration)
+
+	// Stop 停止服务
+	Stop()
+}
+
+// Task 任务, 返回下次执行的时间间隔
+type Task func() time.Duration
+
+// DefaultScheduleServer 默认 access_token 获取服务
+type DefaultScheduleServer struct {
+	done chan bool
+	task Task
+}
+
+// NewDefaultScheduleServer 构造默认 Token Server
+func NewDefaultScheduleServer() *DefaultScheduleServer {
+	return &DefaultScheduleServer{
+		done: make(chan bool),
+	}
+}
+
+// SetTask 设置定时任务
+func (t *DefaultScheduleServer) SetTask(task Task) {
+	t.task = task
+}
+
+// Stop 停止服务
+func (t *DefaultScheduleServer) Stop() {
+	t.done <- true
+}
+
+// Start 启动服务
+func (t *DefaultScheduleServer) Start(delay ...time.Duration) {
+	go t.start(delay...)
+}
+
+/**
+* 定时任务
+* 不依赖文件系统 或者 DB, 依据每次调用反馈的结果，进行下一次任务定时设置
+ */
+
+func (t *DefaultScheduleServer) start(delay ...time.Duration) {
+	go func() {
+		if delay != nil && len(delay) > 0 {
+			time.Sleep(delay[0])
+		}
+
+		d := t.task() // 上次任务, 返回下一次的执行周期
+		for {
+			time.Sleep(d)
+			d = t.task()
+		}
+	}()
+
+	for {
+		select {
+		case done := <-t.done:
+			if done {
+				return
+			}
+		}
+	}
+}
+
+var _ ScheduleTask = &DefaultScheduleServer{}