splinter 0.6.14

// Copyright 2018-2022 Cargill Incorporated
//
// 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.

//! Provides database operations for the `DieselRegistry`.

pub(super) mod add_node;
pub(super) mod count_nodes;
pub(super) mod delete_node;
pub(super) mod get_node;
pub(super) mod has_node;
pub(super) mod list_nodes;
pub(super) mod update_node;

use diesel::{
    dsl::{exists, not},
    prelude::*,
};

use crate::registry::diesel::schema::{splinter_nodes, splinter_nodes_metadata};
use crate::registry::MetadataPredicate;

pub struct RegistryOperations<'a, C> {
    conn: &'a C,
}

impl<'a, C: diesel::Connection> RegistryOperations<'a, C> {
    pub fn new(conn: &'a C) -> Self {
        RegistryOperations { conn }
    }
}

type BoxedNodeQuery<'a, DB> = diesel::query_builder::BoxedSelectStatement<
    'a,
    (diesel::sql_types::Text, diesel::sql_types::Text),
    splinter_nodes::table,
    DB,
>;

/// Takes a query of type `splinter_nodes::table.into_boxed().select(splinter_nodes::all_columns)`
/// and updates the query for any provided predicates
fn apply_predicate_filters<'a, DB: 'a>(
    query: BoxedNodeQuery<'a, DB>,
    predicates: &'a [MetadataPredicate],
) -> BoxedNodeQuery<'a, DB>
where
    DB: diesel::backend::Backend,
{
    let mut query = query;
    for predicate in predicates {
        match predicate {
            MetadataPredicate::Eq(key, val) => {
                // only return values that have key = value
                query = query.filter(exists(
                    splinter_nodes_metadata::table.filter(
                        splinter_nodes_metadata::identity
                            .eq(splinter_nodes::identity)
                            .and(splinter_nodes_metadata::key.eq(key))
                            .and(splinter_nodes_metadata::value.eq(val)),
                    ),
                ));
            }
            MetadataPredicate::Ne(key, val) => {
                // only return values that have key != value or do not have the key
                query = query.filter(
                    not(exists(
                        splinter_nodes_metadata::table.filter(
                            splinter_nodes_metadata::identity
                                .eq(splinter_nodes::identity)
                                .and(splinter_nodes_metadata::key.eq(key)),
                        ),
                    ))
                    .or(exists(
                        splinter_nodes_metadata::table.filter(
                            splinter_nodes_metadata::identity
                                .eq(splinter_nodes::identity)
                                .and(splinter_nodes_metadata::key.eq(key))
                                .and(splinter_nodes_metadata::value.ne(val)),
                        ),
                    )),
                );
            }
            MetadataPredicate::Gt(key, val) => {
                query = query.filter(exists(
                    splinter_nodes_metadata::table.filter(
                        splinter_nodes_metadata::identity
                            .eq(splinter_nodes::identity)
                            .and(splinter_nodes_metadata::key.eq(key))
                            .and(splinter_nodes_metadata::value.gt(val)),
                    ),
                ));
            }

            MetadataPredicate::Ge(key, val) => {
                query = query.filter(exists(
                    splinter_nodes_metadata::table.filter(
                        splinter_nodes_metadata::identity
                            .eq(splinter_nodes::identity)
                            .and(splinter_nodes_metadata::key.eq(key))
                            .and(splinter_nodes_metadata::value.ge(val)),
                    ),
                ));
            }

            MetadataPredicate::Lt(key, val) => {
                query = query.filter(exists(
                    splinter_nodes_metadata::table.filter(
                        splinter_nodes_metadata::identity
                            .eq(splinter_nodes::identity)
                            .and(splinter_nodes_metadata::key.eq(key))
                            .and(splinter_nodes_metadata::value.lt(val)),
                    ),
                ));
            }

            MetadataPredicate::Le(key, val) => {
                query = query.filter(exists(
                    splinter_nodes_metadata::table.filter(
                        splinter_nodes_metadata::identity
                            .eq(splinter_nodes::identity)
                            .and(splinter_nodes_metadata::key.eq(key))
                            .and(splinter_nodes_metadata::value.le(val)),
                    ),
                ));
            }
        }
    }

    query
}