#ifndef CONVEYOR_BELT_GRPC_SERVER_H_
#define CONVEYOR_BELT_GRPC_SERVER_H_
#include <cstdlib>
#include <iostream>
#include <memory>
#include <string>
#include <thread>

#include "conveyorBelt.grpc.pb.h"
#include "conveyorBelt.pb.h"
#include "conveyorBeltControlUnit.h"
#include <glog/logging.h>
#include <grpcpp/grpcpp.h>

using grpc::Server;
using grpc::ServerBuilder;
using grpc::ServerContext;
using grpc::Status;

using ConveyorBelt::ConveyorBeltService;

typedef int RunningState;

class ConveyorBeltServer : public ConveyorBeltService::Service {
private:
    /* data */
public:
    ::grpc::Status open(
        ::grpc::ServerContext* context,
        const ::ConveyorBelt::SpeedInfo* request,
        ::ConveyorBelt::Response* response
    ) override;
    ::grpc::Status close(
        ::grpc::ServerContext* context,
        const ::ConveyorBelt::DeviceID* request,
        ::ConveyorBelt::Response* response
    ) override;
    ::grpc::Status setSpeed(
        ::grpc::ServerContext* context,
        const ::ConveyorBelt::SpeedInfo* request,
        ::ConveyorBelt::Response* response
    ) override;
    ::grpc::Status setDirection(
        ::grpc::ServerContext* context,
        const ::ConveyorBelt::DirectionInfo* request,
        ::ConveyorBelt::Response* response
    ) override;
    ::grpc::Status getWorkState(
        ::grpc::ServerContext* context,
        const ::ConveyorBelt::DeviceID* request,
        ::ConveyorBelt::Response* response
    ) override;
    ConveyorBeltServer();
    ~ConveyorBeltServer();
};

#endif