Examples with NextAction oracle.kubernetes.operator.work.NextAction used on opensource projects

Example 1 with NextAction

use of oracle.kubernetes.operator.work.NextAction in project weblogic-kubernetes-operator by oracle.

the class Main method begin.

private static void begin() {
    // read the operator configuration
    String namespace = System.getenv("OPERATOR_NAMESPACE");
    if (namespace == null) {
        namespace = "default";
    }
    Collection<String> targetNamespaces = getTargetNamespaces(namespace);
    String serviceAccountName = tuningAndConfig.get("serviceaccount");
    if (serviceAccountName == null) {
        serviceAccountName = "default";
    }
    principal = "system:serviceaccount:" + namespace + ":" + serviceAccountName;
    LOGGER.info(MessageKeys.OP_CONFIG_NAMESPACE, namespace);
    StringBuilder tns = new StringBuilder();
    Iterator<String> it = targetNamespaces.iterator();
    while (it.hasNext()) {
        tns.append(it.next());
        if (it.hasNext()) {
            tns.append(", ");
        }
    }
    LOGGER.info(MessageKeys.OP_CONFIG_TARGET_NAMESPACES, tns.toString());
    LOGGER.info(MessageKeys.OP_CONFIG_SERVICE_ACCOUNT, serviceAccountName);
    try {
        // Initialize logging factory with JSON serializer for later logging
        // that includes k8s objects
        LoggingFactory.setJSON(new JSON());
        // start the REST server
        startRestServer(principal, targetNamespaces);
        // create the Custom Resource Definitions if they are not already there
        CRDHelper.checkAndCreateCustomResourceDefinition();
        try {
            HealthCheckHelper healthCheck = new HealthCheckHelper(namespace, targetNamespaces);
            version = healthCheck.performK8sVersionCheck();
            healthCheck.performNonSecurityChecks();
            healthCheck.performSecurityChecks(serviceAccountName);
        } catch (ApiException e) {
            LOGGER.warning(MessageKeys.EXCEPTION, e);
        }
        // check for any existing resources and add the watches on them
        // this would happen when the Domain was running BEFORE the Operator starts up
        LOGGER.info(MessageKeys.LISTING_DOMAINS);
        for (String ns : targetNamespaces) {
            initialized.put(ns, Boolean.TRUE);
            Step domainList = callBuilderFactory.create().listDomainAsync(ns, new ResponseStep<DomainList>(null) {

                @Override
                public NextAction onFailure(Packet packet, ApiException e, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (statusCode == CallBuilder.NOT_FOUND) {
                        return onSuccess(packet, null, statusCode, responseHeaders);
                    }
                    return super.onFailure(packet, e, statusCode, responseHeaders);
                }

                @Override
                public NextAction onSuccess(Packet packet, DomainList result, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (result != null) {
                        for (Domain dom : result.getItems()) {
                            doCheckAndCreateDomainPresence(dom);
                        }
                    }
                    // main logic now happens in the watch handlers
                    domainWatchers.put(ns, createDomainWatcher(ns, result != null ? result.getMetadata().getResourceVersion() : ""));
                    return doNext(packet);
                }
            });
            Step initialize = ConfigMapHelper.createScriptConfigMapStep(ns, new ConfigMapAfterStep(ns, callBuilderFactory.create().with($ -> {
                $.labelSelector = LabelConstants.DOMAINUID_LABEL + "," + LabelConstants.CREATEDBYOPERATOR_LABEL;
            }).listPodAsync(ns, new ResponseStep<V1PodList>(callBuilderFactory.create().with($ -> {
                $.labelSelector = LabelConstants.DOMAINUID_LABEL + "," + LabelConstants.CREATEDBYOPERATOR_LABEL;
            }).listServiceAsync(ns, new ResponseStep<V1ServiceList>(callBuilderFactory.create().with($ -> {
                $.labelSelector = LabelConstants.DOMAINUID_LABEL + "," + LabelConstants.CREATEDBYOPERATOR_LABEL;
            }).listIngressAsync(ns, new ResponseStep<V1beta1IngressList>(domainList) {

                @Override
                public NextAction onFailure(Packet packet, ApiException e, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (statusCode == CallBuilder.NOT_FOUND) {
                        return onSuccess(packet, null, statusCode, responseHeaders);
                    }
                    return super.onFailure(packet, e, statusCode, responseHeaders);
                }

                @Override
                public NextAction onSuccess(Packet packet, V1beta1IngressList result, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (result != null) {
                        for (V1beta1Ingress ingress : result.getItems()) {
                            String domainUID = IngressWatcher.getIngressDomainUID(ingress);
                            String clusterName = IngressWatcher.getIngressClusterName(ingress);
                            if (domainUID != null && clusterName != null) {
                                DomainPresenceInfo created = new DomainPresenceInfo(ns);
                                DomainPresenceInfo info = domains.putIfAbsent(domainUID, created);
                                if (info == null) {
                                    info = created;
                                }
                                info.getIngresses().put(clusterName, ingress);
                            }
                        }
                    }
                    ingressWatchers.put(ns, createIngressWatcher(ns, result != null ? result.getMetadata().getResourceVersion() : ""));
                    return doNext(packet);
                }
            })) {

                @Override
                public NextAction onFailure(Packet packet, ApiException e, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (statusCode == CallBuilder.NOT_FOUND) {
                        return onSuccess(packet, null, statusCode, responseHeaders);
                    }
                    return super.onFailure(packet, e, statusCode, responseHeaders);
                }

                @Override
                public NextAction onSuccess(Packet packet, V1ServiceList result, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (result != null) {
                        for (V1Service service : result.getItems()) {
                            String domainUID = ServiceWatcher.getServiceDomainUID(service);
                            String serverName = ServiceWatcher.getServiceServerName(service);
                            String channelName = ServiceWatcher.getServiceChannelName(service);
                            if (domainUID != null && serverName != null) {
                                DomainPresenceInfo created = new DomainPresenceInfo(ns);
                                DomainPresenceInfo info = domains.putIfAbsent(domainUID, created);
                                if (info == null) {
                                    info = created;
                                }
                                ServerKubernetesObjects csko = new ServerKubernetesObjects();
                                ServerKubernetesObjects current = info.getServers().putIfAbsent(serverName, csko);
                                ServerKubernetesObjects sko = current != null ? current : csko;
                                if (channelName != null) {
                                    sko.getChannels().put(channelName, service);
                                } else {
                                    sko.getService().set(service);
                                }
                            }
                        }
                    }
                    serviceWatchers.put(ns, createServiceWatcher(ns, result != null ? result.getMetadata().getResourceVersion() : ""));
                    return doNext(packet);
                }
            })) {

                @Override
                public NextAction onFailure(Packet packet, ApiException e, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (statusCode == CallBuilder.NOT_FOUND) {
                        return onSuccess(packet, null, statusCode, responseHeaders);
                    }
                    return super.onFailure(packet, e, statusCode, responseHeaders);
                }

                @Override
                public NextAction onSuccess(Packet packet, V1PodList result, int statusCode, Map<String, List<String>> responseHeaders) {
                    if (result != null) {
                        for (V1Pod pod : result.getItems()) {
                            String domainUID = PodWatcher.getPodDomainUID(pod);
                            String serverName = PodWatcher.getPodServerName(pod);
                            if (domainUID != null && serverName != null) {
                                DomainPresenceInfo created = new DomainPresenceInfo(ns);
                                DomainPresenceInfo info = domains.putIfAbsent(domainUID, created);
                                if (info == null) {
                                    info = created;
                                }
                                ServerKubernetesObjects csko = new ServerKubernetesObjects();
                                ServerKubernetesObjects current = info.getServers().putIfAbsent(serverName, csko);
                                ServerKubernetesObjects sko = current != null ? current : csko;
                                sko.getPod().set(pod);
                            }
                        }
                    }
                    podWatchers.put(ns, createPodWatcher(ns, result != null ? result.getMetadata().getResourceVersion() : ""));
                    return doNext(packet);
                }
            })));
            engine.createFiber().start(initialize, new Packet(), new CompletionCallback() {

                @Override
                public void onCompletion(Packet packet) {
                // no-op
                }

                @Override
                public void onThrowable(Packet packet, Throwable throwable) {
                    LOGGER.severe(MessageKeys.EXCEPTION, throwable);
                }
            });
        }
        // delete stranded resources
        for (Map.Entry<String, DomainPresenceInfo> entry : domains.entrySet()) {
            String domainUID = entry.getKey();
            DomainPresenceInfo info = entry.getValue();
            if (info != null) {
                if (info.getDomain() == null) {
                    // no domain resource
                    deleteDomainPresence(info.getNamespace(), domainUID);
                }
            }
        }
    } catch (Throwable e) {
        LOGGER.warning(MessageKeys.EXCEPTION, e);
    } finally {
        LOGGER.info(MessageKeys.OPERATOR_SHUTTING_DOWN);
    }
}

