Add the -checksum flag to calculate and print the checksum for source…

… file

README-CN.md

@@ -298,6 +298,16 @@ $ gofs -source=./source -dest=./dest -log_file -log_level=0 -log_dir="./logs/" -
 $ gofs -conf=./gofs.yaml
 ```
 
+### 校验和
+
+你可以使用`checksum`命令行参数来计算并打印文件的校验和
+
+`chunk_size`和`checkpoint_count`命令行参数在这里同在[本地磁盘](#本地磁盘)中一样有效
+
+```bash
+$ gofs -source=./gofs -checksum
+```
+
 ## 更多信息
 
 ### 帮助信息

README.md

@@ -318,6 +318,16 @@ or the response of [Config API](#config-api).
 $ gofs -conf=./gofs.yaml
 ```
 
+### Checksum
+
+You can use the `checksum` flag to calculate the file checksum and print the result.
+
+The `chunk_size` and `checkpoint_count` flags are effective here the same as in the [Local Disk](#local-disk).
+
+```bash
+$ gofs -source=./gofs -checksum
+```
+
 ## For More Information
 
 ### Help Info

checksum/checksum.go

@@ -0,0 +1,20 @@
+package checksum
+
+import (
+	"github.com/no-src/gofs/util/hashutil"
+	"github.com/no-src/gofs/util/jsonutil"
+	"github.com/no-src/log"
+)
+
+// PrintChecksum calculate and print the checksum for file
+func PrintChecksum(path string, chunkSize int64, checkpointCount int) error {
+	hvs, err := hashutil.CheckpointsMD5FromFileName(path, chunkSize, checkpointCount)
+	if err != nil {
+		log.Error(err, "calculate file checksum error")
+		return err
+	}
+
+	hvsJson, _ := jsonutil.MarshalIndent(hvs)
+	log.Log(string(hvsJson))
+	return err
+}

checksum/checksum_test.go

@@ -0,0 +1,19 @@
+package checksum
+
+import "testing"
+
+func TestPrintChecksum(t *testing.T) {
+	path := "./checksum_test.go"
+	err := PrintChecksum(path, 1024*1024, 10)
+	if err != nil {
+		t.Errorf("test PrintChecksum error => %v", err)
+	}
+}
+
+func TestPrintChecksumError(t *testing.T) {
+	path := "./"
+	err := PrintChecksum(path, 1024*1024, 10)
+	if err == nil {
+		t.Errorf("test PrintChecksum expect get error but get nil")
+	}
+}

cmd/gofs/flag.go

@@ -84,6 +84,9 @@ func parseFlags() {
 	flag.IntVar(&config.RandomPasswordLen, "rand_pwd_len", 10, "the length of the random user's password")
 	flag.StringVar(&config.RandomDefaultPerm, "rand_perm", "r", "the default permission of every random user, like 'rwx'")
 
+	// checksum
+	flag.BoolVar(&config.Checksum, "checksum", false, "calculate and print the checksum for source file")
+
 	flag.Parse()
 }
 

cmd/gofs/main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"github.com/no-src/gofs/about"
 	"github.com/no-src/gofs/auth"
+	"github.com/no-src/gofs/checksum"
 	"github.com/no-src/gofs/conf"
 	"github.com/no-src/gofs/daemon"
 	"github.com/no-src/gofs/fs"
@@ -139,6 +140,12 @@ func executeOnce() (exit bool) {
 		return true
 	}
 
+	// calculate checksum
+	if config.Checksum {
+		checksum.PrintChecksum(config.Source.Path(), config.ChunkSize, config.CheckpointCount)
+		return true
+	}
+
 	return false
 }
 

conf/config.go

@@ -64,4 +64,7 @@ type Config struct {
 	RandomUserNameLen int    `json:"rand_user_len" yaml:"rand_user_len"`
 	RandomPasswordLen int    `json:"rand_pwd_len" yaml:"rand_pwd_len"`
 	RandomDefaultPerm string `json:"rand_perm" yaml:"rand_perm"`
+
+	// checksum
+	Checksum bool `json:"checksum" yaml:"checksum"`
 }

conf/example/gofs-remote-client.json

@@ -38,5 +38,6 @@
   "rand_user_count": 0,
   "rand_user_len": 6,
   "rand_pwd_len": 10,
-  "rand_perm": "r"
+  "rand_perm": "r",
+  "checksum": false
 }

conf/example/gofs-remote-server.yaml

@@ -37,4 +37,5 @@ users: gofs|password|rwx
 rand_user_count: 0
 rand_user_len: 6
 rand_pwd_len: 10
-rand_perm: r
+rand_perm: r
+checksum: false