update to correct block size

1 files changed, 9 insertions, 4 deletions

diff --git a/unmaintained/stress_filer_upload/write_files/write_files.go b/unmaintained/stress_filer_upload/write_files/write_files.go
index 67400eef4..508e37d14 100644
--- a/unmaintained/stress_filer_upload/write_files/write_files.go
+++ b/unmaintained/stress_filer_upload/write_files/write_files.go
@@ -5,12 +5,13 @@ import (
 	"fmt"
 	"math/rand"
 	"os"
+	"time"
 )
 
 var (
 	minSize   = flag.Int("minSize", 1024, "min file size")
-	maxSize   = flag.Int("maxSize", 10*1024*1024, "max file size")
-	fileCount = flag.Int("fileCount", 1, "number of files to write")
+	maxSize   = flag.Int("maxSize", 3*1024*1024, "max file size")
+	fileCount = flag.Int("n", 1, "number of files to write")
 	blockSize = flag.Int("blockSizeKB", 4, "write block size")
 	toDir     = flag.String("dir", ".", "destination directory")
 )
@@ -25,7 +26,7 @@ func main() {
 
 	flag.Parse()
 
-	block := make([]byte, *blockSize)
+	block := make([]byte, *blockSize*1024)
 
 	for i := 0; i < *fileCount; i++ {
 
@@ -33,17 +34,21 @@ func main() {
 		check(err)
 
 		fileSize := *minSize + rand.Intn(*maxSize-*minSize)
+		startTime := time.Now()
+
+		fmt.Printf("write %s %d bytes: ", f.Name(), fileSize)
 
 		for x := 0; x < fileSize; {
 			rand.Read(block)
 			_, err = f.Write(block)
 			check(err)
-			x += *blockSize
+			x += len(block)
 		}
 
 		err = f.Close()
 		check(err)
 
+		fmt.Printf("%.02f MB/sec\n", float64(fileSize)*float64(time.Second)/float64(time.Now().Sub(startTime)*1024*1024))
 	}
 
 }