woj-server/internal/service/problem/query.go

package problem

import (
	"errors"
	"github.com/WHUPRJ/woj-server/internal/e"
	"github.com/WHUPRJ/woj-server/internal/model"
	"github.com/WHUPRJ/woj-server/pkg/utils"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

func (s *service) Query(problemId uint) (*model.Problem, e.Status) {
	problem := new(model.Problem)

	err := s.db.Preload(clause.Associations).First(&problem, problemId).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, e.ProblemNotFound
	}
	if err != nil {
		return nil, e.DatabaseError
	}

	return problem, utils.If(problem.IsEnabled, e.Success, e.ProblemNotAvailable).(e.Status)
}