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

package problem

import (
	"errors"
	"git.0x7f.app/WOJ/woj-server/internal/e"
	"git.0x7f.app/WOJ/woj-server/internal/model"
	"go.uber.org/zap"
	"gorm.io/gorm"
)

func (s *service) QueryVersion(pvid uint, shouldEnable bool) (*model.ProblemVersion, e.Status) {
	problemVersion := new(model.ProblemVersion)

	err := s.db.Get().First(&problemVersion, pvid).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, e.ProblemVersionNotFound
	}
	if err != nil {
		s.log.Warn("DatabaseError", zap.Error(err), zap.Any("pvid", pvid))
		return nil, e.DatabaseError
	}

	if shouldEnable && !problemVersion.IsEnabled {
		return nil, e.ProblemVersionNotAvailable
	}
	return problemVersion, e.Success
}