-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfiles.go
62 lines (55 loc) · 1.48 KB
/
files.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package main
import (
"errors"
"log"
"go.formulabun.club/functional/array"
"go.formulabun.club/functional/strings"
"go.formulabun.club/srb2kart/network"
)
type fileFromServer struct {
file network.File
server *network.ServerInfo
}
func serversToFiles(serversIn chan string, filesOut chan []fileFromServer) {
for server := range serversIn {
localServer := server
go func() {
serverInfo, files, err := getServerData(localServer)
if err != nil {
log.Printf("Could not get files for %s: %s\n", localServer, err)
filesOut <- []fileFromServer{}
} else {
serverName := strings.SafeNullTerminated(serverInfo.ServerName[:])
log.Printf("Got the files for %s\n", serverName)
filesOut <- array.Map(files, func(f network.File) fileFromServer {
return fileFromServer{f, &serverInfo}
})
}
}()
}
}
func getServerData(server string) (network.ServerInfo, []network.File, error) {
// serverinfo
type serverInfoErr struct {
info network.ServerInfo
err error
}
serverInfoChan := make(chan serverInfoErr)
go func() {
serverInfo, _, err := network.AskInfo(server)
serverInfoChan <- serverInfoErr{serverInfo, err}
}()
// files
type filesErr struct {
files []network.File
err error
}
filesChan := make(chan filesErr)
go func() {
files, err := network.TellAllFilesNeeded(server)
filesChan <- filesErr{files, err}
}()
infoErr := <-serverInfoChan
fileErr := <-filesChan
return infoErr.info, fileErr.files, errors.Join(infoErr.err, fileErr.err)
}