Setup de la règle SendErrorRule

This commit is contained in:
Benjamin
2022-03-15 20:27:49 +01:00
parent 4a2a84b874
commit 43844c9b54
4 changed files with 10 additions and 7 deletions

View File

@@ -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)
}

View File

@@ -45,6 +45,7 @@ func HashFileCompare(filePath string, fingerprint string) (bool, error) {
if err != nil {
return false, err
}
println("OKOK ", fileFingerprint, fingerprint)
if strings.Compare(fileFingerprint, fingerprint) == 0 {
return true, nil
}