operator-framework 0.6.2

Tools and helpers used to create Kubernetes operators
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

/*
 * Copyright (c) 2020, 2021 Jens Reimann and others.
 *
 * See the NOTICE file(s) distributed with this work for additional
 * information regarding copyright ownership.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0
 *
 * SPDX-License-Identifier: EPL-2.0
 */

use async_trait::async_trait;
use either::Either::{Left, Right};
use futures::future::FutureExt;
use kube::{
    api::{DeleteParams, Preconditions},
    Api, Error, Resource,
};
use serde::de::DeserializeOwned;
use std::fmt::Debug;

#[async_trait]
pub trait Delete<R: Send> {
    /// Optionally delete a resource. If the resource was already gone, this is not treated as an error.
    ///
    /// The function will return `true` if the resource was deleted (or already gone) and `false` if
    /// the resource is being delete. All other errors are returned unmodified.
    async fn delete_optionally<S>(&self, name: S, dp: &DeleteParams) -> Result<bool, kube::Error>
    where
        S: AsRef<str> + Send + Sync;

    async fn delete_conditionally<F, E, S>(&self, name: S, f: F) -> Result<bool, E>
    where
        F: FnOnce(&R) -> Result<bool, E> + Send,
        E: From<kube::Error>,
        S: AsRef<str> + Send + Sync;
}

#[async_trait]
impl<K> Delete<K> for Api<K>
where
    K: Resource + Clone + DeserializeOwned + Send + Debug,
{
    async fn delete_optionally<S>(&self, name: S, dp: &DeleteParams) -> Result<bool, kube::Error>
    where
        S: AsRef<str> + Send + Sync,
    {
        Ok(self
            .delete(name.as_ref(), dp)
            .map(|future| {
                future
                    .map(|either| match either {
                        Left(_) => false,
                        Right(_) => true,
                    })
                    .or_else(|err| match err {
                        Error::Api(cause) if cause.reason == "NotFound" => Ok(true),
                        _ => Err(err),
                    })
            })
            .await?)
    }

    async fn delete_conditionally<F, E, S>(&self, name: S, f: F) -> Result<bool, E>
    where
        F: FnOnce(&K) -> Result<bool, E> + Send,
        E: From<kube::Error>,
        S: AsRef<str> + Send + Sync,
    {
        let resource = match self.get(name.as_ref()).await {
            Err(Error::Api(cause)) if cause.reason == "NotFound" => return Ok(false),
            result => result?,
        };

        if f(&resource)? {
            let dp = DeleteParams {
                preconditions: Some(Preconditions {
                    resource_version: resource.meta().resource_version.as_ref().cloned(),
                    uid: resource.meta().uid.as_ref().cloned(),
                }),
                ..Default::default()
            };

            self.delete_optionally(name, &dp).await?;

            Ok(true)
        } else {
            Ok(false)
        }
    }
}