diff --git a/pkg/commands/server/run.go b/pkg/commands/server/run.go index 1a97f6561f..1a8ed82530 100644 --- a/pkg/commands/server/run.go +++ b/pkg/commands/server/run.go @@ -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) } diff --git a/pkg/rpc/server/listen.go b/pkg/rpc/server/listen.go index 9e4f33de43..678a1ff71c 100644 --- a/pkg/rpc/server/listen.go +++ b/pkg/rpc/server/listen.go @@ -26,21 +26,23 @@ const updateInterval = 1 * time.Hour // Server represents Trivy server type Server struct { - appVersion string - addr string - cacheDir string - token string - tokenHeader string + appVersion string + addr string + 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, + 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)