refactor: rename PolicyServer to CheckService

1 files changed, 3 insertions, 6 deletions

diff --git a/src/main.rs b/src/main.rs
index 25e2c88f..917cace5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,16 +2,13 @@ use envoy_types::ext_authz::v3::pb::AuthorizationServer;
 use tonic::transport::Server;
 
 pub mod authorization;
-use authorization::PolicyServer;
+use authorization::CheckService;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let addr = "[::1]:50051".parse()?;
-    let authorization_service = AuthorizationServer::new(PolicyServer::default());
-    let (health_reporter, health_service) = tonic_health::server::health_reporter();
-    health_reporter
-        .set_serving::<AuthorizationServer<PolicyServer>>()
-        .await;
+    let authorization_service = AuthorizationServer::new(CheckService::default());
+    let (_health_reporter, health_service) = tonic_health::server::health_reporter();
 
     let reflection_service = tonic_reflection::server::Builder::configure()
         .register_encoded_file_descriptor_set(tonic_health::pb::FILE_DESCRIPTOR_SET)