mirror of
https://github.com/muun/recovery.git
synced 2025-02-23 11:32:33 -05:00
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
// +build windows
|
|
|
|
package headerfs
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
// singleTruncate truncates a single header from the end of the header file.
|
|
// This can be used in the case of a re-org to remove the last header from the
|
|
// end of the main chain.
|
|
//
|
|
// TODO(roasbeef): define this and the two methods above on a headerFile
|
|
// struct?
|
|
func (h *headerStore) singleTruncate() error {
|
|
// In order to truncate the file, we'll need to grab the absolute size
|
|
// of the file as it stands currently.
|
|
fileInfo, err := h.file.Stat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fileSize := fileInfo.Size()
|
|
|
|
// Next, we'll determine the number of bytes we need to truncate from
|
|
// the end of the file.
|
|
var truncateLength int64
|
|
switch h.indexType {
|
|
case Block:
|
|
truncateLength = 80
|
|
case RegularFilter:
|
|
truncateLength = 32
|
|
default:
|
|
return fmt.Errorf("unknown index type: %v", h.indexType)
|
|
}
|
|
|
|
// Finally, we'll use both of these values to calculate the new size of
|
|
// the file.
|
|
newSize := fileSize - truncateLength
|
|
|
|
// On Windows, a file can't be truncated while open, even if using a
|
|
// file handle to truncate it. This means we have to close, truncate,
|
|
// and reopen it.
|
|
fileName := h.file.Name()
|
|
if err = h.file.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = os.Truncate(fileName, newSize); err != nil {
|
|
return err
|
|
}
|
|
|
|
fileFlags := os.O_RDWR | os.O_APPEND | os.O_CREATE
|
|
h.file, err = os.OpenFile(fileName, fileFlags, 0644)
|
|
return err
|
|
}
|