2022-09-20 16:42:57 +08:00
|
|
|
package problem
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
"github.com/WHUPRJ/woj-server/internal/e"
|
2022-10-20 16:50:19 +08:00
|
|
|
"github.com/WHUPRJ/woj-server/internal/model"
|
2022-10-22 17:38:39 +08:00
|
|
|
"go.uber.org/zap"
|
2022-09-20 16:42:57 +08:00
|
|
|
"gorm.io/gorm"
|
|
|
|
"gorm.io/gorm/clause"
|
|
|
|
)
|
|
|
|
|
2022-10-22 17:38:39 +08:00
|
|
|
func (s *service) Query(pid uint, associations bool, shouldEnable bool) (*model.Problem, e.Status) {
|
2022-09-20 16:42:57 +08:00
|
|
|
problem := new(model.Problem)
|
|
|
|
|
2022-10-22 17:38:39 +08:00
|
|
|
query := s.db
|
|
|
|
if associations {
|
|
|
|
query = query.Preload(clause.Associations)
|
|
|
|
}
|
|
|
|
err := query.First(&problem, pid).Error
|
2022-09-20 16:42:57 +08:00
|
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
|
return nil, e.ProblemNotFound
|
|
|
|
}
|
|
|
|
if err != nil {
|
2022-10-22 17:38:39 +08:00
|
|
|
s.log.Warn("DatabaseError", zap.Error(err), zap.Any("pid", pid))
|
2022-09-20 16:42:57 +08:00
|
|
|
return nil, e.DatabaseError
|
|
|
|
}
|
|
|
|
|
2022-10-22 17:38:39 +08:00
|
|
|
if shouldEnable && !problem.IsEnabled {
|
|
|
|
return nil, e.ProblemNotAvailable
|
|
|
|
}
|
|
|
|
return problem, e.Success
|
2022-09-20 16:42:57 +08:00
|
|
|
}
|