refactor: Clear database tables instead of truncating

This commit is contained in:
Felicitas Pojtinger
2021-12-15 00:40:31 +01:00
parent b610287fad
commit 263a4e05d7
2 changed files with 11 additions and 18 deletions

View File

@@ -240,3 +240,11 @@ func (p *MetadataPersister) GetLastIndexedRecordAndBlock(ctx context.Context, re
return header.Lastknownrecord, header.Lastknownblock, nil
}
func (p *MetadataPersister) PurgeAllHeaders(ctx context.Context) error {
if _, err := models.Headers().DeleteAll(ctx, p.db); err != nil {
return err
}
return nil
}

View File

@@ -8,7 +8,6 @@ import (
"io"
"io/ioutil"
"math"
"os"
"strconv"
"github.com/pojntfx/stfs/internal/converters"
@@ -46,23 +45,9 @@ func Index(
onHeader func(hdr *models.Header),
) error {
if overwrite {
f, err := os.OpenFile(metadata.Metadata, os.O_WRONLY|os.O_CREATE, 0600)
if err != nil {
if err := metadata.Metadata.PurgeAllHeaders(context.Background()); err != nil {
return err
}
if err := f.Truncate(0); err != nil {
return err
}
if err := f.Close(); err != nil {
return err
}
}
metadataPersister := persisters.NewMetadataPersister(metadata.Metadata)
if err := metadataPersister.Open(); err != nil {
return err
}
if reader.DriveIsRegular {
@@ -134,7 +119,7 @@ func Index(
return err
}
if err := indexHeader(record, block, hdr, metadataPersister, pipes.Compression, pipes.Encryption, onHeader); err != nil {
if err := indexHeader(record, block, hdr, metadata.Metadata, pipes.Compression, pipes.Encryption, onHeader); err != nil {
return err
}
}
@@ -220,7 +205,7 @@ func Index(
return err
}
if err := indexHeader(record, block, hdr, metadataPersister, pipes.Compression, pipes.Encryption, onHeader); err != nil {
if err := indexHeader(record, block, hdr, metadata.Metadata, pipes.Compression, pipes.Encryption, onHeader); err != nil {
return err
}
}