SEBA-796 report connection error messages
Change-Id: I66b3339c8f9b25147fc6b67756f1afc16e332506
diff --git a/internal/pkg/commands/orm.go b/internal/pkg/commands/orm.go
index 0ac846a..d09f15d 100644
--- a/internal/pkg/commands/orm.go
+++ b/internal/pkg/commands/orm.go
@@ -261,7 +261,7 @@
// Create a model in XOS given a map of fields
func CreateModel(conn *grpc.ClientConn, descriptor grpcurl.DescriptorSource, modelName string, fields map[string]interface{}) error {
- ctx, cancel := context.WithTimeout(context.Background(), GlobalConfig.Grpc.Timeout)
+ ctx, cancel := GrpcTimeoutContext(context.Background())
defer cancel()
headers := GenerateHeaders()
@@ -292,7 +292,7 @@
// Update a model in XOS given a map of fields
func UpdateModel(conn *grpc.ClientConn, descriptor grpcurl.DescriptorSource, modelName string, fields map[string]interface{}) error {
- ctx, cancel := context.WithTimeout(context.Background(), GlobalConfig.Grpc.Timeout)
+ ctx, cancel := GrpcTimeoutContext(context.Background())
defer cancel()
headers := GenerateHeaders()
@@ -320,7 +320,7 @@
// Get a model from XOS given its ID
func GetModel(ctx context.Context, conn *grpc.ClientConn, descriptor grpcurl.DescriptorSource, modelName string, id int32) (*dynamic.Message, error) {
- ctx, cancel := context.WithTimeout(ctx, GlobalConfig.Grpc.Timeout)
+ ctx, cancel := GrpcTimeoutContext(context.Background())
defer cancel()
headers := GenerateHeaders()
@@ -432,7 +432,7 @@
// List all objects of a given model
func ListModels(ctx context.Context, conn *grpc.ClientConn, descriptor grpcurl.DescriptorSource, modelName string) ([]*dynamic.Message, error) {
- ctx, cancel := context.WithTimeout(ctx, GlobalConfig.Grpc.Timeout)
+ ctx, cancel := GrpcTimeoutContext(context.Background())
defer cancel()
headers := GenerateHeaders()
@@ -465,7 +465,7 @@
// For example,
// map[string]string{"name": "==mysite"}
func FilterModels(ctx context.Context, conn *grpc.ClientConn, descriptor grpcurl.DescriptorSource, modelName string, kind string, queries map[string]string) ([]*dynamic.Message, error) {
- ctx, cancel := context.WithTimeout(ctx, GlobalConfig.Grpc.Timeout)
+ ctx, cancel := GrpcTimeoutContext(context.Background())
defer cancel()
headers := GenerateHeaders()
@@ -613,7 +613,7 @@
// Get a model from XOS given its ID
func DeleteModel(conn *grpc.ClientConn, descriptor grpcurl.DescriptorSource, modelName string, id int32) error {
- ctx, cancel := context.WithTimeout(context.Background(), GlobalConfig.Grpc.Timeout)
+ ctx, cancel := GrpcTimeoutContext(context.Background())
defer cancel()
headers := GenerateHeaders()