Skip to content

Commit b5578ce

Browse files
committed
error on model reg not found
1 parent 2332592 commit b5578ce

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

src/viam/sdk/module/service.cpp

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,8 @@ struct ModuleService::ServiceImpl : viam::module::v1::ModuleService::Service {
8383
} catch (const std::exception& exc) {
8484
return grpc::Status(::grpc::INTERNAL, exc.what());
8585
}
86-
};
86+
}
87+
8788
try {
8889
parent.server_->add_resource(res, ctx->deadline());
8990
} catch (const std::exception& exc) {
@@ -131,13 +132,16 @@ struct ModuleService::ServiceImpl : viam::module::v1::ModuleService::Service {
131132
const std::shared_ptr<const ModelRegistration> reg =
132133
Registry::get().lookup_model(cfg.api(), cfg.model());
133134

134-
if (reg) {
135-
try {
136-
std::shared_ptr<Resource> resource = reg->construct_resource(deps, cfg);
137-
manager->replace_one(cfg.resource_name(), std::move(resource));
138-
} catch (const std::exception& exc) {
139-
return grpc::Status(::grpc::INTERNAL, exc.what());
140-
}
135+
if (!reg) {
136+
return grpc::Status(::grpc::INTERNAL,
137+
"Unable to rebuild resource: model registration not found");
138+
}
139+
140+
try {
141+
std::shared_ptr<Resource> resource = reg->construct_resource(deps, cfg);
142+
manager->replace_one(cfg.resource_name(), std::move(resource));
143+
} catch (const std::exception& exc) {
144+
return grpc::Status(::grpc::INTERNAL, exc.what());
141145
}
142146

143147
return grpc::Status();

0 commit comments

Comments
 (0)