ProtoHandler refactor

This commit is contained in:
netbenix 2021-11-09 18:22:52 +01:00
parent 40e0d99596
commit 150037aa57

View file

@ -46,35 +46,37 @@ func parsePackages(res *proto.SearchResponse, reg Lotus.Registry) []Package{
return packages
}
func fetchPackages(platform string, architecture []string, query string) []Package{
logger := Lotus.DetailedLogger("ProtoHandler", "receiveSearchResponse")
func fetchPackages(platform string, architecture []string, query string, registry Lotus.Registry) *proto.SearchResponse {
logger := Lotus.DetailedLogger("ProtoHandler", "fetchPackages")
regisitries := Lotus.GetRegistries()
var packages []Package
for _, reg := range regisitries{
conn := createConnection(reg.Address)
conn := createConnection(registry.Address)
defer conn.Close()
client, ctx, cancel := createRegistryClient(conn)
defer cancel()
res, connErr := client.Search(ctx, &proto.SearchRequest{Platform: platform, Architecture: architecture, Query: query})
if connErr != nil{
logger.Fatal(connErr)
logger.Error(connErr)
}
for _, pkg := range parsePackages(res, reg){
packages = append(packages, pkg)
}
cancel()
conn.Close()
}
return packages
return res
}
func SearchPackages(platform string, architecture []string, query string) []Package {
return fetchPackages(platform, architecture, query)
registries := Lotus.GetRegistries()
var packages []Package
for _, reg := range registries{
res := fetchPackages(platform, architecture, query, reg)
if res != nil{
for _, pkg := range parsePackages(res, reg){
packages = append(packages, pkg)
}
}
}
return packages
}