RedisClientFactory.java

/*
 * Copyright (c) 2017. Sunghyouk Bae <sunghyouk.bae@gmail.com>
 * 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 org.hibernate.cache.redis.client;

import lombok.NonNull;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.redisson.Redisson;
import org.redisson.api.RedissonClient;
import org.redisson.codec.SnappyCodec;
import org.redisson.config.Config;

import java.io.File;
import java.io.IOException;
import java.net.URL;

/**
 * Factory class for RedisClient.
 *
 * @author debop sunghyouk.bae@gmail.com
 */
@Slf4j
public final class RedisClientFactory {

  private RedisClientFactory() {}

  public static RedisClient createRedisClient(@NonNull final Config config) {
    try {
      if (config.getCodec() == null) {
        config.setCodec(new SnappyCodec());
      }
      log.debug("Set Redisson Codec = {}", config.getCodec().getClass().getName());
      RedissonClient redisson = Redisson.create(config);
      return new RedisClient(redisson);
    } catch (Exception e) {
      log.error("Fail to create RedisClient.", e);
      throw new RuntimeException(e);
    }
  }

  /**
   * Create {@link RedisClient} instance by properties
   *
   * @param redissonYamlUrl redisson yaml setting file URL
   * @return {@link RedisClient} instance
   */
  public static RedisClient createRedisClient(@NonNull final URL redissonYamlUrl) {
    try {
      Config config = Config.fromYAML(redissonYamlUrl);
      return createRedisClient(config);
    } catch (IOException e) {
      log.error("Error in create RedisClient. redissonYamlUrl={}", redissonYamlUrl, e);
      throw new RuntimeException(e);
    }
  }

  @SneakyThrows
  public static RedisClient createRedisClient(@NonNull final String redissonYamlPath) {
    log.trace("load redisson config yaml file. path={}", redissonYamlPath);

    String path = null;
    URL url = null;

    if (redissonYamlPath.startsWith("classpath:")) {
      path = redissonYamlPath.substring("classpath:".length());
      url = Thread.currentThread().getContextClassLoader().getResource(path);

      log.trace("load redisson yaml. path={}, url={}", path, url);
      return createRedisClient(url);
    }

    path = redissonYamlPath;
    if (redissonYamlPath.startsWith("file:")) {
      path = path.substring("file:".length());
    }
    url = new File(path).toURI().toURL();
    log.trace("load redisson yaml. path={}, url={}", path, url);
    return createRedisClient(url);

  }
}