Compare commits
No commits in common. "38a73fd0034d7ee613716f20797eadf13581f1d8" and "0ccaa4840cf23783fdac17eeba8ce99d2ae69ea8" have entirely different histories.
38a73fd003
...
0ccaa4840c
@ -94,7 +94,7 @@ impl NewAppReq {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
#[derive(Debug, Serialize, Deserialize)]
|
||||
pub struct AppNodeWithReports {
|
||||
pub id: RecordId,
|
||||
pub operator: RecordId,
|
||||
@ -116,7 +116,6 @@ impl AppNodeWithReports {
|
||||
pub async fn find_by_filters(
|
||||
db: &Surreal<Client>,
|
||||
filters: app_proto::AppNodeFilters,
|
||||
limit_one: bool,
|
||||
) -> Result<Vec<Self>, Error> {
|
||||
let mut filter_query = format!(
|
||||
"select *, <-report.* from {APP_NODE} where
|
||||
@ -124,7 +123,7 @@ impl AppNodeWithReports {
|
||||
max_ports_per_app >= {} &&
|
||||
avail_vcpus >= {} &&
|
||||
avail_mem_mb >= {} &&
|
||||
avail_storage_gbs >= {} ",
|
||||
avail_storage_gbs >= {}\n",
|
||||
filters.free_ports,
|
||||
filters.free_ports,
|
||||
filters.vcpus,
|
||||
@ -144,12 +143,8 @@ impl AppNodeWithReports {
|
||||
if !filters.ip.is_empty() {
|
||||
filter_query += &format!("&& ip = '{}' ", filters.ip);
|
||||
}
|
||||
|
||||
if limit_one {
|
||||
filter_query += "limit 1";
|
||||
}
|
||||
|
||||
filter_query += ";";
|
||||
dbg!(&filter_query);
|
||||
let mut query_resp = db.query(filter_query).await?;
|
||||
let app_nodes: Vec<Self> = query_resp.take(0)?;
|
||||
Ok(app_nodes)
|
||||
@ -233,12 +228,6 @@ pub struct ActiveAppWithNode {
|
||||
}
|
||||
|
||||
impl ActiveAppWithNode {
|
||||
pub async fn get_by_uuid(db: &Surreal<Client>, uuid: &str) -> Result<Option<Self>, Error> {
|
||||
let contract: Option<Self> =
|
||||
db.query(format!("select * from {ACTIVE_APP}:{uuid} fetch out;")).await?.take(0)?;
|
||||
Ok(contract)
|
||||
}
|
||||
|
||||
pub async fn list_by_node(db: &Surreal<Client>, node_pubkey: &str) -> Result<Vec<Self>, Error> {
|
||||
let mut query_result = db
|
||||
.query(format!(
|
||||
@ -250,38 +239,10 @@ impl ActiveAppWithNode {
|
||||
Ok(contract)
|
||||
}
|
||||
|
||||
pub async fn list_by_admin(db: &Surreal<Client>, admin: &str) -> Result<Vec<Self>, Error> {
|
||||
let mut query_result = db
|
||||
.query(format!("select * from {ACTIVE_APP} where in = {ACCOUNT}:{admin} fetch out;"))
|
||||
.await?;
|
||||
|
||||
let app_contracts: Vec<Self> = query_result.take(0)?;
|
||||
|
||||
Ok(app_contracts)
|
||||
}
|
||||
|
||||
pub async fn list_by_operator(
|
||||
db: &Surreal<Client>,
|
||||
operator: &str,
|
||||
) -> Result<Vec<Self>, Error> {
|
||||
let mut query_result = db
|
||||
.query(format!(
|
||||
"select
|
||||
(select * from ->operator->app_node<-{ACTIVE_APP} fetch out)
|
||||
as app_contracts
|
||||
from {ACCOUNT}:{operator}"
|
||||
))
|
||||
.await?;
|
||||
#[derive(Deserialize)]
|
||||
struct Wrapper {
|
||||
app_contracts: Vec<ActiveAppWithNode>,
|
||||
}
|
||||
|
||||
let c: Option<Wrapper> = query_result.take(0)?;
|
||||
match c {
|
||||
Some(contracts_wrapper) => Ok(contracts_wrapper.app_contracts),
|
||||
None => Ok(vec![]),
|
||||
}
|
||||
pub async fn get_by_uuid(db: &Surreal<Client>, uuid: &str) -> Result<Option<Self>, Error> {
|
||||
let contract: Option<Self> =
|
||||
db.query(format!("select * from {ACTIVE_APP}:{uuid} fetch out;")).await?.take(0)?;
|
||||
Ok(contract)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@ -113,7 +113,7 @@ pub struct Kick {
|
||||
contract: RecordId,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
#[derive(Debug, Serialize, Deserialize)]
|
||||
pub struct Report {
|
||||
#[serde(rename = "in")]
|
||||
from_account: RecordId,
|
||||
|
||||
@ -210,36 +210,7 @@ impl BrainAppCli for AppCliServer {
|
||||
let req = check_sig_from_req(req)?;
|
||||
info!("list_app_contracts process starting for {:?}", req);
|
||||
|
||||
let mut app_contracts = Vec::new();
|
||||
|
||||
if !req.uuid.is_empty() {
|
||||
if let Some(app_contract) =
|
||||
db::ActiveAppWithNode::get_by_uuid(&self.db, &req.uuid).await?
|
||||
{
|
||||
if app_contract.admin.key().to_string() == req.admin_pubkey {
|
||||
app_contracts.push(app_contract);
|
||||
}
|
||||
// TODO: allow operator to inspect contracts
|
||||
}
|
||||
} else if req.as_operator {
|
||||
app_contracts.append(
|
||||
&mut db::ActiveAppWithNode::list_by_operator(&self.db, &req.admin_pubkey).await?,
|
||||
);
|
||||
} else {
|
||||
app_contracts.append(
|
||||
&mut db::ActiveAppWithNode::list_by_admin(&self.db, &req.admin_pubkey).await?,
|
||||
);
|
||||
}
|
||||
|
||||
let (tx, rx) = mpsc::channel(6);
|
||||
tokio::spawn(async move {
|
||||
for app_contract in app_contracts {
|
||||
let _ = tx.send(Ok(app_contract.into())).await;
|
||||
}
|
||||
});
|
||||
|
||||
let resp_stream = ReceiverStream::new(rx);
|
||||
Ok(Response::new(Box::pin(resp_stream)))
|
||||
todo!()
|
||||
}
|
||||
|
||||
async fn list_app_nodes(
|
||||
@ -248,7 +219,7 @@ impl BrainAppCli for AppCliServer {
|
||||
) -> Result<tonic::Response<<Self as BrainAppCli>::ListAppNodesStream>, tonic::Status> {
|
||||
let req = check_sig_from_req(req)?;
|
||||
info!("list_app_nodes process starting for {:?}", req);
|
||||
let app_nodes = db::AppNodeWithReports::find_by_filters(&self.db, req, false).await?;
|
||||
let app_nodes = db::AppNodeWithReports::find_by_filters(&self.db, req).await?;
|
||||
let (tx, rx) = mpsc::channel(6);
|
||||
tokio::spawn(async move {
|
||||
for app_node in app_nodes {
|
||||
@ -265,12 +236,7 @@ impl BrainAppCli for AppCliServer {
|
||||
) -> Result<tonic::Response<AppNodeListResp>, tonic::Status> {
|
||||
let req = check_sig_from_req(req)?;
|
||||
info!("get_one_app_node process starting for {:?}", req);
|
||||
let app_node = db::AppNodeWithReports::find_by_filters(&self.db, req, true)
|
||||
.await?
|
||||
.first()
|
||||
.ok_or(Status::not_found("No app node found"))?
|
||||
.clone();
|
||||
|
||||
Ok(Response::new(app_node.into()))
|
||||
todo!()
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user