diff --git a/server/controller/salary.go b/server/controller/salary.go index e1e4487..4ae1f95 100644 --- a/server/controller/salary.go +++ b/server/controller/salary.go @@ -69,7 +69,12 @@ func (s *Salary) Download(ctx *gin.Context) { corp, err := dao.NewCorpDao().Get(sess.GetCorpId()) session.CheckDBError(err) - session.CheckNilError(corp, "企业不存在") + if corp == nil { + corp, err = dao.NewCorpDao().GetByHost(sess.GetHeader().Host) + session.CheckDBError(err) + session.CheckNilError(corp, "企业不存在") + } + month := time.Now().AddDate(0, -1, 0).Format("200601") month = strings.ReplaceAll(month, "-", "") diff --git a/server/session/session.go b/server/session/session.go index 071bd8c..b03c707 100644 --- a/server/session/session.go +++ b/server/session/session.go @@ -42,7 +42,10 @@ func (s *AdminSession) GetAdmin() *model.StaffUser { } func (s *AdminSession) GetCorpId() int64 { - return s.adminUser.CorpId + if s.adminUser != nil { + return s.adminUser.CorpId + } + return 0 } func (s *AdminSession) GetUsername() string {