Example 2 with NextAction

use of oracle.kubernetes.operator.work.NextAction in project weblogic-kubernetes-operator by oracle.

the class ResponseStep method apply.

@Override
public final NextAction apply(Packet packet) {
    NextAction nextAction = null;
    @SuppressWarnings("unchecked") CallResponse<T> callResponse = packet.getSPI(CallResponse.class);
    if (callResponse != null) {
        if (callResponse.result != null) {
            // success
            nextAction = onSuccess(packet, callResponse.result, callResponse.statusCode, callResponse.responseHeaders);
        }
        if (callResponse.e != null) {
            // exception
            nextAction = onFailure(packet, callResponse.e, callResponse.statusCode, callResponse.responseHeaders);
        }
    }
    if (nextAction == null) {
        // call timed-out
        nextAction = doPotentialRetry(null, packet, null, 0, null);
        if (nextAction == null) {
            nextAction = doEnd(packet);
        }
    }
    if (previousStep != nextAction.getNext()) {
        // not a retry, clear out old response
        packet.getComponents().remove(CallBuilder.RESPONSE_COMPONENT_NAME);
    }
    return nextAction;
}

Example 3 with NextAction

use of oracle.kubernetes.operator.work.NextAction in project weblogic-kubernetes-operator by oracle.

the class DomainStatusUpdater method doDomainUpdate.

