44 lines
950 B
Go
44 lines
950 B
Go
|
package grpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"log/slog"
|
||
|
|
|
||
|
|
pb "schedule_converter/proto/runner"
|
||
|
|
|
||
|
|
"schedule_converter/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (h *TaskHandler) getSchedule(ctx context.Context, payload []byte) ([]byte, error) {
|
||
|
|
var req pb.GetSchedulePayload
|
||
|
|
if err := unmarshalPayload(payload, &req); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
log := slog.With("username", req.Username, "semester", req.Semester)
|
||
|
|
log.Info("获取课表任务")
|
||
|
|
|
||
|
|
httpClient, err := createHTTPClient()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
cfg := config.Get()
|
||
|
|
setTWFIDCookie(httpClient, cfg.BaseURL, "")
|
||
|
|
|
||
|
|
courses, err := executeLoginAndFetchSchedule(ctx, httpClient, req.Username, req.Password, req.Semester)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result := &pb.ScheduleResultData{
|
||
|
|
StudentId: req.Username,
|
||
|
|
Semester: req.Semester,
|
||
|
|
Courses: convertCourses(courses),
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Info("课表获取成功", "course_count", len(courses))
|
||
|
|
return json.Marshal(result)
|
||
|
|
}
|