新增统计睡眠时间和添加协程里的异常捕获 (#84)

* fix:捕获协程异常 * feat:添加睡眠时间 * feat:添加睡眠管理功能 * fix:修改对标时间 * fix:修改注释函数名，使用一个时间 * feat:让大家能够更加清楚的看到每一个功能 * feat:把群管放在control,增加退群信息 * fix:不会recover * fix:原本的正则限制机器人回复回复的消息，故去掉正则 * fix:添加服务详情,整理manager * feat:chat进control Co-authored-by: Guohuiyuan <haibaraguo@yeahka.com>
2026-02-12 02:00:24 +00:00 · 2021-12-04 20:46:24 +08:00
parent d0d07604e5
commit 793cac09cb
15 changed files with 339 additions and 77 deletions
--- a/plugin_sleep_manage/data.go
+++ b/plugin_sleep_manage/data.go
@@ -0,0 +1,24 @@
+package plugin_sleep_manage
+
+import (
+	"github.com/FloatTech/ZeroBot-Plugin/plugin_sleep_manage/model"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/process"
+	log "github.com/sirupsen/logrus"
+	"os"
+)
+
+func init() {
+	go func() {
+		defer func() {
+			//recover() //可以打印panic的错误信息
+			if err := recover(); err != nil { //产生了panic异常
+				log.Println(err)
+			}
+
+		}() //别忘了(), 调用此匿名函数
+		process.SleepAbout1sTo2s()
+		_ = os.MkdirAll(dbpath, 0755)
+		model.Initialize(dbfile)
+	}()
+
+}
--- a/plugin_sleep_manage/model/model.go
+++ b/plugin_sleep_manage/model/model.go
@@ -0,0 +1,109 @@
+package model
+
+import (
+	"github.com/jinzhu/gorm"
+	_ "github.com/logoove/sqlite"
+	log "github.com/sirupsen/logrus"
+	"os"
+	"time"
+)
+
+type SleepDB gorm.DB
+
+func Initialize(dbpath string) *SleepDB {
+	var err error
+	if _, err = os.Stat(dbpath); err != nil || os.IsNotExist(err) {
+		// 生成文件
+		f, err := os.Create(dbpath)
+		if err != nil {
+			return nil
+		}
+		defer f.Close()
+	}
+	gdb, err := gorm.Open("sqlite3", dbpath)
+	if err != nil {
+		panic(err)
+	}
+	gdb.AutoMigrate(&SleepManage{})
+	return (*SleepDB)(gdb)
+}
+
+func Open(dbpath string) (*SleepDB, error) {
+	db, err := gorm.Open("sqlite3", dbpath)
+	if err != nil {
+		return nil, err
+	} else {
+		return (*SleepDB)(db), nil
+	}
+}
+
+func (sdb *SleepDB) Close() error {
+	db := (*gorm.DB)(sdb)
+	return db.Close()
+}
+
+type SleepManage struct {
+	gorm.Model
+	GroupId   int64     `gorm:"column:group_id"`
+	UserId    int64     `gorm:"column:user_id"`
+	SleepTime time.Time `gorm:"column:sleep_time"`
+}
+
+func (SleepManage) TableName() string {
+	return "sleep_manage"
+}
+
+// 更新睡眠时间
+func (sdb *SleepDB) Sleep(groupId, userId int64) (position int, awakeTime time.Duration) {
+	db := (*gorm.DB)(sdb)
+	now := time.Now()
+
+	today := now.Add(-time.Hour*time.Duration(3+now.Hour()) - time.Minute*time.Duration(now.Minute()) - time.Second*time.Duration(now.Second()))
+	st := SleepManage{
+		GroupId:   groupId,
+		UserId:    userId,
+		SleepTime: now,
+	}
+	if err := db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", groupId, userId).First(&st).Error; err != nil {
+		// error handling...
+		if gorm.IsRecordNotFoundError(err) {
+			db.Debug().Model(&SleepManage{}).Create(&st) // newUser not user
+		}
+	} else {
+		log.Println("sleeptime为", st)
+		awakeTime = now.Sub(st.SleepTime)
+		db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", groupId, userId).Update(
+			map[string]interface{}{
+				"sleep_time": now,
+			})
+	}
+	db.Debug().Model(&SleepManage{}).Where("group_id = ? and sleep_time <= ? and sleep_time >= ?", groupId, now, today).Count(&position)
+	return position, awakeTime
+}
+
+// 更新起床时间
+func (sdb *SleepDB) GetUp(groupId, userId int64) (position int, sleepTime time.Duration) {
+	db := (*gorm.DB)(sdb)
+	now := time.Now()
+	today := now.Add(-time.Hour*time.Duration(-6+now.Hour()) - time.Minute*time.Duration(now.Minute()) - time.Second*time.Duration(now.Second()))
+	st := SleepManage{
+		GroupId:   groupId,
+		UserId:    userId,
+		SleepTime: now,
+	}
+	if err := db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", groupId, userId).First(&st).Error; err != nil {
+		// error handling...
+		if gorm.IsRecordNotFoundError(err) {
+			db.Debug().Model(&SleepManage{}).Create(&st) // newUser not user
+		}
+	} else {
+		log.Println("sleeptime为", st)
+		sleepTime = now.Sub(st.SleepTime)
+		db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", groupId, userId).Update(
+			map[string]interface{}{
+				"get_up_time": now,
+			})
+	}
+	db.Debug().Model(&SleepManage{}).Where("group_id = ? and sleep_time <= ? and sleep_time >= ?", groupId, now, today).Count(&position)
+	return position, sleepTime
+}
--- a/plugin_sleep_manage/sleep_manage.go
+++ b/plugin_sleep_manage/sleep_manage.go
@@ -0,0 +1,78 @@
+package plugin_sleep_manage
+
+import (
+	"fmt"
+	"github.com/FloatTech/ZeroBot-Plugin/control"
+	"github.com/FloatTech/ZeroBot-Plugin/plugin_sleep_manage/model"
+	log "github.com/sirupsen/logrus"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"time"
+)
+
+const dbpath = "data/SleepManage/"
+const dbfile = dbpath + "sleepmanage.db"
+const prio = 4
+
+var engine = control.Register("sleepmanage", &control.Options{
+	DisableOnDefault: false,
+	Help:             "sleepmanage\n- 早安\n- 晚安",
+})
+
+func init() {
+	engine.OnFullMatch("早安", isMorning, zero.OnlyGroup).SetBlock(true).SetPriority(prio).
+		Handle(func(ctx *zero.Ctx) {
+			db, err := model.Open(dbfile)
+			if err != nil {
+				log.Errorln(err)
+				return
+			}
+			position, getUpTime := db.GetUp(ctx.Event.GroupID, ctx.Event.UserID)
+			log.Println(position, getUpTime)
+			hour, minute, second := timeDuration(getUpTime)
+			if (hour == 0 && minute == 0 && second == 0) || hour >= 24 {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("早安成功！你是今天第%d个起床的", position)))
+			} else {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("早安成功！你的睡眠时长为%d时%d分%d秒,你是今天第%d个起床的", hour, minute, second, position)))
+			}
+			db.Close()
+
+		})
+	engine.OnFullMatch("晚安", isEvening, zero.OnlyGroup).SetBlock(true).SetPriority(prio).
+		Handle(func(ctx *zero.Ctx) {
+			db, err := model.Open(dbfile)
+			if err != nil {
+				log.Errorln(err)
+				return
+			}
+			position, sleepTime := db.Sleep(ctx.Event.GroupID, ctx.Event.UserID)
+			log.Println(position, sleepTime)
+			hour, minute, second := timeDuration(sleepTime)
+			if (hour == 0 && minute == 0 && second == 0) || hour >= 24 {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("晚安成功！你是今天第%d个睡觉的", position)))
+			} else {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("晚安成功！你的清醒时长为%d时%d分%d秒,你是今天第%d个睡觉的", hour, minute, second, position)))
+			}
+			db.Close()
+
+		})
+}
+
+func timeDuration(time time.Duration) (hour, minute, second int64) {
+	hour = int64(time) / (1000 * 1000 * 1000 * 60 * 60)
+	minute = (int64(time) - hour*(1000*1000*1000*60*60)) / (1000 * 1000 * 1000 * 60)
+	second = (int64(time) - hour*(1000*1000*1000*60*60) - minute*(1000*1000*1000*60)) / (1000 * 1000 * 1000)
+	return hour, minute, second
+}
+
+//只统计6点到12点的早安
+func isMorning(ctx *zero.Ctx) bool {
+	now := time.Now().Hour()
+	return now >= 6 && now <= 12
+}
+
+//只统计21点到凌晨3点的晚安
+func isEvening(ctx *zero.Ctx) bool {
+	now := time.Now().Hour()
+	return now >= 21 || now <= 3
+}