基于Golang与MongoDB的日期范围及特殊周一日程处理咨询

一、用Golang找出指定日期范围内的所有周一 #

咱们可以直接借助Go标准库的time包来遍历日期范围，精准筛选出所有周一。这里要注意Go的time.Weekday枚举值：周一对应time.Monday（值为1，周日是0），判断的时候别搞混就行。

下面是完整的可运行代码示例：

package main

import (
	"fmt"
	"time"
)

func getMondaysInRange(startStr, endStr string) ([]time.Time, error) {
	// 定义日期解析格式（Go的参考时间固定为2006-01-02）
	layout := "2006-01-02"
	startDate, err := time.Parse(layout, startStr)
	if err != nil {
		return nil, fmt.Errorf("解析起始日期失败: %v", err)
	}
	endDate, err := time.Parse(layout, endStr)
	if err != nil {
		return nil, fmt.Errorf("解析结束日期失败: %v", err)
	}

	var mondays []time.Time
	currentDate := startDate

	// 先把起始日期调整到范围内的第一个周一（如果起始日不是周一的话）
	for currentDate.Weekday() != time.Monday {
		currentDate = currentDate.AddDate(0, 0, 1)
		// 极端情况：起始日已经在结束日之后，直接返回空列表
		if currentDate.After(endDate) {
			return mondays, nil
		}
	}

	// 每周加7天，收集所有不超过结束日期的周一
	for !currentDate.After(endDate) {
		mondays = append(mondays, currentDate)
		currentDate = currentDate.AddDate(0, 0, 7)
	}

	return mondays, nil
}

func main() {
	start := "2018-05-01"
	end := "2018-07-14"
	mondays, err := getMondaysInRange(start, end)
	if err != nil {
		fmt.Printf("错误: %v\n", err)
		return
	}

	fmt.Println("范围内的所有周一：")
	for _, m := range mondays {
		fmt.Println(m.Format("2006-01-02"))
	}
}

运行这段代码后，会输出2018年5月1日到7月14日之间的所有周一，比如2018-05-07、2018-05-14直到2018-07-13。

二、MongoDB Schema调整与特殊日程存储 #

现有Schema的小问题分析 #

你当前的Schedule字段已经覆盖了基础需求，但有几个可以优化的点：

1. Id用int可能会遇到扩容和唯一性冲突的问题，MongoDB原生的ObjectId（对应Go的primitive.ObjectID）更适合做主键，自带唯一性和时间戳信息
2. Start_hours/Start_minutes和Start_date存在冗余——Start_date作为时间戳已经包含了完整的时间信息，拆分出小时分钟虽然直观，但会增加维护成本（比如修改时间时要同步更新多个字段）
3. 缺少标识特殊日程的字段，后续查询和区分普通/调整后的日程会很麻烦

优化后的Schema建议 #

我推荐调整后的Schema（用Go结构体对应MongoDB文档）：

import "go.mongodb.org/mongo-driver/bson/primitive"

type Schedule struct {
	ID          primitive.ObjectID `bson:"_id,omitempty"` // MongoDB原生主键
	DayOfWeek   string             `bson:"day_of_week"`   // 比如"Monday"，方便按周几筛选
	IsSpecial   bool               `bson:"is_special"`    // 标记是否为特殊调整的日程
	StartTime   time.Time          `bson:"start_time"`    // 完整时间戳（包含日期+时分）
	EndTime     time.Time          `bson:"end_time"`      // 同上
	CreatedAt   time.Time          `bson:"created_at"`    // 可选，记录创建时间
}

这样调整的好处：

• 用ObjectID做主键，符合MongoDB最佳实践
• 合并日期和时间到StartTime/EndTime，避免冗余，查询时也能直接按时间范围筛选
• 新增IsSpecial字段，快速区分普通和特殊日程，后续查询特殊周一的日程效率更高

存储日程的代码示例 #

假设咱们已经通过前面的函数拿到了周一列表，现在插入前两个普通日程和第三个特殊日程：

package main

import (
	"context"
	"fmt"
	"time"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

type Schedule struct {
	ID          primitive.ObjectID `bson:"_id,omitempty"`
	DayOfWeek   string             `bson:"day_of_week"`
	IsSpecial   bool               `bson:"is_special"`
	StartTime   time.Time          `bson:"start_time"`
	EndTime     time.Time          `bson:"end_time"`
	CreatedAt   time.Time          `bson:"created_at"`
}

func main() {
	// 连接MongoDB（替换成你的数据库地址）
	client, err := mongo.Connect(context.TODO(), options.Client().ApplyURI("mongodb://localhost:27017"))
	if err != nil {
		panic(err)
	}
	defer client.Disconnect(context.TODO())

	// 获取Schedule集合（替换成你的数据库名）
	coll := client.Database("your_db_name").Collection("Schedule")

	// 获取范围内的周一列表
	mondays, _ := getMondaysInRange("2018-05-01", "2018-07-14")

	// 插入前两个普通周一的日程
	for i := 0; i < 2; i++ {
		if i >= len(mondays) {
			break // 防止数组越界
		}
		monday := mondays[i]
		// 构造当天8:00-18:00的时间
		startTime := time.Date(monday.Year(), monday.Month(), monday.Day(), 8, 0, 0, 0, monday.Location())
		endTime := time.Date(monday.Year(), monday.Month(), monday.Day(), 18, 0, 0, 0, monday.Location())

		schedule := Schedule{
			DayOfWeek: "Monday",
			IsSpecial: false,
			StartTime: startTime,
			EndTime:   endTime,
			CreatedAt: time.Now(),
		}

		_, err := coll.InsertOne(context.TODO(), schedule)
		if err != nil {
			fmt.Printf("插入第%d个日程失败: %v\n", i+1, err)
		} else {
			fmt.Printf("第%d个日程插入成功\n", i+1)
		}
	}

	// 插入第三个特殊周一的日程
	if len(mondays) >= 3 {
		monday := mondays[2]
		// 构造当天9:00-17:00的时间
		startTime := time.Date(monday.Year(), monday.Month(), monday.Day(), 9, 0, 0, 0, monday.Location())
		endTime := time.Date(monday.Year(), monday.Month(), monday.Day(), 17, 0, 0, 0, monday.Location())

		schedule := Schedule{
			DayOfWeek: "Monday",
			IsSpecial: true,
			StartTime: startTime,
			EndTime:   endTime,
			CreatedAt: time.Now(),
		}

		_, err := coll.InsertOne(context.TODO(), schedule)
		if err != nil {
			fmt.Printf("插入特殊日程失败: %v\n", err)
		} else {
			fmt.Println("特殊日程插入成功")
		}
	}
}

可选：兼容原Schema的方案 #

如果你不想大幅修改现有结构，也可以直接新增一个IsSpecial布尔字段来标记特殊日程，前两个日程设为false，第三个设为true即可，这样既能兼容原有数据，又能满足特殊日程的区分需求。

内容的提问来源于stack exchange，提问作者gourav