private static NextAction doDomainUpdate(Domain dom, DomainPresenceInfo info, Packet packet, Step conflictStep, Step next) {
    V1ObjectMeta meta = dom.getMetadata();
    NextAction na = new NextAction();
    CallBuilderFactory factory = ContainerResolver.getInstance().getContainer().getSPI(CallBuilderFactory.class);
    na.invoke(factory.create().replaceDomainAsync(meta.getName(), meta.getNamespace(), dom, new ResponseStep<Domain>(next) {

        @Override
        public NextAction onFailure(Packet packet, ApiException e, int statusCode, Map<String, List<String>> responseHeaders) {
            if (statusCode == CallBuilder.NOT_FOUND) {
                // Just ignore update
                return doNext(packet);
            }
            return super.onFailure(conflictStep, packet, e, statusCode, responseHeaders);
        }

        @Override
        public NextAction onSuccess(Packet packet, Domain result, int statusCode, Map<String, List<String>> responseHeaders) {
            info.setDomain(result);
            return doNext(packet);
        }
    }), packet);
    return na;
}

Example 4 with NextAction

use of oracle.kubernetes.operator.work.NextAction in project weblogic-kubernetes-operator by oracle.

the class PodWatcherTest method WhenWaitForReadyAppliedToReadyPod_performNextStep.

@Test
public void WhenWaitForReadyAppliedToReadyPod_performNextStep() throws Exception {
    AtomicBoolean stopping = new AtomicBoolean(false);
    PodWatcher watcher = PodWatcher.create(Executors.defaultThreadFactory(), "ns", Integer.toString(INITIAL_RESOURCE_VERSION), this, stopping);
    makePodReady(pod);
    ListeningTerminalStep listeningStep = new ListeningTerminalStep(stopping);
    Step step = watcher.waitForReady(pod, listeningStep);
    NextAction nextAction = step.apply(packet);
    nextAction.getNext().apply(packet);
    assertThat(listeningStep.wasPerformed, is(true));
}