Examples with RabbitMQMessage com.redhat.jenkins.plugins.ci.messaging.data.RabbitMQMessage used on opensource projects

Example 1 with RabbitMQMessage

use of com.redhat.jenkins.plugins.ci.messaging.data.RabbitMQMessage in project jms-messaging-plugin by jenkinsci.

the class RabbitMQMessagingWorker method sendMessage.

@Override
public SendResult sendMessage(Run<?, ?> build, TaskListener listener, ProviderData pdata) {
    RabbitMQPublisherProviderData pd = (RabbitMQPublisherProviderData) pdata;
    try {
        if (connection == null || !connection.isOpen()) {
            connect();
        }
        if (channel == null || !channel.isOpen()) {
            this.channel = connection.createChannel();
            log.info("Channel created.");
        }
    } catch (Exception ex) {
        ex.printStackTrace();
    }
    String body = "";
    String msgId = "";
    // Fedora messaging wire format support
    Map<String, Object> headers = new HashMap<>();
    if (pd.isFedoraMessaging()) {
        headers.put("fedora_messaging_severity", pd.getSeverity());
        headers.put("fedora_messaging_schema", pd.getSchema());
        headers.put("sent_at", ZonedDateTime.now().toString());
    }
    System.out.println(headers);
    try {
        EnvVars env = new EnvVars();
        env.putAll(build.getEnvironment(listener));
        env.put("CI_NAME", build.getParent().getName());
        if (!build.isBuilding()) {
            env.put("CI_STATUS", (build.getResult() == Result.SUCCESS ? "passed" : "failed"));
            env.put("BUILD_STATUS", build.getResult().toString());
        }
        RabbitMQMessage msg = new RabbitMQMessage(PluginUtils.getSubstitutedValue(getTopic(provider), build.getEnvironment(listener)), PluginUtils.getSubstitutedValue(pd.getMessageContent(), env));
        msg.setTimestamp(System.currentTimeMillis() / 1000L);
        body = msg.getBodyJson();
        msgId = msg.getMsgId();
        try {
            channel.exchangeDeclarePassive(exchangeName);
            channel.basicPublish(exchangeName, msg.getTopic(), new AMQP.BasicProperties.Builder().headers(headers).messageId(msgId).build(), body.getBytes(StandardCharsets.UTF_8));
        } catch (IOException e) {
            if (pd.isFailOnError()) {
                log.severe("Unhandled exception in perform: Failed to send message!");
                return new SendResult(false, msgId, body);
            }
        }
        log.fine("Message headers:\n" + headers);
        log.fine("JSON message:\n" + msg.toJson());
        listener.getLogger().println("Message id: " + msg.getMsgId());
        listener.getLogger().println("Message topic: " + msg.getTopic());
        listener.getLogger().println("Message headers:\n" + headers);
        listener.getLogger().println("JSON message body:\n" + body);
    } catch (Exception e) {
        if (pd.isFailOnError()) {
            log.severe("Unhandled exception in perform: ");
            log.severe(ExceptionUtils.getStackTrace(e));
            listener.fatalError("Unhandled exception in perform: ");
            listener.fatalError(ExceptionUtils.getStackTrace(e));
            return new SendResult(false, msgId, body);
        } else {
            log.warning("Unhandled exception in perform: ");
            log.warning(ExceptionUtils.getStackTrace(e));
            listener.error("Unhandled exception in perform: ");
            listener.error(ExceptionUtils.getStackTrace(e));
            return new SendResult(true, msgId, body);
        }
    } finally {
        try {
            channel.close();
        } catch (Exception e) {
            log.warning("Unhandled exception when closing channel: ");
            log.warning(ExceptionUtils.getStackTrace(e));
            listener.getLogger().println("exception in finally");
        }
    }
    return new SendResult(true, msgId, body);
}

Example 2 with RabbitMQMessage

use of com.redhat.jenkins.plugins.ci.messaging.data.RabbitMQMessage in project jms-messaging-plugin by jenkinsci.

the class RabbitMQMessagingWorker method receive.

