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

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)
 }

pkg/rpc/server/listen.go

@@ -26,21 +26,23 @@ const updateInterval = 1 * time.Hour
 
 // Server represents Trivy server
 type Server struct {
-	appVersion  string
+	appVersion   string
-	addr        string
+	addr         string
-	cacheDir    string
+	cacheDir     string
-	token       string
+	token        string
-	tokenHeader 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,
+		appVersion:   appVersion,
-		addr:        addr,
+		addr:         addr,
-		cacheDir:    cacheDir,
+		cacheDir:     cacheDir,
-		token:       token,
+		token:        token,
-		tokenHeader: tokenHeader,
+		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)