Examples with V1OwnerReference io.kubernetes.client.models.V1OwnerReference used on opensource projects

Example 1 with V1OwnerReference

use of io.kubernetes.client.models.V1OwnerReference in project seldon-core by SeldonIO.

the class DeploymentWatcher method watchDeployments.

public int watchDeployments(int resourceVersion, int resourceVersionProcessed) throws ApiException, IOException {
    String rs = null;
    if (resourceVersion > 0)
        rs = "" + resourceVersion;
    logger.debug("Watching with rs " + rs);
    int maxResourceVersion = resourceVersion;
    ApiClient client = Config.defaultClient();
    ExtensionsV1beta1Api api = new ExtensionsV1beta1Api(client);
    Watch<ExtensionsV1beta1Deployment> watch = Watch.createWatch(client, api.listNamespacedDeploymentCall(namespace, null, null, null, false, SeldonDeploymentOperatorImpl.LABEL_SELDON_TYPE_KEY + "=" + SeldonDeploymentOperatorImpl.LABEL_SELDON_TYPE_VAL, null, rs, 10, true, null, null), new TypeToken<Watch.Response<ExtensionsV1beta1Deployment>>() {
    }.getType());
    try {
        for (Watch.Response<ExtensionsV1beta1Deployment> item : watch) {
            int resourceVersionNew = Integer.parseInt(item.object.getMetadata().getResourceVersion());
            if (resourceVersionNew <= resourceVersionProcessed) {
                logger.warn("Looking at already processed request - skipping");
            } else {
                if (resourceVersionNew > maxResourceVersion)
                    maxResourceVersion = resourceVersionNew;
                if (item.object.getMetadata().getOwnerReferences() == null) {
                    logger.warn("Found possible seldon controlled deployment which has no owner reference. Ignoring.");
                } else {
                    switch(item.type) {
                        case "ADDED":
                        case "MODIFIED":
                            for (V1OwnerReference ownerRef : item.object.getMetadata().getOwnerReferences()) {
                                if (ownerRef.getKind().equals(KubeCRDHandlerImpl.KIND) && item.object.getStatus() != null) {
                                    String mlDepName = ownerRef.getName();
                                    String depName = item.object.getMetadata().getName();
                                    statusUpdater.updateStatus(mlDepName, depName, item.object.getStatus().getReplicas(), item.object.getStatus().getReadyReplicas());
                                }
                            }
                            break;
                        case "DELETED":
                            for (V1OwnerReference ownerRef : item.object.getMetadata().getOwnerReferences()) {
                                if (ownerRef.getKind().equals(KubeCRDHandlerImpl.KIND) && item.object.getStatus() != null) {
                                    String mlDepName = ownerRef.getName();
                                    String depName = item.object.getMetadata().getName();
                                    statusUpdater.removeStatus(mlDepName, depName);
                                }
                            }
                            break;
                        default:
                            logger.error("Unknown type " + item.type);
                    }
                // for modified get owner reference and determine which predictor it is
                // get the MLDeployment from API or local cache and update status
                // put this logic in new class
                }
            }
        }
    } catch (RuntimeException e) {
        if (e.getCause() instanceof SocketTimeoutException)
            return maxResourceVersion;
        else
            throw e;
    } finally {
        watch.close();
    }
    return maxResourceVersion;
}