@Override
public void receive(String jobname, ProviderData pdata) {
    RabbitMQSubscriberProviderData pd = (RabbitMQSubscriberProviderData) pdata;
    int timeout = (pd.getTimeout() != null ? pd.getTimeout() : RabbitMQSubscriberProviderData.DEFAULT_TIMEOUT_IN_MINUTES) * 60 * 1000;
    if (interrupt) {
        log.info("we have been interrupted at start of receive");
        return;
    }
    // subscribe job
    while (!subscribe(jobname)) {
        if (!Thread.currentThread().isInterrupted()) {
            try {
                int WAIT_SECONDS = 2;
                Thread.sleep(WAIT_SECONDS * 1000);
            } catch (InterruptedException e) {
                // We were interrupted while waiting to retry. We will
                // jump ship on the next iteration.
                // NB: The interrupt flag was cleared when
                // InterruptedException was thrown. We have to
                // re-install it to make sure we eventually leave this
                // thread.
                Thread.currentThread().interrupt();
            }
        }
    }
    long lastSeenMessage = new Date().getTime();
    try {
        while ((new Date().getTime() - lastSeenMessage) < timeout) {
            if (!messageQueue.isEmpty()) {
                RabbitMQMessage data = messageQueue.poll();
                // Reset timer
                lastSeenMessage = data.getTimestamp().getTime();
                // 
                if (provider.verify(data.getBodyJson(), pd.getChecks(), jobname)) {
                    Map<String, String> params = new HashMap<>();
                    params.put("CI_MESSAGE", data.getBodyJson());
                    trigger(jobname, data.getBodyJson(), params);
                }
                channel.basicAck(data.getDeliveryTag(), false);
            } else {
                if (interrupt) {
                    log.info("We have been interrupted...");
                    break;
                }
            }
            TimeUnit.MILLISECONDS.sleep(500);
        }
    } catch (Exception e) {
        // for more info
        if (e.getClass() == InterruptedException.class) {
            Thread.currentThread().interrupt();
        }
        if (!Thread.currentThread().isInterrupted()) {
            // Something other than an interrupt causes this.
            // Unsubscribe, but stay in our loop and try to reconnect..
            log.log(Level.WARNING, "JMS exception raised, going to re-subscribe for job '" + jobname + "'.", e);
            // Try again next time.
            unsubscribe(jobname);
        }
    }
}

Example 3 with RabbitMQMessage

use of com.redhat.jenkins.plugins.ci.messaging.data.RabbitMQMessage in project jms-messaging-plugin by jenkinsci.

the class RabbitMQMessagingWorker method waitForMessage.

@Override
public String waitForMessage(Run<?, ?> build, TaskListener listener, ProviderData pdata) {
    RabbitMQSubscriberProviderData pd = (RabbitMQSubscriberProviderData) pdata;
    try {
        if (connection == null || !connection.isOpen()) {
            connect();
        }
        if (channel == null || !channel.isOpen()) {
            this.channel = connection.createChannel();
        }
        channel.exchangeDeclarePassive(exchangeName);
        channel.queueBind(getQueue(provider), exchangeName, this.topic);
    } catch (Exception ex) {
        log.severe("Connection to broker can't be established!");
        log.severe(ExceptionUtils.getStackTrace(ex));
        listener.error("Connection to broker can't be established!");
        listener.error(ExceptionUtils.getStackTrace(ex));
        return null;
    }
    log.info("Waiting for message.");
    listener.getLogger().println("Waiting for message.");
    for (MsgCheck msgCheck : pd.getChecks()) {
        log.info(" with check: " + msgCheck.toString());
        listener.getLogger().println(" with check: " + msgCheck);
    }
    Integer timeout = (pd.getTimeout() != null ? pd.getTimeout() : RabbitMQSubscriberProviderData.DEFAULT_TIMEOUT_IN_MINUTES);
    log.info(" with timeout: " + timeout + " minutes");
    listener.getLogger().println(" with timeout: " + timeout + " minutes");
    // Create deliver callback to listen for messages
    DeliverCallback deliverCallback = (consumerTag, delivery) -> {
        String json = new String(delivery.getBody(), StandardCharsets.UTF_8);
        listener.getLogger().println("Received '" + delivery.getEnvelope().getRoutingKey() + "':\n" + "Message id: '" + delivery.getProperties().getMessageId() + "'\n'" + json + "'");
        log.info("Received '" + delivery.getEnvelope().getRoutingKey() + "':\n" + "Message id: '" + delivery.getProperties().getMessageId() + "'\n'" + json + "'");
        RabbitMQMessage message = new RabbitMQMessage(delivery.getEnvelope().getRoutingKey(), json, delivery.getProperties().getMessageId());
        message.setTimestamp(new Date().getTime());
        message.setDeliveryTag(delivery.getEnvelope().getDeliveryTag());
        messageQueue.add(message);
    };
    String consumerTag = null;
    long startTime = new Date().getTime();
    int timeoutInMs = timeout * 60 * 1000;
    try {
        consumerTag = channel.basicConsume(getQueue(provider), deliverCallback, (CancelCallback) null);
        while ((new Date().getTime() - startTime) < timeoutInMs) {
            if (!messageQueue.isEmpty()) {
                RabbitMQMessage message = messageQueue.poll();
                log.info("Obtained message from queue: " + message.toJson());
                if (!provider.verify(message.getBodyJson(), pd.getChecks(), jobname)) {
                    channel.basicAck(message.getDeliveryTag(), false);
                    continue;
                }
                listener.getLogger().println("Message: '" + message.getMsgId() + "' was succesfully checked.");
                if (build != null) {
                    if (StringUtils.isNotEmpty(pd.getVariable())) {
                        EnvVars vars = new EnvVars();
                        vars.put(pd.getVariable(), message.getBodyJson());
                        build.addAction(new CIEnvironmentContributingAction(vars));
                    }
                }
                channel.basicAck(message.getDeliveryTag(), false);
                return message.getBodyJson();
            }
            if (interrupt) {
                return null;
            }
            TimeUnit.MILLISECONDS.sleep(500);
        }
        log.severe("Timed out waiting for message!");
        listener.getLogger().println("Timed out waiting for message!");
    } catch (Exception e) {
        // for more info
        if (e.getClass() == InterruptedException.class) {
            Thread.currentThread().interrupt();
        }
        log.log(Level.SEVERE, "Unhandled exception waiting for message.", e);
    } finally {
        try {
            if (consumerTag != null) {
                channel.basicCancel(consumerTag);
            }
            channel.close();
        } catch (Exception e) {
            listener.getLogger().println("exception in finally");
        }
    }
    return null;
}

