2022-10-22 17:38:39 +08:00
|
|
|
package submission
|
|
|
|
|
|
|
|
import (
|
2022-10-23 17:29:35 +08:00
|
|
|
"errors"
|
2023-07-14 21:47:11 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/e"
|
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/model"
|
2022-10-22 17:38:39 +08:00
|
|
|
"go.uber.org/zap"
|
2022-10-23 17:29:35 +08:00
|
|
|
"gorm.io/gorm"
|
2022-10-22 17:38:39 +08:00
|
|
|
"gorm.io/gorm/clause"
|
|
|
|
)
|
|
|
|
|
2023-12-23 15:36:42 +08:00
|
|
|
func (s *service) Query(pid uint, uid uint, offset int, limit int, count *int64) ([]*model.Submission, e.Status) {
|
2022-10-22 17:38:39 +08:00
|
|
|
submissions := make([]*model.Submission, 0)
|
|
|
|
|
|
|
|
submission := &model.Submission{
|
|
|
|
ProblemID: pid,
|
|
|
|
UserID: uid,
|
|
|
|
}
|
|
|
|
|
2023-07-15 16:19:49 +08:00
|
|
|
err := s.db.Get().Preload(clause.Associations).
|
2022-10-22 17:38:39 +08:00
|
|
|
Where(submission).
|
2023-12-23 15:36:42 +08:00
|
|
|
Order("created_at DESC").
|
|
|
|
Offset(offset).Limit(limit).Find(&submissions).
|
|
|
|
Offset(-1).Limit(-1).Count(count).
|
|
|
|
Error
|
2022-10-22 17:38:39 +08:00
|
|
|
|
|
|
|
//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
|
|
|
|
}
|
2022-10-23 17:29:35 +08:00
|
|
|
|
|
|
|
func (s *service) QueryBySid(sid uint, associations bool) (*model.Submission, e.Status) {
|
|
|
|
submission := new(model.Submission)
|
|
|
|
|
2023-07-15 16:19:49 +08:00
|
|
|
query := s.db.Get()
|
2022-10-23 17:29:35 +08:00
|
|
|
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
|
|
|
|
}
|