Line data    Source code

       1              : use control_plane::attachment_service::{NodeAvailability, NodeSchedulingPolicy};
       2              : use utils::id::NodeId;
       3              : 
       4              : use crate::persistence::NodePersistence;
       5              : 
       6          961 : #[derive(Clone)]
       7              : pub(crate) struct Node {
       8              :     pub(crate) id: NodeId,
       9              : 
      10              :     pub(crate) availability: NodeAvailability,
      11              :     pub(crate) scheduling: NodeSchedulingPolicy,
      12              : 
      13              :     pub(crate) listen_http_addr: String,
      14              :     pub(crate) listen_http_port: u16,
      15              : 
      16              :     pub(crate) listen_pg_addr: String,
      17              :     pub(crate) listen_pg_port: u16,
      18              : }
      19              : 
      20              : impl Node {
      21         1410 :     pub(crate) fn base_url(&self) -> String {
      22         1410 :         format!("http://{}:{}", self.listen_http_addr, self.listen_http_port)
      23         1410 :     }
      24              : 
      25              :     /// Is this node elegible to have work scheduled onto it?
      26         1433 :     pub(crate) fn may_schedule(&self) -> bool {
      27         1433 :         match self.availability {
      28         1429 :             NodeAvailability::Active => {}
      29            4 :             NodeAvailability::Offline => return false,
      30              :         }
      31              : 
      32         1429 :         match self.scheduling {
      33            0 :             NodeSchedulingPolicy::Active => true,
      34           13 :             NodeSchedulingPolicy::Draining => false,
      35         1416 :             NodeSchedulingPolicy::Filling => true,
      36            0 :             NodeSchedulingPolicy::Pause => false,
      37              :         }
      38         1433 :     }
      39              : 
      40          410 :     pub(crate) fn to_persistent(&self) -> NodePersistence {
      41          410 :         NodePersistence {
      42          410 :             node_id: self.id.0 as i64,
      43          410 :             scheduling_policy: self.scheduling.into(),
      44          410 :             listen_http_addr: self.listen_http_addr.clone(),
      45          410 :             listen_http_port: self.listen_http_port as i32,
      46          410 :             listen_pg_addr: self.listen_pg_addr.clone(),
      47          410 :             listen_pg_port: self.listen_pg_port as i32,
      48          410 :         }
      49          410 :     }
      50              : }