59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
|
package writers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"StoreBackEnd/pkg/protocol"
|
||
|
|
"StoreBackEnd/pkg/utils"
|
||
|
|
"bufio"
|
||
|
|
"fmt"
|
||
|
|
"strconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RetrieveOkRulePrefix Rule command prefix
|
||
|
|
const RetrieveOkRulePrefix = "RETRIEVE_OK"
|
||
|
|
|
||
|
|
// RetrieveOkRule Storage structure for the RetrieveOkRule
|
||
|
|
type RetrieveOkRule struct {
|
||
|
|
// cmd Rule name
|
||
|
|
cmd string
|
||
|
|
// matcher Allow to make a regex match
|
||
|
|
matcher *protocol.RegexMatcher
|
||
|
|
// storagePath Chemin de stockage du fichier
|
||
|
|
storagePath string
|
||
|
|
}
|
||
|
|
|
||
|
|
// CreateRetrieveOkRule Creating an instance of RetrieveOkRule
|
||
|
|
func CreateRetrieveOkRule(pattern string, storagePath string) protocol.IProtocolWriter {
|
||
|
|
return &RetrieveOkRule{
|
||
|
|
cmd: RetrieveOkRulePrefix,
|
||
|
|
matcher: protocol.CreateRegexMatcher(pattern),
|
||
|
|
storagePath: storagePath,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetCmd retrieve the command name.
|
||
|
|
func (rule RetrieveOkRule) GetCmd() string {
|
||
|
|
return rule.cmd
|
||
|
|
}
|
||
|
|
|
||
|
|
// Execute the Rule with a string command.
|
||
|
|
func (rule RetrieveOkRule) Execute(argsData ...string) *protocol.ProtocolWriterResult {
|
||
|
|
buildedCmd, _ := rule.matcher.Build(RetrieveOkRulePrefix, argsData...)
|
||
|
|
|
||
|
|
return &protocol.ProtocolWriterResult{
|
||
|
|
Cmd: buildedCmd,
|
||
|
|
Write: func(writer *bufio.Writer) {
|
||
|
|
fileSize, _ := strconv.Atoi(argsData[1])
|
||
|
|
rule.onWrite(writer, argsData[0], fileSize)
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rule RetrieveOkRule) onWrite(writer *bufio.Writer, hashFileName string, fileSize int) {
|
||
|
|
path := fmt.Sprintf("%s/%s", rule.storagePath, hashFileName)
|
||
|
|
if utils.SendFile(path, fileSize, writer) {
|
||
|
|
println("Fichier envoyé au client")
|
||
|
|
} else {
|
||
|
|
println("Fichier non-envoyé au client")
|
||
|
|
}
|
||
|
|
}
|