parent
6ea4d4700c
commit
0c3419a1fb
|
|
@ -1,6 +1,7 @@
|
||||||
package broker
|
package broker
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/base64"
|
||||||
"fmt"
|
"fmt"
|
||||||
"local/truckstop/config"
|
"local/truckstop/config"
|
||||||
"local/truckstop/logtr"
|
"local/truckstop/logtr"
|
||||||
|
|
@ -26,6 +27,18 @@ type JobLocation struct {
|
||||||
State string
|
State string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (j Job) UID() string {
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"%v-%s-%s-%s-%s-%v",
|
||||||
|
j.ID,
|
||||||
|
j.Pickup.State,
|
||||||
|
base64.StdEncoding.EncodeToString([]byte(j.Pickup.City)),
|
||||||
|
j.Dropoff.State,
|
||||||
|
base64.StdEncoding.EncodeToString([]byte(j.Dropoff.City)),
|
||||||
|
j.Pickup.Date.Unix(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
func (j *Job) Secrets() {
|
func (j *Job) Secrets() {
|
||||||
if j.secrets == nil {
|
if j.secrets == nil {
|
||||||
return
|
return
|
||||||
|
|
|
||||||
5
main.go
5
main.go
|
|
@ -291,12 +291,13 @@ func once() error {
|
||||||
jobs[i].Secrets()
|
jobs[i].Secrets()
|
||||||
}
|
}
|
||||||
logtr.Infof("once: sending jobs: %+v", jobs)
|
logtr.Infof("once: sending jobs: %+v", jobs)
|
||||||
|
db := config.Get().DB()
|
||||||
for i := range jobs {
|
for i := range jobs {
|
||||||
if ok, err := sendJob(jobs[i]); err != nil {
|
if ok, err := sendJob(jobs[i]); err != nil {
|
||||||
return err
|
return err
|
||||||
} else if ok {
|
} else if ok {
|
||||||
logtr.Debugf("sent job", jobs[i])
|
logtr.Debugf("sent job", jobs[i])
|
||||||
if err := config.Get().DB().Set(jobs[i].ID, []byte(`sent`)); err != nil {
|
if err := db.Set(jobs[i].UID(), []byte(`sent`)); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -368,7 +369,7 @@ func updateDeadJobs(jobs []broker.Job) error {
|
||||||
func dropStaleJobs(jobs []broker.Job) ([]broker.Job, error) {
|
func dropStaleJobs(jobs []broker.Job) ([]broker.Job, error) {
|
||||||
db := config.Get().DB()
|
db := config.Get().DB()
|
||||||
for i := len(jobs) - 1; i >= 0; i-- {
|
for i := len(jobs) - 1; i >= 0; i-- {
|
||||||
if _, err := db.Get(jobs[i].ID); err == storage.ErrNotFound {
|
if _, err := db.Get(jobs[i].UID()); err == storage.ErrNotFound {
|
||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
} else {
|
} else {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue