Examples with CompletionCallback oracle.kubernetes.operator.work.Fiber.CompletionCallback used on opensource projects

Example 1 with CompletionCallback

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

the class StepTest method testSuspendAndThrow.

@Test
public void testSuspendAndThrow() throws InterruptedException {
    Step stepline = Step.createStepline(Arrays.asList(Step1.class, Step2.class, Step3.class));
    Packet p = new Packet();
    Map<Class<? extends BaseStep>, Command> commandMap = new HashMap<>();
    commandMap.put(Step2.class, new Command(RELEASE_SEMAPHORE, ACQUIRE_SEMAPHORE, SUSPEND_AND_THROW));
    p.put(NA, commandMap);
    Semaphore releaseSemaphore = new Semaphore(0);
    Semaphore acquireSemaphore = new Semaphore(0);
    p.put(Step2.class.getName() + RELEASE_SEMAPHORE_SUFFIX, releaseSemaphore);
    p.put(Step2.class.getName() + ACQUIRE_SEMAPHORE_SUFFIX, acquireSemaphore);
    Semaphore signal = new Semaphore(0);
    List<Throwable> throwables = new ArrayList<Throwable>();
    engine.createFiber().start(stepline, p, new CompletionCallback() {

        @Override
        public void onCompletion(Packet packet) {
            signal.release();
        }

        @Override
        public void onThrowable(Packet packet, Throwable throwable) {
            throwables.add(throwable);
            signal.release();
        }
    });
    releaseSemaphore.acquire();
    // Fiber is inside Step2
    acquireSemaphore.release();
    @SuppressWarnings({ "unchecked", "rawtypes" }) List<Step> called = (List) p.get(MARK);
    boolean result = signal.tryAcquire(1, TimeUnit.SECONDS);
    assertFalse(result);
    assertEquals(2, called.size());
    assertTrue(called.get(0) instanceof Step1);
    assertTrue(called.get(1) instanceof Step2);
// assertEquals(1, throwables.size());
// assertTrue(throwables.get(0) instanceof NullPointerException);
}

Example 2 with CompletionCallback

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

the class StepTest method testThrow.

@Test
public void testThrow() throws InterruptedException {
    Step stepline = Step.createStepline(Arrays.asList(Step1.class, Step2.class, Step3.class));
    Packet p = new Packet();
    Map<Class<? extends BaseStep>, Command> commandMap = new HashMap<>();
    commandMap.put(Step2.class, new Command(THROW));
    p.put(NA, commandMap);
    Semaphore signal = new Semaphore(0);
    List<Step> called = new ArrayList<Step>();
    List<Throwable> throwables = new ArrayList<Throwable>();
    engine.createFiber().start(stepline, p, new CompletionCallback() {

        @SuppressWarnings({ "unchecked", "rawtypes" })
        @Override
        public void onCompletion(Packet packet) {
            List<Step> l = (List) packet.get(MARK);
            if (l != null) {
                called.addAll(l);
            }
            signal.release();
        }

        @SuppressWarnings({ "unchecked", "rawtypes" })
        @Override
        public void onThrowable(Packet packet, Throwable throwable) {
            List<Step> l = (List) packet.get(MARK);
            if (l != null) {
                called.addAll(l);
            }
            throwables.add(throwable);
            signal.release();
        }
    });
    boolean result = signal.tryAcquire(5, TimeUnit.SECONDS);
    assertTrue(result);
    assertEquals(2, called.size());
    assertTrue(called.get(0) instanceof Step1);
    assertTrue(called.get(1) instanceof Step2);
    assertEquals(1, throwables.size());
    assertTrue(throwables.get(0) instanceof NullPointerException);
}

Example 3 with CompletionCallback

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

the class StepTest method testCancel.

@Test
public void testCancel() throws InterruptedException {
    Step stepline = Step.createStepline(Arrays.asList(Step1.class, Step2.class, Step3.class));
    Packet p = new Packet();
    Map<Class<? extends BaseStep>, Command> commandMap = new HashMap<>();
    commandMap.put(Step2.class, new Command(RELEASE_SEMAPHORE, ACQUIRE_SEMAPHORE, INVOKE_NEXT));
    p.put(NA, commandMap);
    Semaphore releaseSemaphore = new Semaphore(0);
    Semaphore acquireSemaphore = new Semaphore(0);
    p.put(Step2.class.getName() + RELEASE_SEMAPHORE_SUFFIX, releaseSemaphore);
    p.put(Step2.class.getName() + ACQUIRE_SEMAPHORE_SUFFIX, acquireSemaphore);
    Semaphore signal = new Semaphore(0);
    List<Throwable> throwables = new ArrayList<Throwable>();
    Fiber f = engine.createFiber();
    f.start(stepline, p, new CompletionCallback() {

        @Override
        public void onCompletion(Packet packet) {
            signal.release();
        }

        @Override
        public void onThrowable(Packet packet, Throwable throwable) {
            throwables.add(throwable);
            signal.release();
        }
    });
    releaseSemaphore.acquire();
    // Fiber is inside Step2
    f.cancel(false);
    acquireSemaphore.release();
    @SuppressWarnings({ "unchecked", "rawtypes" }) List<Step> called = (List) p.get(MARK);
    boolean result = signal.tryAcquire(1, TimeUnit.SECONDS);
    assertFalse(result);
    assertTrue(f.isCancelled());
    assertFalse(f.isDone());
    assertEquals(2, called.size());
    assertTrue(called.get(0) instanceof Step1);
    assertTrue(called.get(1) instanceof Step2);
    assertTrue(throwables.isEmpty());
}

Example 4 with CompletionCallback

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

the class StepTest method test.

@Test
public void test() throws InterruptedException {
    Step stepline = Step.createStepline(Arrays.asList(Step1.class, Step2.class, Step3.class));
    Packet p = new Packet();
    Semaphore signal = new Semaphore(0);
    List<Step> called = new ArrayList<Step>();
    List<Throwable> throwables = new ArrayList<Throwable>();
    engine.createFiber().start(stepline, p, new CompletionCallback() {

        @SuppressWarnings({ "unchecked", "rawtypes" })
        @Override
        public void onCompletion(Packet packet) {
            List<Step> l = (List) packet.get(MARK);
            if (l != null) {
                called.addAll(l);
            }
            signal.release();
        }

        @SuppressWarnings({ "unchecked", "rawtypes" })
        @Override
        public void onThrowable(Packet packet, Throwable throwable) {
            List<Step> l = (List) packet.get(MARK);
            if (l != null) {
                called.addAll(l);
            }
            throwables.add(throwable);
            signal.release();
        }
    });
    boolean result = signal.tryAcquire(5, TimeUnit.SECONDS);
    assertTrue(result);
    assertEquals(3, called.size());
    assertTrue(called.get(0) instanceof Step1);
    assertTrue(called.get(1) instanceof Step2);
    assertTrue(called.get(2) instanceof Step3);
    assertTrue(throwables.isEmpty());
}

Example 5 with CompletionCallback

use of oracle.kubernetes.operator.work.Fiber.CompletionCallback 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);
    }
}