grid-sdk 0.1.3

Hyperledger Grid is a platform for building supply chain solutions that include distributed ledger components. It provides a growing set of tools that accelerate development for supply chain smart contractsand client interfaces.
Documentation
// Copyright 2018-2020 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.

use super::CommitStoreOperations;
use crate::commits::store::diesel::models::{CommitModel, NewCommitModel};
use crate::commits::store::diesel::{schema::commits, CommitStoreError};
use crate::error::{ConstraintViolationError, ConstraintViolationType, InternalError};

use diesel::{
    dsl::insert_into,
    prelude::*,
    result::{DatabaseErrorKind, Error as dsl_error},
};

pub(in crate::commits) trait CommitStoreAddCommitOperation {
    fn add_commit(&self, commit: NewCommitModel) -> Result<(), CommitStoreError>;
}

#[cfg(feature = "postgres")]
impl<'a> CommitStoreAddCommitOperation for CommitStoreOperations<'a, diesel::pg::PgConnection> {
    fn add_commit(&self, commit: NewCommitModel) -> Result<(), CommitStoreError> {
        let duplicate_commit = commits::table
            .filter(commits::commit_id.eq(&commit.commit_id))
            .first::<CommitModel>(self.conn)
            .map(Some)
            .or_else(|err| {
                if err == dsl_error::NotFound {
                    Ok(None)
                } else {
                    Err(err)
                }
            })
            .map_err(|err| {
                CommitStoreError::InternalError(InternalError::from_source(Box::new(err)))
            })?;
        if duplicate_commit.is_some() {
            return Err(CommitStoreError::ConstraintViolationError(
                ConstraintViolationError::with_violation_type(ConstraintViolationType::Unique),
            ));
        }

        insert_into(commits::table)
            .values(commit)
            .execute(self.conn)
            .map(|_| ())
            .map_err(|err| match err {
                dsl_error::DatabaseError(DatabaseErrorKind::UniqueViolation, _) => {
                    CommitStoreError::ConstraintViolationError(
                        ConstraintViolationError::from_source_with_violation_type(
                            ConstraintViolationType::Unique,
                            Box::new(err),
                        ),
                    )
                }
                dsl_error::DatabaseError(DatabaseErrorKind::ForeignKeyViolation, _) => {
                    CommitStoreError::ConstraintViolationError(
                        ConstraintViolationError::from_source_with_violation_type(
                            ConstraintViolationType::ForeignKey,
                            Box::new(err),
                        ),
                    )
                }
                _ => CommitStoreError::InternalError(InternalError::from_source(Box::new(err))),
            })?;
        Ok(())
    }
}

#[cfg(feature = "sqlite")]
impl<'a> CommitStoreAddCommitOperation
    for CommitStoreOperations<'a, diesel::sqlite::SqliteConnection>
{
    fn add_commit(&self, commit: NewCommitModel) -> Result<(), CommitStoreError> {
        let duplicate_commit = commits::table
            .filter(commits::commit_id.eq(&commit.commit_id))
            .first::<CommitModel>(self.conn)
            .map(Some)
            .or_else(|err| {
                if err == dsl_error::NotFound {
                    Ok(None)
                } else {
                    Err(err)
                }
            })
            .map_err(|err| {
                CommitStoreError::InternalError(InternalError::from_source(Box::new(err)))
            })?;
        if duplicate_commit.is_some() {
            return Err(CommitStoreError::ConstraintViolationError(
                ConstraintViolationError::with_violation_type(ConstraintViolationType::Unique),
            ));
        }

        insert_into(commits::table)
            .values(commit)
            .execute(self.conn)
            .map(|_| ())
            .map_err(|err| match err {
                dsl_error::DatabaseError(DatabaseErrorKind::UniqueViolation, _) => {
                    CommitStoreError::ConstraintViolationError(
                        ConstraintViolationError::from_source_with_violation_type(
                            ConstraintViolationType::Unique,
                            Box::new(err),
                        ),
                    )
                }
                dsl_error::DatabaseError(DatabaseErrorKind::ForeignKeyViolation, _) => {
                    CommitStoreError::ConstraintViolationError(
                        ConstraintViolationError::from_source_with_violation_type(
                            ConstraintViolationType::ForeignKey,
                            Box::new(err),
                        ),
                    )
                }
                _ => CommitStoreError::InternalError(InternalError::from_source(Box::new(err))),
            })?;
        Ok(())
    }
}