Browse Source

En cours de développement

master
losyme 2 months ago
parent
commit
956dd9cb6f
  1. 35
      internal/storage/mongo/jobs.go

35
internal/storage/mongo/jobs.go

@ -7,6 +7,7 @@
package mongo
import (
"context"
"time"
"forge.chapril.org/dune/jw"
@ -20,21 +21,7 @@ func (ms *mongoStorage) ValidateJob(_ *jw.Job) error {
return nil
}
func (ms *mongoStorage) InsertJob(job *jw.Job) (bool, error) {
col := ms.Collection("jobs")
ctx, cancel := util.CtxWithTimeout(5 * time.Second)
defer cancel()
if job.Category == nil {
_, err := col.InsertOne(ctx, job)
if err != nil {
return false, err
}
return true, nil
}
func (ms *mongoStorage) txInsertJob(ctx context.Context, job *jw.Job, col *mongo.Collection) (bool, error) {
session, err := ms.client.StartSession()
if err != nil {
return false, err
@ -75,6 +62,24 @@ func (ms *mongoStorage) InsertJob(job *jw.Job) (bool, error) {
return result.(bool), err
}
func (ms *mongoStorage) InsertJob(job *jw.Job) (bool, error) {
col := ms.Collection("jobs")
ctx, cancel := util.CtxWithTimeout(5 * time.Second)
defer cancel()
if job.Category == nil {
_, err := col.InsertOne(ctx, job)
if err != nil {
return false, err
}
return true, nil
}
return ms.txInsertJob(ctx, job, col)
}
func (ms *mongoStorage) UpdateJob(job *jw.Job) error {
return nil
}

Loading…
Cancel
Save