Fix legacy unit schema migration

Signed-off-by: Sean Du <do4suki@gmail.com>
2026-06-28 01:20:05 +08:00
parent b276fada4b
commit 1bb8f20c88
5 changed files with 29 additions and 1 deletions
@@ -19,6 +19,10 @@ func (ss *Session) GetBasicUnitInfo() *xorm.Session {
 }

 func (ss *Session) GetUnitInfo(unitID int) (bool, *unitmodel.CommonUnitData) {
+	if ss.UserEng == nil {
+		return false, nil
+	}
+
 	unitInfo := unitmodel.CommonUnitData{}
 	has, err := ss.UserEng.Table(new(unitmodel.CommonUnitData)).
 		Where("unit_id = ?", unitID).Get(&unitInfo)