dkforest

commit 7ba7b754f8da20b7af083fb44a18aeb3a658360d
parent 1cad3d4e5e0b230f4d3f0a7295b162b5c5d6db43
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Mon, 30 Jan 2023 15:43:36 -0800

fix nb chunks needed

Diffstat:
M
cmd/dkfupload/main.go
 | 
8
++++++--

1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/cmd/dkfupload/main.go b/cmd/dkfupload/main.go
@@ -4,6 +4,7 @@ import (
 	"flag"
 	"github.com/dustin/go-humanize"
 	"log"
+	"math"
 	"os"
 	"sync"
 )
@@ -25,13 +26,16 @@ func main() {
 	f, _ := os.Open(fileName)
 	fs, _ := f.Stat()
 	fileSize := fs.Size()
-	nbChunks := fileSize / maxChunkSize
+	nbChunks := int64(math.Ceil(float64(fileSize) / float64(maxChunkSize)))
 
 	log.Println("file: ", fs.Name())
 	log.Printf("file size: %s (%s) \n", humanize.Bytes(uint64(fileSize)), humanize.Comma(fileSize))
 	log.Printf("Chunks size: %s (%s)\n", humanize.Bytes(uint64(maxChunkSize)), humanize.Comma(maxChunkSize))
 	log.Println("Nb chunks: ", nbChunks)
 	log.Println("Nb threads: ", nbThreads)
+	if dry {
+		log.Println("dry run")
+	}
 
 	ch := make(chan int64)
 	go func() {
@@ -49,7 +53,7 @@ func main() {
 				offset := int64(chunkNum) * maxChunkSize
 				buf := make([]byte, maxChunkSize)
 				n, _ := f.ReadAt(buf, offset)
-				log.Printf("Thread #%d read %d | from %d to %d \n", i, n, offset, offset+maxChunkSize)
+				log.Printf("Thread #%03d | chunk #%03d | read %d | from %d to %d \n", i, chunkNum, n, offset, offset+maxChunkSize)
 				if !dry {
 					// TODO: http post the chunk to server
 				}