task_log.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. package db
  2. import (
  3. "context"
  4. "time"
  5. "youngee_b_api/model/gorm_model"
  6. "github.com/sirupsen/logrus"
  7. )
  8. func CreateTaskLog(ctx context.Context, taskId string, log string) error {
  9. db := GetReadDB(ctx)
  10. taskLog := gorm_model.YounggeeTaskLog{
  11. TaskID: taskId,
  12. Content: log,
  13. LogAt: time.Now(),
  14. }
  15. err := db.Model(gorm_model.YounggeeTaskLog{}).Create(&taskLog).Error
  16. if err != nil {
  17. logrus.WithContext(ctx).Errorf("[Sketch db] Insert YounggeeTaskLog error,err:%+v", err)
  18. return err
  19. }
  20. return nil
  21. }
  22. // GetTaskLogsByTaskId 获取达人日志
  23. func GetTaskLogsByTaskId(ctx context.Context, taskId string) ([]*gorm_model.YounggeeTaskLog, int64, error) {
  24. db := GetReadDB(ctx)
  25. var total int64
  26. var taskLogs []*gorm_model.YounggeeTaskLog
  27. whereCondition := gorm_model.YounggeeTaskLog{TaskID: taskId}
  28. err := db.Model(gorm_model.YounggeeTaskLog{}).Where(whereCondition).Find(&taskLogs).Count(&total).Error
  29. if err != nil {
  30. return nil, 0, err
  31. }
  32. if total == 0 {
  33. return nil, 0, err
  34. }
  35. return taskLogs, total, nil
  36. }