Compare commits

...

3 Commits

Author SHA1 Message Date
Bel LaPointe
27605997c1 emailer configurable interval 2022-01-11 07:50:58 -05:00
Bel LaPointe
6c455764e1 durations are now optionally spans for jitter 2022-01-11 07:39:24 -05:00
Bel LaPointe
31a3d4948b separate ok and err intervals for sleep, dont exit on err 2022-01-11 07:39:07 -05:00
6 changed files with 181 additions and 27 deletions

View File

@@ -1,5 +1,9 @@
{
"Interval": "6h0m0s",
"Interval": {
"OK": "6h0m0s",
"Error": "6h",
"Email": "15m"
},
"States": [
"GA"
],

View File

@@ -10,11 +10,15 @@ import (
)
type Config struct {
Interval Duration
States []State
Storage []string
Client string
Message struct {
Interval struct {
Email Duration
OK Duration
Error Duration
}
States []State
Storage []string
Client string
Message struct {
Matrix struct {
Mock bool
Homeserver string

View File

@@ -2,17 +2,29 @@ package config
import (
"encoding/json"
"errors"
"math/rand"
"strings"
"time"
)
type Duration time.Duration
type Duration struct {
least time.Duration
most time.Duration
}
func (d Duration) Get() time.Duration {
return time.Duration(d)
jitter := d.most - d.least
if jitter >= time.Second {
jitter = time.Second * time.Duration(rand.Int()%int(jitter.Seconds()))
} else {
jitter = 0
}
return time.Duration(d.least + jitter)
}
func (d Duration) MarshalJSON() ([]byte, error) {
return json.Marshal(d.Get().String())
return json.Marshal(d.least.String() + ".." + d.most.String())
}
func (d *Duration) UnmarshalJSON(b []byte) error {
@@ -20,10 +32,25 @@ func (d *Duration) UnmarshalJSON(b []byte) error {
if err := json.Unmarshal(b, &s); err != nil {
return err
}
d2, err := time.ParseDuration(s)
if !strings.Contains(s, "..") {
s = s + ".." + s
}
splits := strings.Split(s, "..")
if len(splits) != 2 {
return errors.New("unexpected ..")
}
least, err := time.ParseDuration(splits[0])
if err != nil {
return err
}
*d = Duration(d2)
most, err := time.ParseDuration(splits[1])
if err != nil {
return err
}
if least > most {
most, least = least, most
}
d.most = most
d.least = least
return nil
}

105
config/duration_test.go Normal file
View File

@@ -0,0 +1,105 @@
package config
import (
"encoding/json"
"testing"
"time"
)
func TestDurationMarshal(t *testing.T) {
cases := map[string]struct {
in string
want Duration
wantJS string
}{
"empty string": {
wantJS: `"0s..0s"`,
},
"random in": {
in: "abc",
want: Duration{},
wantJS: `"0s..0s"`,
},
"1 dur": {
in: "1m",
want: Duration{
least: time.Minute,
most: time.Minute,
},
wantJS: `"1m0s..1m0s"`,
},
"descending": {
in: "2m..1m",
want: Duration{
least: time.Minute,
most: 2 * time.Minute,
},
wantJS: `"1m0s..2m0s"`,
},
"happy span": {
in: "1m..2m",
want: Duration{
least: time.Minute,
most: 2 * time.Minute,
},
wantJS: `"1m0s..2m0s"`,
},
"multi unit descending": {
in: "1h1m..2m3ms",
want: Duration{
least: 2*time.Minute + 3*time.Millisecond,
most: time.Hour + time.Minute,
},
wantJS: `"2m0.003s..1h1m0s"`,
},
}
for name, d := range cases {
c := d
t.Run(name, func(t *testing.T) {
var got Duration
err := json.Unmarshal([]byte(`"`+c.in+`"`), &got)
if got != c.want {
t.Fatalf("err %v: want %+v, got %+v", err, c.want, got)
}
js, _ := json.Marshal(got)
if string(js) != c.wantJS {
t.Fatalf("want marshalled %s, got %s", c.wantJS, js)
}
var got2 Duration
json.Unmarshal(js, &got2)
if got != got2 {
t.Fatalf("want unmarshal-marshal-unmarshal %+v, got %+v", got, got2)
}
})
}
}
func TestDurationGet(t *testing.T) {
t.Run("same", func(t *testing.T) {
d := Duration{least: time.Second, most: time.Second}
for i := 0; i < 50; i++ {
if got := d.Get(); got != time.Second {
t.Error(got)
}
}
})
t.Run("1ms span", func(t *testing.T) {
d := Duration{least: time.Second, most: time.Second + time.Millisecond}
for i := 0; i < 50; i++ {
if got := d.Get(); got != time.Second {
t.Error(got)
}
}
})
t.Run("5s span", func(t *testing.T) {
d := Duration{least: time.Second, most: 6 * time.Second}
for i := 0; i < 50; i++ {
if got := d.Get(); got > time.Second*6 || got < time.Second {
t.Error(got)
}
}
})
}

43
main.go
View File

@@ -21,16 +21,15 @@ var stateFinder = regexp.MustCompile(`[A-Za-z]+`)
func main() {
lock := &sync.Mutex{}
go func() {
c := time.NewTicker(time.Minute)
for range c.C {
if !config.Get().EmailerEnabled {
continue
for {
if config.Get().EmailerEnabled {
lock.Lock()
if err := email(); err != nil {
log.Print(err)
}
lock.Unlock()
}
lock.Lock()
if err := email(); err != nil {
log.Print(err)
}
lock.Unlock()
time.Sleep(config.Get().Interval.Email.Get())
}
}()
if err := _main(); err != nil {
@@ -94,16 +93,28 @@ func parseOutStates(b []byte) []config.State {
func _main() error {
for {
if err := config.Refresh(); err != nil {
return err
}
if err := once(); err != nil {
return err
err := _mainOne()
if err != nil {
log.Println(err)
}
if config.Get().Once {
break
return err
}
time.Sleep(config.Get().Interval.Get())
if err != nil {
time.Sleep(config.Get().Interval.Error.Get())
} else {
time.Sleep(config.Get().Interval.OK.Get())
}
}
return nil
}
func _mainOne() error {
if err := config.Refresh(); err != nil {
return err
}
if err := once(); err != nil {
return err
}
return nil
}

View File

@@ -4,7 +4,10 @@ todo:
- banlist criteria like vendors, brokers, metadata
- quiet hours
- setup pa on element
- accept states via element for one system
done:
- configurable email interval
- jitter on intervals, including dedicated err span
- email doesnt get all matches
- send jobs
- read jobs