Example 4 with RabbitMQMessage

use of com.redhat.jenkins.plugins.ci.messaging.data.RabbitMQMessage in project jms-messaging-plugin by jenkinsci.

the class RabbitMQMessagingWorker method subscribe.

@Override
public boolean subscribe(String jobname, String selector) {
    if (interrupt) {
        return true;
    }
    if (this.topic != null) {
        while (!Thread.currentThread().isInterrupted()) {
            try {
                if (connection == null || !connection.isOpen()) {
                    if (!connect()) {
                        return false;
                    }
                }
                if (channel == null || !channel.isOpen()) {
                    this.channel = connection.createChannel();
                    log.info("Subscribing job '" + jobname + "' to " + this.topic + " topic.");
                    String queueName = getQueue(provider);
                    try {
                        // Check if queue exists
                        channel.queueDeclarePassive(queueName);
                    } catch (IOException e) {
                        // Request new queue - durable false, exclusive true, autodelete true
                        this.channel = connection.createChannel();
                        channel.queueDeclare(queueName, false, true, true, null);
                    }
                    channel.exchangeDeclarePassive(exchangeName);
                    channel.queueBind(queueName, exchangeName, this.topic);
                    // Create deliver callback to listen for messages
                    DeliverCallback deliverCallback = (consumerTag, delivery) -> {
                        String json = new String(delivery.getBody(), StandardCharsets.UTF_8);
                        log.info("Received '" + delivery.getEnvelope().getRoutingKey() + "':\n" + "Message id: '" + delivery.getProperties().getMessageId() + "'\n'" + json + "'");
                        RabbitMQMessage message = new RabbitMQMessage(delivery.getEnvelope().getRoutingKey(), json, delivery.getProperties().getMessageId());
                        message.setTimestamp(new Date().getTime());
                        message.setDeliveryTag(delivery.getEnvelope().getDeliveryTag());
                        messageQueue.add(message);
                    };
                    this.consumerTag = channel.basicConsume(queueName, deliverCallback, (CancelCallback) null);
                    log.info("Successfully subscribed job '" + jobname + "' to topic '" + this.topic + "'.");
                } else {
                    log.info("Already subscribed job '" + jobname + "' to topic '" + this.topic + "'.");
                }
                return true;
            } catch (Exception ex) {
                // Either we were interrupted, or something else went
                // wrong. If we were interrupted, then we will jump ship
                // on the next iteration. If something else happened,
                // then we just unsubscribe here, sleep, so that we may
                // try again on the next iteration.
                log.log(Level.SEVERE, "Eexception raised while subscribing job '" + jobname + "', retrying in " + RETRY_MINUTES + " minutes.", ex);
                if (!Thread.currentThread().isInterrupted()) {
                    unsubscribe(jobname);
                    try {
                        Thread.sleep(RETRY_MINUTES * 60 * 1000);
                    } catch (InterruptedException ie) {
                        // We were interrupted while waiting to retry.
                        // We will jump ship on the next iteration.
                        // NB: The interrupt flag was cleared when
                        // InterruptedException was thrown. We have to
                        // re-install it to make sure we eventually
                        // leave this thread.
                        Thread.currentThread().interrupt();
                    }
                }
            }
        }
    }
    return false;
}