package handler import ( "fmt" "github.com/gocarina/gocsv" "github.com/golang-module/carbon/v2" "github.com/labstack/echo/v4" "net/http" "os" "zsxyww.com/scheduler/config" "zsxyww.com/scheduler/model" "zsxyww.com/scheduler/signals" ) func GetAssignment(i echo.Context) error { if (carbon.Now().ToDateString() != signals.Table.LastUpdated.ToDateString()) || signals.Table.NeedUpdate == true { fmt.Printf("At %v:start regenerate table", carbon.Now()) data, err := generateTable() if err != nil { i.String(http.StatusInternalServerError, err.Error()) return echo.ErrInternalServerError } i.Render(http.StatusOK, "table.html", data) return nil } return nil } func generateTable() (*[7][]string, error) { table := [7][]string{} //结果放入这里 members := []*model.Member{} //包含所有成员信息的切片 err := readTableData(&members) if err != nil { return nil, err } dayOfWeek := carbon.Now().DayOfWeek() //今天星期几 today := []*model.Member{} //今天值班的人 female := []*model.Member{} //今天的女生 male := []*model.Member{} //今天的男生 for _, i := range members { if i.FreeDay == dayOfWeek { today = append(today, i) } } for _, i := range today { if i.Sex == false { female = append(female, i) } else if i.Sex == true { male = append(male, i) } } //为女生分配负责人 for c, i := range female { if i.Access < model.FRESH { //是正式成员 table[c%4] = append(table[c%4], i.Name) //轮流分配到女生片区 i.Arranged = true } } //分配剩下的所有女生到女生片区 for c, i := range female { if i.Arranged != true { table[c%4] = append(table[c%4], i.Name) i.Arranged = true } } //为男生分配负责人 for c, i := range male { if i.Access < model.FRESH { table[(c%3)+4] = append(table[(c%3)+4], i.Name) i.Arranged = true } } //分配剩下的所有男生 for c, i := range male { if i.Arranged == false { table[c%7] = append(table[c%7], i.Name) } } fmt.Printf("today:%v\n", today) fmt.Printf("table:%v\n", table) return &table, nil } func readTableData(m *[]*model.Member) error { data, err := os.OpenFile(config.File, os.O_RDWR|os.O_CREATE, os.ModePerm) if err != nil { return err } defer data.Close() err = gocsv.UnmarshalFile(data, m) if err != nil { return err } for index, member := range *m { fmt.Printf("%v:%v\n", index, member) // for debug concerns } return nil }