upstream/icingadb
1
0
Fork 0
mirror of https://github.com/Icinga/icingadb.git synced 2026-06-08 16:34:29 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Merge pull request #471 from Icinga/embed.lua

Browse source 
go:embed LUA script
This commit is contained in:
Julian Brost 2022-05-10 12:53:07 +02:00 committed by GitHub
commit c1bbc037de
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 38 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
pkg/icingadb/overdue/get_overdues.lua Normal file
View file

@ -0,0 +1,30 @@
-- get_overdues.lua takes the following KEYS:
-- * either icinga:nextupdate:host or icinga:nextupdate:service
-- * either icingadb:overdue:host or icingadb:overdue:service
-- * a random one
--
-- It takes the following ARGV:
-- * the current date and time as *nix timestamp float in seconds
--
-- It returns the following:
-- * overdue monitored objects not yet marked overdue
-- * not overdue monitored objects not yet unmarked overdue
local icingaNextupdate = KEYS[1]
local icingadbOverdue = KEYS[2]
local tempOverdue = KEYS[3]
local now = ARGV[1]
redis.call('DEL', tempOverdue)
local zrbs = redis.call('ZRANGEBYSCORE', icingaNextupdate, '-inf', '(' .. now)
for i = 1, #zrbs do
redis.call('SADD', tempOverdue, zrbs[i])
end
zrbs = nil
local res = { redis.call('SDIFF', tempOverdue, icingadbOverdue), redis.call('SDIFF', icingadbOverdue, tempOverdue) }
redis.call('DEL', tempOverdue)
return res

40
pkg/icingadb/overdue/sync.go
View file

@ -2,6 +2,7 @@ package overdue
import (
"context"
_ "embed"
"fmt"
"github.com/go-redis/redis/v8"
"github.com/google/uuid"
@ -16,7 +17,9 @@ import (
"github.com/icinga/icingadb/pkg/periodic"
"github.com/pkg/errors"
"golang.org/x/sync/errgroup"
"regexp"
"strconv"
"strings"
"time"
)
@ -129,39 +132,12 @@ func (s Sync) log(ctx context.Context, objectType string, counter *com.Counter)
})
}
// luaGetOverdues takes the following KEYS:
// * either icinga:nextupdate:host or icinga:nextupdate:service
// * either icingadb:overdue:host or icingadb:overdue:service
// * a random one
//
// It takes the following ARGV:
// * the current date and time as *nix timestamp float in seconds
//
// It returns the following:
// * overdue monitored objects not yet marked overdue
// * not overdue monitored objects not yet unmarked overdue
var luaGetOverdues = redis.NewScript(`
//go:embed get_overdues.lua
var getOverduesLua string
local icingaNextupdate = KEYS[1]
local icingadbOverdue = KEYS[2]
local tempOverdue = KEYS[3]
local now = ARGV[1]
redis.call('DEL', tempOverdue)
local zrbs = redis.call('ZRANGEBYSCORE', icingaNextupdate, '-inf', '(' .. now)
for i = 1, #zrbs do
redis.call('SADD', tempOverdue, zrbs[i])
end
zrbs = nil
local res = {redis.call('SDIFF', tempOverdue, icingadbOverdue), redis.call('SDIFF', icingadbOverdue, tempOverdue)}
redis.call('DEL', tempOverdue)
return res
`)
var luaGetOverdues = redis.NewScript(strings.TrimSpace(
regexp.MustCompile(`(?m)^--.*?$`).ReplaceAllString(getOverduesLua, ""),
))
// sync synchronizes Redis overdue sets from s.redis to s.db for objectType.
func (s Sync) sync(ctx context.Context, objectType string, factory factory, counter *com.Counter) error {