Adjust check for rpm/deb archive type

This commit is contained in:
Ahrav Dutta
2024-04-15 11:51:49 -07:00
parent 5bcaffb7e6
commit 91d2665647
2 changed files with 8 additions and 13 deletions
+6 -4
View File
@@ -97,12 +97,14 @@ func HandleFile(ctx logContext.Context, reReader *diskbufferreader.DiskBufferRea
return false
}
_, arReader, err := archiver.Identify("", reReader)
if errors.Is(err, archiver.ErrNoMatch) {
return false
if !(mimeT == arMimeType || mimeT == rpmMimeType) {
_, _, err := archiver.Identify("", reReader)
if errors.Is(err, archiver.ErrNoMatch) {
return false
}
}
archiveChan, err := handler.HandleFile(ctx, arReader) // Delegate to the specific handler to process the file.
archiveChan, err := handler.HandleFile(ctx, reReader) // Delegate to the specific handler to process the file.
if err != nil {
ctx.Logger().Error(err, "error handling file")
return false
+2 -9
View File
@@ -36,15 +36,8 @@ func (h *RPMHandler) HandleFile(ctx logContext.Context, input io.Reader) (chan [
return
}
for {
select {
case <-ctx.Done():
return
default:
if err := h.processRPMFiles(ctx, reader, archiveChan); err != nil {
ctx.Logger().Error(err, "error processing RPM files")
}
}
if err := h.processRPMFiles(ctx, reader, archiveChan); err != nil {
ctx.Logger().Error(err, "error processing RPM files")
}
}()