mlmodel_server.hpp

// Copyright 2023 Viam Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
#pragma once
 
#include <viam/api/common/v1/common.grpc.pb.h>
#include <viam/api/service/mlmodel/v1/mlmodel.grpc.pb.h>
 
#include <viam/sdk/resource/resource_manager.hpp>
#include <viam/sdk/resource/resource_server_base.hpp>
#include <viam/sdk/services/mlmodel.hpp>
 
namespace viam {
namespace sdk {
namespace impl {
 
class MLModelServiceServer : public ResourceServer,
                             public ::viam::service::mlmodel::v1::MLModelService::Service {
   public:
    using interface_type = MLModelService;
    using service_type = service::mlmodel::v1::MLModelService;
    explicit MLModelServiceServer(std::shared_ptr<ResourceManager> manager);
 
    ::grpc::Status Infer(::grpc::ServerContext* context,
                         const ::viam::service::mlmodel::v1::InferRequest* request,
                         ::viam::service::mlmodel::v1::InferResponse* response) noexcept override;
 
    ::grpc::Status Metadata(
        ::grpc::ServerContext* context,
        const ::viam::service::mlmodel::v1::MetadataRequest* request,
        ::viam::service::mlmodel::v1::MetadataResponse* response) noexcept override;
 
    ::grpc::Status GetStatus(::grpc::ServerContext* context,
                             const ::viam::common::v1::GetStatusRequest* request,
                             ::viam::common::v1::GetStatusResponse* response) noexcept override;
};
 
}  // namespace impl
}  // namespace sdk
}  // namespace viam