Setup de la règle SendErrorRule

2022-03-15 20:27:49 +01:00
parent 4a2a84b874
commit 43844c9b54
4 changed files with 10 additions and 7 deletions
--- a/pkg/protocol/rules/readers/SendFileRule.go
+++ b/pkg/protocol/rules/readers/SendFileRule.go
@@ -65,12 +65,14 @@ func (rule SendFileRule) onRead(fileName string, fileSize int, fingerPrint strin

 	return func(reader *bufio.Reader) *protocol.ProtocolWriterResult {
 		path := fmt.Sprintf("%s/%s", rule.storagePath, fileName)
+
 		if utils.ReceiveFile(path, fileSize, reader) {
 			hashCompare, err := utils.HashFileCompare(path, fingerPrint)
 			if err == nil && hashCompare {
 				return rule.protocolRepo.ExecuteWriter(writers.SendOkRulePrefix)
 			}
 		}
+
 		os.Remove(path)
 		return rule.protocolRepo.ExecuteWriter(writers.SendErrorRulePrefix)
 	}