io.fency.redis.RedisMessageService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fency-redis Show documentation
Idempotency barrier for RabbitMQ consumers.
There is a newer version: 0.1.2
/*
 * Copyright 2016-2019 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.fency.redis;

import java.util.Optional;
import java.util.Set;
import java.util.StringJoiner;

import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.transaction.annotation.Transactional;

import io.fency.Message;
import io.fency.MessageService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;

/**
 * Redis implementation of {@link MessageService}.
 * @author Gilles Robert
 */
@Transactional
@RequiredArgsConstructor
@Slf4j
public class RedisMessageService implements MessageService {

  private static final String PREFIX = "message";
  private final RedisTemplate redisTemplate;

  /**
   * Save a Message metadata in a data store.
   * @param message The message to save.
   */
  @Override
  public void save(Message message) {
        log.debug("Saving metadata for message with id {} and consumer queue name {}", message.getId(),
            message.getConsumerQueueName());
    String key = getKey(message.getId(), message.getConsumerQueueName());
    redisTemplate.opsForValue().set(key, message);
  }

  /**
   * Retrieve a Message metadata in a data store.
   * @param messageId The message id.
   * @param consumerQueueName The target consumer queue name.
   */
  @Override
  public Optional find(String messageId, String consumerQueueName) {
    log.debug("Retrieving metadata for message with id {} and consumer queue name {}", messageId, consumerQueueName);
    String key = getKey(messageId, consumerQueueName);
    return Optional.ofNullable(redisTemplate.opsForValue().get(key));
  }

  /**
   * Clean the message metadata in a data store.
   */
  @Override
  public void clean() {
    log.debug("Cleaning up messages metadata stored in Redis...");
    Set keys = redisTemplate.keys(PREFIX + "*");
    redisTemplate.delete(keys);
    log.debug("{} messages deleted.", keys.size());
  }

  private String getKey(String messageId, String consumerQueueName) {
    StringJoiner stringJoiner = new StringJoiner(":");
    stringJoiner.add(PREFIX);
    stringJoiner.add(messageId);
    stringJoiner.add(consumerQueueName);
    return stringJoiner.toString();
  }
}