feat(v1.2): 亮暗主题、按尾号快速完成、加急置顶、按套餐分日历、大屏升级
1) 全站亮色 / 暗色 - 管理端与大屏都提供切换按钮(顶栏 ☾/☀、大屏右上角),偏好写 cookie,服务端首屏直出,不闪 - 未选择时跟随系统(prefers-color-scheme);两套配色都改成 CSS 变量,色签、热力、表格、抽屉全部适配 - 大屏默认仍是深色记分板,浅色为纸感背景 2) 工作台 → 当日待完成:按手机尾号快速完成 - 输入尾号回车:唯一匹配直接完成并提示 - 同尾号多客户 / 一单多服务:跳到选择面板,默认全选,可多选后一次完成 - 新增 POST /checkins/quick 与 POST /checkins/batch;抽屉标题改为「未完成数 / 总单数」 3) 加急 - 每条任务右侧可勾选/取消加急;加急任务红 tag + 置顶(抽屉与大屏「今日待打」) - 日历格子出现红点,单据色签加红边;大屏热力格也带红点 - 新增表 task_flags(sub_id, service_date, urgent),按 订阅+日期 存,避免和打卡状态耦合 4) 套餐「在日历中独立显示」 - 套餐表单新增勾选项,列表页显示「独立日历」标记 - 勾选后工作台与大屏顶部出现套餐标签页(全部 / 每日任务 / …),可只看某类委托的排单、热量与待办 - 新增列 packages.calendar_separate;老库打开时自动 ALTER,无需手工迁移 5) 大屏 - 热力默认改为「每日待完成单量」,右上角可切回「到期热力」 - 鼠标悬停任意日期格,浮出当日待办明细(客户、套餐、剩余、加急),数据来自 GET /api/screen/day - 底部面板改为「今日到期」(左)+「今日待打」(右,加急置顶),最近追加改为最底部细条 数据库兼容性: - 只做加法:新增 task_flags 表与 packages.calendar_separate 列,原有表结构、字段、接口与已有数据均不变 - 迁移可重复执行(CREATE TABLE IF NOT EXISTS + PRAGMA 检查后 ALTER) 测试: - 新增单元测试:套餐独立日历标记、加急增删查、旧库自动加列与重复迁移 - 新增功能冒烟 33 项(主题 cookie、套餐标签、快速完成单/多匹配、加急排序、大屏接口)全部通过 - 原有 119 项验收回归全绿(3 条断言按新 UI 更新:抽屉标题、热力默认口径、今日待打数取值方式) - go vet / go test ./... 通过;版本号 1.2.0
This commit is contained in:
@@ -199,3 +199,36 @@ func (db *DB) CustomerCount() (int, error) {
|
||||
err := db.sql.QueryRow(`SELECT COUNT(*) FROM customers`).Scan(&n)
|
||||
return n, err
|
||||
}
|
||||
|
||||
// CustomersByTail 按手机尾号精确匹配客户;没有精确命中时退回模糊匹配。
|
||||
func (db *DB) CustomersByTail(tail string) ([]Customer, error) {
|
||||
tail = strings.TrimSpace(tail)
|
||||
if tail == "" {
|
||||
return nil, nil
|
||||
}
|
||||
out, err := db.queryCustomers(`SELECT `+customerCols+` FROM customers WHERE tail = ? ORDER BY id`, tail)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(out) > 0 {
|
||||
return out, nil
|
||||
}
|
||||
return db.queryCustomers(`SELECT `+customerCols+` FROM customers WHERE tail LIKE ? ORDER BY id LIMIT 20`, "%"+tail+"%")
|
||||
}
|
||||
|
||||
func (db *DB) queryCustomers(q string, args ...any) ([]Customer, error) {
|
||||
rows, err := db.sql.Query(q, args...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
out := []Customer{}
|
||||
for rows.Next() {
|
||||
c, err := scanCustomer(rows)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out = append(out, *c)
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
@@ -116,6 +116,41 @@ func (db *DB) Migrate() error {
|
||||
return fmt.Errorf("迁移失败: %w\nSQL: %s", err, stmt)
|
||||
}
|
||||
}
|
||||
// 增量列:老库升级时补上(SQLite 没有 ADD COLUMN IF NOT EXISTS)
|
||||
if err := db.ensureColumn("packages", "calendar_separate",
|
||||
"ALTER TABLE packages ADD COLUMN calendar_separate INTEGER NOT NULL DEFAULT 0"); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ensureColumn 在列不存在时执行 ddl,保证迁移可重复执行。
|
||||
func (db *DB) ensureColumn(table, column, ddl string) error {
|
||||
rows, err := db.sql.Query("PRAGMA table_info(" + table + ")")
|
||||
if err != nil {
|
||||
return fmt.Errorf("读取 %s 表结构失败: %w", table, err)
|
||||
}
|
||||
found := false
|
||||
for rows.Next() {
|
||||
var cid int
|
||||
var name, ctype string
|
||||
var notnull, pk int
|
||||
var dflt any
|
||||
if err := rows.Scan(&cid, &name, &ctype, ¬null, &dflt, &pk); err != nil {
|
||||
rows.Close()
|
||||
return err
|
||||
}
|
||||
if strings.EqualFold(name, column) {
|
||||
found = true
|
||||
}
|
||||
}
|
||||
rows.Close()
|
||||
if found {
|
||||
return nil
|
||||
}
|
||||
if _, err := db.sql.Exec(ddl); err != nil {
|
||||
return fmt.Errorf("迁移失败(%s.%s): %w", table, column, err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -146,6 +181,7 @@ var schemaStatements = []string{
|
||||
note TEXT NOT NULL DEFAULT '',
|
||||
archived INTEGER NOT NULL DEFAULT 0,
|
||||
sort_order INTEGER NOT NULL DEFAULT 0,
|
||||
calendar_separate INTEGER NOT NULL DEFAULT 0,
|
||||
created_at TEXT NOT NULL,
|
||||
updated_at TEXT NOT NULL
|
||||
)`,
|
||||
@@ -224,6 +260,16 @@ var schemaStatements = []string{
|
||||
ip TEXT NOT NULL DEFAULT '',
|
||||
created_at TEXT NOT NULL
|
||||
)`,
|
||||
`CREATE TABLE IF NOT EXISTS task_flags (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
sub_id INTEGER NOT NULL REFERENCES subscriptions(id) ON DELETE CASCADE,
|
||||
service_date TEXT NOT NULL,
|
||||
urgent INTEGER NOT NULL DEFAULT 0,
|
||||
created_at TEXT NOT NULL,
|
||||
updated_at TEXT NOT NULL,
|
||||
UNIQUE (sub_id, service_date)
|
||||
)`,
|
||||
`CREATE INDEX IF NOT EXISTS idx_task_flags_date ON task_flags(service_date)`,
|
||||
`CREATE TABLE IF NOT EXISTS settings (
|
||||
key TEXT PRIMARY KEY,
|
||||
value TEXT NOT NULL
|
||||
|
||||
+27
-15
@@ -20,10 +20,12 @@ type Package struct {
|
||||
Note string
|
||||
Archived bool
|
||||
SortOrder int
|
||||
CreatedAt string
|
||||
UpdatedAt string
|
||||
SubCount int
|
||||
ActiveCount int
|
||||
// CalendarSeparate 勾选后在工作台/大屏日历中单独成页
|
||||
CalendarSeparate bool
|
||||
CreatedAt string
|
||||
UpdatedAt string
|
||||
SubCount int
|
||||
ActiveCount int
|
||||
}
|
||||
|
||||
// UnitLabel 返回类型中文名。
|
||||
@@ -37,9 +39,9 @@ func (p *Package) UnitLabel() string {
|
||||
func scanPackage(row interface{ Scan(dest ...any) error }) (*Package, error) {
|
||||
var p Package
|
||||
var price sql.NullInt64
|
||||
var archived int
|
||||
var archived, separate int
|
||||
err := row.Scan(&p.ID, &p.Name, &p.Unit, &p.DefaultAmount, &price, &p.Color, &p.Note,
|
||||
&archived, &p.SortOrder, &p.CreatedAt, &p.UpdatedAt)
|
||||
&archived, &p.SortOrder, &separate, &p.CreatedAt, &p.UpdatedAt)
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return nil, ErrNotFound
|
||||
@@ -47,6 +49,7 @@ func scanPackage(row interface{ Scan(dest ...any) error }) (*Package, error) {
|
||||
return nil, err
|
||||
}
|
||||
p.Archived = archived == 1
|
||||
p.CalendarSeparate = separate == 1
|
||||
if price.Valid {
|
||||
p.PriceCents = int(price.Int64)
|
||||
p.HasPrice = true
|
||||
@@ -56,7 +59,7 @@ func scanPackage(row interface{ Scan(dest ...any) error }) (*Package, error) {
|
||||
|
||||
// ListPackages 列出套餐(含订阅计数)。
|
||||
func (db *DB) ListPackages(includeArchived bool) ([]Package, error) {
|
||||
q := `SELECT p.id,p.name,p.unit,p.default_amount,p.price_cents,p.color,p.note,p.archived,p.sort_order,p.created_at,p.updated_at,
|
||||
q := `SELECT p.id,p.name,p.unit,p.default_amount,p.price_cents,p.color,p.note,p.archived,p.sort_order,p.calendar_separate,p.created_at,p.updated_at,
|
||||
(SELECT COUNT(*) FROM subscriptions s WHERE s.package_id=p.id),
|
||||
(SELECT COUNT(*) FROM subscriptions s WHERE s.package_id=p.id AND s.status<>'closed')
|
||||
FROM packages p`
|
||||
@@ -73,12 +76,13 @@ func (db *DB) ListPackages(includeArchived bool) ([]Package, error) {
|
||||
for rows.Next() {
|
||||
var p Package
|
||||
var price sql.NullInt64
|
||||
var archived int
|
||||
var archived, separate int
|
||||
if err := rows.Scan(&p.ID, &p.Name, &p.Unit, &p.DefaultAmount, &price, &p.Color, &p.Note,
|
||||
&archived, &p.SortOrder, &p.CreatedAt, &p.UpdatedAt, &p.SubCount, &p.ActiveCount); err != nil {
|
||||
&archived, &p.SortOrder, &separate, &p.CreatedAt, &p.UpdatedAt, &p.SubCount, &p.ActiveCount); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
p.Archived = archived == 1
|
||||
p.CalendarSeparate = separate == 1
|
||||
if price.Valid {
|
||||
p.PriceCents = int(price.Int64)
|
||||
p.HasPrice = true
|
||||
@@ -90,7 +94,7 @@ func (db *DB) ListPackages(includeArchived bool) ([]Package, error) {
|
||||
|
||||
// Package 取单个套餐。
|
||||
func (db *DB) Package(id int64) (*Package, error) {
|
||||
return scanPackage(db.sql.QueryRow(`SELECT id,name,unit,default_amount,price_cents,color,note,archived,sort_order,created_at,updated_at
|
||||
return scanPackage(db.sql.QueryRow(`SELECT id,name,unit,default_amount,price_cents,color,note,archived,sort_order,calendar_separate,created_at,updated_at
|
||||
FROM packages WHERE id=?`, id))
|
||||
}
|
||||
|
||||
@@ -108,9 +112,9 @@ func (db *DB) CreatePackage(p *Package) (int64, error) {
|
||||
if p.HasPrice {
|
||||
price = p.PriceCents
|
||||
}
|
||||
res, err := db.sql.Exec(`INSERT INTO packages(name,unit,default_amount,price_cents,color,note,archived,sort_order,created_at,updated_at)
|
||||
VALUES(?,?,?,?,?,?,0,?,?,?)`,
|
||||
p.Name, p.Unit, p.DefaultAmount, price, p.Color, p.Note, p.SortOrder, now, now)
|
||||
res, err := db.sql.Exec(`INSERT INTO packages(name,unit,default_amount,price_cents,color,note,archived,sort_order,calendar_separate,created_at,updated_at)
|
||||
VALUES(?,?,?,?,?,?,0,?,?,?,?)`,
|
||||
p.Name, p.Unit, p.DefaultAmount, price, p.Color, p.Note, p.SortOrder, b2i(p.CalendarSeparate), now, now)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
@@ -130,11 +134,19 @@ func (db *DB) UpdatePackage(p *Package) error {
|
||||
if p.HasPrice {
|
||||
price = p.PriceCents
|
||||
}
|
||||
_, err := db.sql.Exec(`UPDATE packages SET name=?,unit=?,default_amount=?,price_cents=?,color=?,note=?,sort_order=?,updated_at=?
|
||||
WHERE id=?`, p.Name, p.Unit, p.DefaultAmount, price, p.Color, p.Note, p.SortOrder, nowStamp(), p.ID)
|
||||
_, err := db.sql.Exec(`UPDATE packages SET name=?,unit=?,default_amount=?,price_cents=?,color=?,note=?,sort_order=?,calendar_separate=?,updated_at=?
|
||||
WHERE id=?`, p.Name, p.Unit, p.DefaultAmount, price, p.Color, p.Note, p.SortOrder, b2i(p.CalendarSeparate), nowStamp(), p.ID)
|
||||
return err
|
||||
}
|
||||
|
||||
// b2i 把 bool 转成 SQLite 的 0/1。
|
||||
func b2i(v bool) int {
|
||||
if v {
|
||||
return 1
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
// SetPackageArchived 归档/取消归档。
|
||||
func (db *DB) SetPackageArchived(id int64, archived bool) error {
|
||||
v := 0
|
||||
|
||||
@@ -647,3 +647,163 @@ func TestOpenLegacyWALDatabaseKeepsData(t *testing.T) {
|
||||
t.Fatalf("旧库未切到单文件模式: %s", mode)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPackageCalendarSeparate(t *testing.T) {
|
||||
db := newTestDB(t)
|
||||
pkg := mustPackage(t, db, "每日任务", domain.UnitDay, 30)
|
||||
if pkg.CalendarSeparate {
|
||||
t.Fatal("新建套餐默认不应勾选独立日历")
|
||||
}
|
||||
pkg.CalendarSeparate = true
|
||||
if err := db.UpdatePackage(pkg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
got, err := db.Package(pkg.ID)
|
||||
if err != nil || !got.CalendarSeparate {
|
||||
t.Fatalf("独立日历标记未保存: %+v err=%v", got, err)
|
||||
}
|
||||
list, err := db.ListPackages(true)
|
||||
if err != nil || len(list) != 1 || !list[0].CalendarSeparate {
|
||||
t.Fatalf("列表未带独立日历标记: %+v err=%v", list, err)
|
||||
}
|
||||
got.CalendarSeparate = false
|
||||
if err := db.UpdatePackage(got); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if again, _ := db.Package(pkg.ID); again.CalendarSeparate {
|
||||
t.Fatal("取消勾选失败")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTaskUrgentFlags(t *testing.T) {
|
||||
db := newTestDB(t)
|
||||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||||
cid := mustCustomer(t, db, "4321", "老王")
|
||||
sub1, err := db.CreateSub(CreateSubInput{
|
||||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(domain.Today()), Days: 30,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
pkg2 := mustPackage(t, db, "按次", domain.UnitCount, 10)
|
||||
sub2, err := db.CreateSub(CreateSubInput{
|
||||
CustomerID: cid, PackageID: pkg2.ID, StartDate: domain.Date(domain.Today()),
|
||||
TotalCount: 10, EndDate: domain.Date(domain.AddDays(domain.Today(), 30)),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
d1 := domain.Date(domain.Today())
|
||||
d2 := domain.Date(domain.AddDays(domain.Today(), 1))
|
||||
|
||||
if err := db.SetUrgent(sub1.ID, d1, true); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := db.SetUrgent(sub2.ID, d2, true); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
flags, err := db.UrgentFlags(d1, d2)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !flags[sub1.ID][d1] || !flags[sub2.ID][d2] {
|
||||
t.Fatalf("加急标记缺失: %+v", flags)
|
||||
}
|
||||
if ok, _ := db.IsUrgent(sub1.ID, d1); !ok {
|
||||
t.Fatal("IsUrgent 应为 true")
|
||||
}
|
||||
if n, _ := db.CountUrgent(d1, d2); n != 2 {
|
||||
t.Fatalf("加急计数=%d 期望 2", n)
|
||||
}
|
||||
// 重复设置不报错(UPSERT)
|
||||
if err := db.SetUrgent(sub1.ID, d1, true); err != nil {
|
||||
t.Fatalf("重复标记加急失败: %v", err)
|
||||
}
|
||||
// 取消
|
||||
if err := db.SetUrgent(sub1.ID, d1, false); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if ok, _ := db.IsUrgent(sub1.ID, d1); ok {
|
||||
t.Fatal("取消加急后仍为 true")
|
||||
}
|
||||
if n, _ := db.CountUrgent(d1, d2); n != 1 {
|
||||
t.Fatalf("取消后计数=%d 期望 1", n)
|
||||
}
|
||||
// 非法日期
|
||||
if err := db.SetUrgent(sub1.ID, "2026-13-40", true); err == nil {
|
||||
t.Fatal("非法日期应报错")
|
||||
}
|
||||
// 不存在的订阅
|
||||
if err := db.SetUrgent(99999, d1, true); err == nil {
|
||||
t.Fatal("不存在的订阅应报错")
|
||||
}
|
||||
}
|
||||
|
||||
// TestMigrateOldDatabaseAddsColumn 模拟 v1.1 的旧库:packages 表没有 calendar_separate。
|
||||
func TestMigrateOldDatabaseAddsColumn(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
path := filepath.Join(dir, "old.db")
|
||||
dsn := strings.ReplaceAll(path, "\\", "/")
|
||||
old, err := sql.Open("sqlite", dsn)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
stmts := []string{
|
||||
`CREATE TABLE packages (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
name TEXT NOT NULL,
|
||||
unit TEXT NOT NULL,
|
||||
default_amount INTEGER NOT NULL DEFAULT 30,
|
||||
price_cents INTEGER,
|
||||
color TEXT NOT NULL DEFAULT '#2F5D50',
|
||||
note TEXT NOT NULL DEFAULT '',
|
||||
archived INTEGER NOT NULL DEFAULT 0,
|
||||
sort_order INTEGER NOT NULL DEFAULT 0,
|
||||
created_at TEXT NOT NULL,
|
||||
updated_at TEXT NOT NULL
|
||||
)`,
|
||||
`INSERT INTO packages(name,unit,default_amount,color,note,archived,sort_order,created_at,updated_at)
|
||||
VALUES('老套餐','day',30,'#2F5D50','',0,1,'2026-01-01 00:00:00','2026-01-01 00:00:00')`,
|
||||
}
|
||||
for _, s := range stmts {
|
||||
if _, err := old.Exec(s); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
if err := old.Close(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
db, err := Open(path)
|
||||
if err != nil {
|
||||
t.Fatalf("打开旧库失败: %v", err)
|
||||
}
|
||||
defer db.Close()
|
||||
if err := db.Migrate(); err != nil {
|
||||
t.Fatalf("迁移失败: %v", err)
|
||||
}
|
||||
pkg, err := db.Package(1)
|
||||
if err != nil {
|
||||
t.Fatalf("读取老套餐失败: %v", err)
|
||||
}
|
||||
if pkg.Name != "老套餐" || pkg.CalendarSeparate {
|
||||
t.Fatalf("老数据/默认值不对: %+v", pkg)
|
||||
}
|
||||
// 新列可写
|
||||
pkg.CalendarSeparate = true
|
||||
if err := db.UpdatePackage(pkg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if got, _ := db.Package(1); !got.CalendarSeparate {
|
||||
t.Fatal("迁移后新列不可写")
|
||||
}
|
||||
// 新表 task_flags 已建立
|
||||
var tbl string
|
||||
if err := db.sql.QueryRow(`SELECT name FROM sqlite_master WHERE type='table' AND name='task_flags'`).Scan(&tbl); err != nil {
|
||||
t.Fatalf("task_flags 表未建立: %v", err)
|
||||
}
|
||||
// 迁移可重复执行
|
||||
if err := db.Migrate(); err != nil {
|
||||
t.Fatalf("重复迁移失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,67 @@
|
||||
package store
|
||||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"daili/internal/domain"
|
||||
)
|
||||
|
||||
// UrgentFlags 返回 [from,to] 区间内的加急标记,索引为 [subID][date]。
|
||||
// 表里只存“加急=1”的行,未出现即未加急。
|
||||
func (db *DB) UrgentFlags(from, to string) (map[int64]map[string]bool, error) {
|
||||
out := map[int64]map[string]bool{}
|
||||
rows, err := db.sql.Query(`SELECT sub_id, service_date FROM task_flags
|
||||
WHERE urgent=1 AND service_date BETWEEN ? AND ?`, from, to)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
for rows.Next() {
|
||||
var subID int64
|
||||
var date string
|
||||
if err := rows.Scan(&subID, &date); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if out[subID] == nil {
|
||||
out[subID] = map[string]bool{}
|
||||
}
|
||||
out[subID][date] = true
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
// SetUrgent 设置或取消某天某订阅的加急标记。
|
||||
func (db *DB) SetUrgent(subID int64, date string, urgent bool) error {
|
||||
if _, err := domain.ParseDate(date); err != nil {
|
||||
return errors.New("日期格式不正确")
|
||||
}
|
||||
if _, err := db.Sub(subID); err != nil {
|
||||
return errors.New("订阅不存在")
|
||||
}
|
||||
now := nowStamp()
|
||||
if urgent {
|
||||
_, err := db.sql.Exec(`INSERT INTO task_flags(sub_id, service_date, urgent, created_at, updated_at)
|
||||
VALUES(?,?,1,?,?)
|
||||
ON CONFLICT(sub_id, service_date) DO UPDATE SET urgent=1, updated_at=excluded.updated_at`,
|
||||
subID, date, now, now)
|
||||
return err
|
||||
}
|
||||
_, err := db.sql.Exec(`DELETE FROM task_flags WHERE sub_id=? AND service_date=?`, subID, date)
|
||||
return err
|
||||
}
|
||||
|
||||
// IsUrgent 查询单条加急状态。
|
||||
func (db *DB) IsUrgent(subID int64, date string) (bool, error) {
|
||||
var n int
|
||||
err := db.sql.QueryRow(`SELECT COUNT(*) FROM task_flags WHERE sub_id=? AND service_date=? AND urgent=1`,
|
||||
subID, date).Scan(&n)
|
||||
return n > 0, err
|
||||
}
|
||||
|
||||
// CountUrgent 统计区间内加急任务数。
|
||||
func (db *DB) CountUrgent(from, to string) (int, error) {
|
||||
var n int
|
||||
err := db.sql.QueryRow(`SELECT COUNT(*) FROM task_flags WHERE urgent=1 AND service_date BETWEEN ? AND ?`,
|
||||
from, to).Scan(&n)
|
||||
return n, err
|
||||
}
|
||||
Reference in New Issue
Block a user