Updated Handler

This commit is contained in:
netbenix 2021-11-07 21:16:32 +01:00
parent 3a5166eece
commit eddda81fcf

View file

@ -18,26 +18,26 @@ func createConnection() *grpc.ClientConn{
logger.Fatal(err) logger.Fatal(err)
} }
defer conn.Close()
return conn return conn
} }
func createRegistryClient() (proto.LpmRegistryClient, context.Context){ func createRegistryClient(conn *grpc.ClientConn) (proto.LpmRegistryClient, context.Context, context.CancelFunc){
lrc := proto.NewLpmRegistryClient(createConnection()) lrc := proto.NewLpmRegistryClient(conn)
ctx, cancel := context.WithTimeout(context.Background(), time.Second * 10) ctx, cancel := context.WithTimeout(context.Background(), time.Second * 10)
defer cancel() return lrc, ctx, cancel
return lrc, ctx
} }
func SearchPackages(platform string, architecture []string, query string) []PackageMetadata { func SearchPackages(platform string, architecture []string, query string) []PackageMetadata {
logger := Lotus.DetailedLogger("ProtoHandler", "sendSearchRequest") logger := Lotus.DetailedLogger("ProtoHandler", "sendSearchRequest")
client, ctx := createRegistryClient() conn := createConnection()
defer conn.Close()
client, ctx, cancel := createRegistryClient(conn)
defer cancel()
res, connErr := client.Search(ctx, &proto.SearchRequest{Platform: platform, Architecture: architecture, Query: query}) res, connErr := client.Search(ctx, &proto.SearchRequest{Platform: platform, Architecture: architecture, Query: query})
if connErr != nil{ if connErr != nil{
logger.Fatal(connErr) logger.Fatal(connErr)