修改

handler/assignments.go

@@ -0,0 +1,104 @@
+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())
+		err := generateTable()
+		if err != nil {
+			i.String(http.StatusInternalServerError, err.Error())
+			return echo.ErrInternalServerError
+		}
+		return nil
+	}
+	return nil
+}
+func generateTable() error {
+
+	table := [7][]string{}       //结果放入这里
+	members := []*model.Member{} //包含所有成员信息的切片
+
+	err := readTableData(&members)
+	if err != nil {
+		return 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 i := 0; i < 4; i++ {
+		for _, x := range female {
+			if (x.Arranged == false) && (x.Access < model.FRESH) {
+				table[i] = append(table[i], x.Name)
+				x.Arranged = true
+			}
+		}
+	}
+	//男生负责人：
+	for i := 0; i < 7; i++ {
+		for _, x := range male {
+			if (x.Arranged == false) && (x.Access < model.FRESH) {
+				table[i] = append(table[i], x.Name)
+				x.Arranged = true
+			}
+		}
+	}
+	//女生成员
+	for c, i := range female {
+		if i.Arranged == false {
+			table[(c % 4)] = append(table[(c%4)], i.Name)
+		}
+	}
+	//男生成员
+	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 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
+}