gitea-mirror/trivy
1
0
Fork 0
mirror of https://github.com/aquasecurity/trivy.git synced 2025-12-22 23:26:39 -08:00
Code Issues Packages Projects Releases Wiki Activity

fix(server): hot update the db from custom repository (#2406)

Browse Source
This commit is contained in:
afdesk
2022-06-27 16:45:48 +06:00
committed by GitHub
parent 57ed577459
commit e2c3bc4124
2 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/commands/server/run.go
View File

@@ -60,6 +60,6 @@ func run(c Option) (err error) {
}
m.Register()
server := rpcServer.NewServer(c.AppVersion, c.Listen, c.CacheDir, c.Token, c.TokenHeader)
server := rpcServer.NewServer(c.AppVersion, c.Listen, c.CacheDir, c.Token, c.TokenHeader, c.DBRepository)
return server.ListenAndServe(cache, c.Insecure)
}

6
pkg/rpc/server/listen.go
View File

@@ -31,16 +31,18 @@ type Server struct {
cacheDir string
token string
tokenHeader string
dbRepository string
}
// NewServer returns an instance of Server
func NewServer(appVersion, addr, cacheDir, token, tokenHeader string) Server {
func NewServer(appVersion, addr, cacheDir, token, tokenHeader, dbRepository string) Server {
return Server{
appVersion: appVersion,
addr: addr,
cacheDir: cacheDir,
token: token,
tokenHeader: tokenHeader,
dbRepository: dbRepository,
}
}
@@ -50,7 +52,7 @@ func (s Server) ListenAndServe(serverCache cache.Cache, insecure bool) error {
dbUpdateWg := &sync.WaitGroup{}
go func() {
worker := newDBWorker(dbc.NewClient(s.cacheDir, true, insecure))
worker := newDBWorker(dbc.NewClient(s.cacheDir, true, insecure, dbc.WithDBRepository(s.dbRepository)))
ctx := context.Background()
for {
time.Sleep(updateInterval)