[#255] services/audit: Complete audit report after the last check

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
Leonard Lyubich 2020-12-22 11:08:31 +03:00 committed by Alex Vanin
parent 4dc09b19f3
commit 33804e024d
3 changed files with 19 additions and 4 deletions

View file

@ -71,6 +71,10 @@ func (c *Context) expired() bool {
} }
} }
func (c *Context) complete() {
c.report.Complete()
}
func (c *Context) writeReport() { func (c *Context) writeReport() {
c.log.Debug("writing audit report...") c.log.Debug("writing audit report...")

View file

@ -8,21 +8,27 @@ import (
func (c *Context) Execute() { func (c *Context) Execute() {
c.init() c.init()
for _, check := range []struct { checks := []struct {
name string name string
exec func() exec func()
}{ }{
{name: "PoR", exec: c.executePoR}, {name: "PoR", exec: c.executePoR},
{name: "PoP", exec: c.executePoP}, {name: "PoP", exec: c.executePoP},
{name: "PDP", exec: c.executePDP}, {name: "PDP", exec: c.executePDP},
} { }
c.log.Debug(fmt.Sprintf("executing %s check...", check.name))
for i := range checks {
c.log.Debug(fmt.Sprintf("executing %s check...", checks[i].name))
if c.expired() { if c.expired() {
break break
} }
check.exec() checks[i].exec()
if i == len(checks)-1 {
c.complete()
}
} }
c.writeReport() c.writeReport()

View file

@ -31,3 +31,8 @@ func NewReport(cid *container.ID) *Report {
func (r *Report) Result() *audit.Result { func (r *Report) Result() *audit.Result {
return r.res return r.res
} }
// Complete completes audit report.
func (r *Report) Complete() {
r.res.SetComplete(true)
}