core v1
This commit is contained in:
96
server/internal/db/active_class.go
Normal file
96
server/internal/db/active_class.go
Normal file
@@ -0,0 +1,96 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"server/internal/models"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func scheduleDayIndex(t time.Time) int {
|
||||
wd := t.Weekday()
|
||||
if wd == time.Sunday {
|
||||
return 6
|
||||
}
|
||||
return int(wd) - 1
|
||||
}
|
||||
|
||||
func parseTimeMinutes(tStr string) int {
|
||||
parts := strings.Split(tStr, ":")
|
||||
if len(parts) != 2 {
|
||||
return 0
|
||||
}
|
||||
hours, _ := strconv.Atoi(parts[0])
|
||||
mins, _ := strconv.Atoi(parts[1])
|
||||
return hours*60 + mins
|
||||
}
|
||||
|
||||
func studentClassIDs(db *gorm.DB, studentRkID int64) []int64 {
|
||||
var classRkIDs []int64
|
||||
_ = db.Model(&models.ClassStudent{}).
|
||||
Where("student_rk_id = ?", studentRkID).
|
||||
Pluck("class_rk_id", &classRkIDs).Error
|
||||
return classRkIDs
|
||||
}
|
||||
|
||||
// FindActiveClassAndPeriodForStudent trả về lớp + ca đang trong khung giờ học.
|
||||
func FindActiveClassAndPeriodForStudent(db *gorm.DB, studentRkID int64) (int64, int) {
|
||||
classRkIDs := studentClassIDs(db, studentRkID)
|
||||
if len(classRkIDs) == 0 {
|
||||
return 0, 0
|
||||
}
|
||||
|
||||
var schedules []models.ClassSchedule
|
||||
if err := db.Where("class_rk_id IN ? AND is_active = ?", classRkIDs, true).Find(&schedules).Error; err != nil || len(schedules) == 0 {
|
||||
return classRkIDs[0], 1
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
day := scheduleDayIndex(now)
|
||||
currMin := now.Hour()*60 + now.Minute()
|
||||
|
||||
for _, s := range schedules {
|
||||
if s.DayOfWeek != day {
|
||||
continue
|
||||
}
|
||||
startMin := parseTimeMinutes(s.StartTime)
|
||||
endMin := parseTimeMinutes(s.EndTime)
|
||||
if currMin >= startMin && currMin <= endMin {
|
||||
period := s.Period
|
||||
if period < 1 {
|
||||
period = 1
|
||||
}
|
||||
return s.ClassRkID, period
|
||||
}
|
||||
}
|
||||
|
||||
return classRkIDs[0], 1
|
||||
}
|
||||
|
||||
func FindActiveClassForStudent(db *gorm.DB, studentRkID int64) int64 {
|
||||
classID, _ := FindActiveClassAndPeriodForStudent(db, studentRkID)
|
||||
return classID
|
||||
}
|
||||
|
||||
// ResolveSessionPeriodForDate lấy ca theo ngày (dùng khi xem nhật ký quá khứ).
|
||||
func ResolveSessionPeriodForDate(db *gorm.DB, classRkID int64, date string, period int) int {
|
||||
if period > 0 {
|
||||
return period
|
||||
}
|
||||
day, err := time.Parse("2006-01-02", date)
|
||||
if err != nil {
|
||||
return 1
|
||||
}
|
||||
var slots []models.ClassSchedule
|
||||
if err := db.Where("class_rk_id = ? AND day_of_week = ? AND is_active = ?", classRkID, scheduleDayIndex(day), true).
|
||||
Order("period asc, start_time asc").Find(&slots).Error; err != nil || len(slots) == 0 {
|
||||
return 1
|
||||
}
|
||||
if slots[0].Period > 0 {
|
||||
return slots[0].Period
|
||||
}
|
||||
return 1
|
||||
}
|
||||
85
server/internal/db/courses.go
Normal file
85
server/internal/db/courses.go
Normal file
@@ -0,0 +1,85 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"server/internal/models"
|
||||
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/clause"
|
||||
)
|
||||
|
||||
type ClassCourseDTO struct {
|
||||
ID int64 `json:"id"`
|
||||
Name string `json:"name"`
|
||||
CourseCode string `json:"courseCode"`
|
||||
}
|
||||
|
||||
func UpsertClassCourses(db *gorm.DB, classRkID int64, items []ClassCourseDTO) error {
|
||||
if len(items) == 0 {
|
||||
return nil
|
||||
}
|
||||
rows := make([]models.ClassCourse, 0, len(items))
|
||||
for _, it := range items {
|
||||
if it.ID <= 0 {
|
||||
continue
|
||||
}
|
||||
rows = append(rows, models.ClassCourse{
|
||||
ClassRkID: classRkID,
|
||||
CourseRkID: it.ID,
|
||||
Name: it.Name,
|
||||
CourseCode: it.CourseCode,
|
||||
})
|
||||
}
|
||||
if len(rows) == 0 {
|
||||
return nil
|
||||
}
|
||||
return db.Clauses(clause.OnConflict{
|
||||
Columns: []clause.Column{{Name: "class_rk_id"}, {Name: "course_rk_id"}},
|
||||
DoUpdates: clause.AssignmentColumns([]string{"name", "course_code", "updated_at"}),
|
||||
}).Create(&rows).Error
|
||||
}
|
||||
|
||||
func ListClassCourses(db *gorm.DB, classRkID int64) ([]ClassCourseDTO, error) {
|
||||
var rows []models.ClassCourse
|
||||
if err := db.Where("class_rk_id = ?", classRkID).Order("name asc").Find(&rows).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out := make([]ClassCourseDTO, 0, len(rows))
|
||||
for _, r := range rows {
|
||||
out = append(out, ClassCourseDTO{
|
||||
ID: r.CourseRkID,
|
||||
Name: r.Name,
|
||||
CourseCode: r.CourseCode,
|
||||
})
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func MergeClassCourses(base []ClassCourseDTO, extra []ClassCourseDTO) []ClassCourseDTO {
|
||||
merged := make(map[int64]ClassCourseDTO, len(base)+len(extra))
|
||||
for _, c := range base {
|
||||
if c.ID > 0 {
|
||||
merged[c.ID] = c
|
||||
}
|
||||
}
|
||||
for _, c := range extra {
|
||||
if c.ID <= 0 {
|
||||
continue
|
||||
}
|
||||
if existing, ok := merged[c.ID]; ok {
|
||||
if c.Name != "" {
|
||||
existing.Name = c.Name
|
||||
}
|
||||
if c.CourseCode != "" {
|
||||
existing.CourseCode = c.CourseCode
|
||||
}
|
||||
merged[c.ID] = existing
|
||||
} else {
|
||||
merged[c.ID] = c
|
||||
}
|
||||
}
|
||||
out := make([]ClassCourseDTO, 0, len(merged))
|
||||
for _, c := range merged {
|
||||
out = append(out, c)
|
||||
}
|
||||
return out
|
||||
}
|
||||
@@ -3,6 +3,9 @@ package db
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"server/internal/models"
|
||||
|
||||
"gorm.io/driver/mysql"
|
||||
@@ -35,10 +38,110 @@ func ConnectDB(host, port, user, password, dbName string) (*gorm.DB, error) {
|
||||
|
||||
func AutoMigrate(db *gorm.DB) error {
|
||||
log.Println("Running AutoMigrate...")
|
||||
return db.AutoMigrate(
|
||||
if err := migrateLegacyWifiTables(db); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.AutoMigrate(
|
||||
&models.Class{},
|
||||
&models.ClassCourseLink{},
|
||||
&models.Student{},
|
||||
&models.ClassStudent{},
|
||||
)
|
||||
&models.ClassSchedule{},
|
||||
&models.ClassCourse{},
|
||||
&models.ClassAllowedApp{},
|
||||
&models.AppPoolEntry{},
|
||||
&models.WifiPoolEntry{},
|
||||
&models.AcceptedWifi{},
|
||||
&models.StudentSession{},
|
||||
&models.AttendanceResult{},
|
||||
); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := purgeLegacyWifiWithoutBSSID(db); err != nil {
|
||||
return err
|
||||
}
|
||||
return migrateLegacyDiscoveredApps(db)
|
||||
}
|
||||
|
||||
// Bảng WiFi cũ dùng ssid_key làm PK; schema mới cần id + bssid_key — GORM không tự đổi PK.
|
||||
func migrateLegacyWifiTables(db *gorm.DB) error {
|
||||
if db.Migrator().HasTable("accepted_wifis") {
|
||||
needsReset := !db.Migrator().HasColumn(&models.AcceptedWifi{}, "id") ||
|
||||
!db.Migrator().HasColumn(&models.AcceptedWifi{}, "bssid_key")
|
||||
if needsReset {
|
||||
log.Println("Dropping legacy accepted_wifis (SSID-only schema) — will recreate with BSSID...")
|
||||
if err := db.Migrator().DropTable("accepted_wifis"); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
if db.Migrator().HasTable("wifi_pool") {
|
||||
needsReset := !db.Migrator().HasColumn(&models.WifiPoolEntry{}, "bssid_key")
|
||||
if needsReset {
|
||||
log.Println("Dropping legacy wifi_pool (SSID-only schema) — will recreate with BSSID...")
|
||||
if err := db.Migrator().DropTable("wifi_pool"); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func purgeLegacyWifiWithoutBSSID(db *gorm.DB) error {
|
||||
_ = db.Where("bssid = '' OR bssid_key = '' OR LENGTH(bssid_key) <> 12").Delete(&models.AcceptedWifi{}).Error
|
||||
_ = db.Where("bssid = '' OR bssid_key = '' OR LENGTH(bssid_key) <> 12").Delete(&models.WifiPoolEntry{}).Error
|
||||
return nil
|
||||
}
|
||||
|
||||
// Gộp dữ liệu cũ (theo lớp) vào kho global nếu bảng discovered_apps còn tồn tại.
|
||||
func migrateLegacyDiscoveredApps(db *gorm.DB) error {
|
||||
if !db.Migrator().HasTable("discovered_apps") {
|
||||
return nil
|
||||
}
|
||||
type legacyRow struct {
|
||||
ProcessName string
|
||||
WindowTitle string
|
||||
Keyword string
|
||||
HitCount int
|
||||
LastSeenAt time.Time
|
||||
StudentRkID int64
|
||||
ClassRkID int64
|
||||
}
|
||||
var rows []legacyRow
|
||||
if err := db.Table("discovered_apps").Find(&rows).Error; err != nil {
|
||||
return nil
|
||||
}
|
||||
for _, r := range rows {
|
||||
pKey := strings.ToLower(strings.TrimSpace(r.ProcessName))
|
||||
if pKey == "" {
|
||||
continue
|
||||
}
|
||||
kw := strings.TrimSuffix(pKey, ".exe")
|
||||
if r.Keyword != "" {
|
||||
kw = r.Keyword
|
||||
}
|
||||
var existing models.AppPoolEntry
|
||||
if err := db.Where("process_key = ?", pKey).First(&existing).Error; err == nil {
|
||||
existing.HitCount += r.HitCount
|
||||
if r.LastSeenAt.After(existing.LastSeenAt) {
|
||||
existing.LastSeenAt = r.LastSeenAt
|
||||
existing.WindowTitle = r.WindowTitle
|
||||
existing.ProcessName = r.ProcessName
|
||||
existing.LastStudentRkID = r.StudentRkID
|
||||
existing.LastClassRkID = r.ClassRkID
|
||||
}
|
||||
_ = db.Save(&existing).Error
|
||||
} else {
|
||||
_ = db.Create(&models.AppPoolEntry{
|
||||
ProcessName: r.ProcessName,
|
||||
ProcessKey: pKey,
|
||||
WindowTitle: r.WindowTitle,
|
||||
Keyword: kw,
|
||||
HitCount: r.HitCount,
|
||||
LastSeenAt: r.LastSeenAt,
|
||||
LastStudentRkID: r.StudentRkID,
|
||||
LastClassRkID: r.ClassRkID,
|
||||
}).Error
|
||||
}
|
||||
}
|
||||
return db.Migrator().DropTable("discovered_apps")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user