org.apache.shenyu.sync.data.zookeeper.ZookeeperClient Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.shenyu.sync.data.zookeeper;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.framework.recipes.cache.ChildData;
import org.apache.curator.framework.recipes.cache.TreeCache;
import org.apache.curator.framework.recipes.cache.TreeCacheListener;
import org.apache.curator.retry.ExponentialBackoffRetry;
import org.apache.curator.utils.CloseableUtils;
import org.apache.shenyu.common.exception.ShenyuException;
import org.apache.shenyu.common.utils.GsonUtils;
import org.apache.zookeeper.CreateMode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.nio.charset.StandardCharsets;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
public class ZookeeperClient implements AutoCloseable {
private static final Logger LOGGER = LoggerFactory.getLogger(ZookeeperClient.class);
private final ZookeeperConfig config;
private final CuratorFramework client;
private final Map caches = new ConcurrentHashMap<>();
public ZookeeperClient(final ZookeeperConfig zookeeperConfig) {
this.config = zookeeperConfig;
ExponentialBackoffRetry retryPolicy = new ExponentialBackoffRetry(config.getBaseSleepTimeMilliseconds(), config.getMaxRetries(), config.getMaxSleepTimeMilliseconds());
CuratorFrameworkFactory.Builder builder = CuratorFrameworkFactory.builder()
.connectString(config.getServerLists())
.retryPolicy(retryPolicy)
.connectionTimeoutMs(config.getConnectionTimeoutMilliseconds())
.sessionTimeoutMs(config.getSessionTimeoutMilliseconds())
.namespace(config.getNamespace());
if (!StringUtils.isEmpty(config.getDigest())) {
builder.authorization("digest", config.getDigest().getBytes(StandardCharsets.UTF_8));
}
this.client = builder.build();
}
/**
* start.
*/
public void start() {
this.client.start();
try {
this.client.blockUntilConnected();
} catch (InterruptedException e) {
LOGGER.warn("Interrupted during zookeeper client starting.");
Thread.currentThread().interrupt();
}
}
/**
* start.
*/
@Override
public void close() {
// close all caches
for (Map.Entry cache : caches.entrySet()) {
CloseableUtils.closeQuietly(cache.getValue());
}
// close client
CloseableUtils.closeQuietly(client);
}
/**
* get curator framework.
*
* @return curator framework client.
*/
public CuratorFramework getClient() {
return client;
}
/**
* check if key exist.
*
* @param key zookeeper path
* @return if exist.
*/
public boolean isExist(final String key) {
try {
return null != client.checkExists().forPath(key);
} catch (Exception e) {
throw new ShenyuException(e);
}
}
/**
* get from zk directly.
*
* @param key zookeeper path
* @return value.
*/
public String getDirectly(final String key) {
try {
byte[] ret = client.getData().forPath(key);
return Objects.isNull(ret) ? null : new String(ret, StandardCharsets.UTF_8);
} catch (Exception e) {
throw new ShenyuException(e);
}
}
/**
* get value for specific key.
*
* @param key zookeeper path
* @return value.
*/
public String get(final String key) {
TreeCache cache = findFromcache(key);
if (Objects.isNull(cache)) {
return getDirectly(key);
}
ChildData data = cache.getCurrentData(key);
if (Objects.isNull(data)) {
return getDirectly(key);
}
return Objects.isNull(data.getData()) ? null : new String(data.getData(), StandardCharsets.UTF_8);
}
/**
* create or update key with value.
*
* @param key zookeeper path key.
* @param value string value.
* @param mode creation mode.
*/
public void createOrUpdate(final String key, final String value, final CreateMode mode) {
String val = StringUtils.isEmpty(value) ? "" : value;
try {
client.create().orSetData().creatingParentsIfNeeded().withMode(mode).forPath(key, val.getBytes(StandardCharsets.UTF_8));
} catch (Exception e) {
throw new ShenyuException(e);
}
}
/**
* create or update key with value.
*
* @param key zookeeper path key.
* @param value object value.
* @param mode creation mode.
*/
public void createOrUpdate(final String key, final Object value, final CreateMode mode) {
if (value != null) {
String val = GsonUtils.getInstance().toJson(value);
createOrUpdate(key, val, mode);
} else {
createOrUpdate(key, "", mode);
}
}
/**
* delete a node with specific key.
*
* @param key zookeeper path key.
*/
public void delete(final String key) {
try {
client.delete().guaranteed().deletingChildrenIfNeeded().forPath(key);
} catch (Exception e) {
throw new ShenyuException(e);
}
}
/**
* get children with specific key.
*
* @param key zookeeper key.
* @return children node name.
*/
public List getChildren(final String key) {
try {
return client.getChildren().forPath(key);
} catch (Exception e) {
LOGGER.error("zookeeper get child error=", e);
return Collections.emptyList();
}
}
/**
* get created cache.
* @param path path.
* @return cache.
*/
public TreeCache getCache(final String path) {
return caches.get(path);
}
/**
* add new curator cache.
* @param path path.
* @param listeners listeners.
* @return cache.
*/
public TreeCache addCache(final String path, final TreeCacheListener... listeners) {
TreeCache cache = TreeCache.newBuilder(client, path).build();
caches.put(path, cache);
if (ArrayUtils.isNotEmpty(listeners)) {
for (TreeCacheListener listener : listeners) {
cache.getListenable().addListener(listener);
}
}
try {
cache.start();
} catch (Exception e) {
throw new ShenyuException("failed to add curator cache.", e);
}
return cache;
}
/**
* find cache with key.
* @param key key.
* @return cache.
*/
private TreeCache findFromcache(final String key) {
for (Map.Entry cache : caches.entrySet()) {
if (key.startsWith(cache.getKey())) {
return cache.getValue();
}
}
return null;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy