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

package submission

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"
	"gorm.io/gorm/clause"
)

func (s *service) Query(pid uint, uid uint, offset int, limit int, count *int64) ([]*model.Submission, e.Status) {
	submissions := make([]*model.Submission, 0)

	submission := &model.Submission{
		ProblemID: pid,
		UserID:    uid,
	}

	err := s.db.Get().Preload(clause.Associations).
		Where(submission).
		Order("created_at DESC").
		Offset(offset).Limit(limit).Find(&submissions).
		Offset(-1).Limit(-1).Count(count).
		Error

	//if errors.Is(err, gorm.ErrRecordNotFound) {
	//	return nil, e.ProblemNotFound
	//}

	if err != nil {
		s.log.Warn("DatabaseError", zap.Error(err), zap.Any("pid", pid), zap.Any("uid", uid))
		return nil, e.DatabaseError
	}
	return submissions, e.Success
}

func (s *service) QueryBySid(sid uint, associations bool) (*model.Submission, e.Status) {
	submission := new(model.Submission)

	query := s.db.Get()
	if associations {
		query = query.Preload(clause.Associations)
	}

	err := query.First(&submission, sid).Error

	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, e.SubmissionNotFound
	}

	if err != nil {
		s.log.Warn("DatabaseError", zap.Error(err), zap.Any("sid", sid))
		return nil, e.DatabaseError
	}

	return submission, e.Success
}