io.gravitee.management.services.dictionary.DictionaryRefresher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gravitee-management-api-services-dictionary Show documentation
There is a newer version: 1.30.31
Show newest version
/**
 * Copyright (C) 2015 The Gravitee team (http://gravitee.io)
 *
 * 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.gravitee.management.services.dictionary;

import io.gravitee.definition.model.Property;
import io.gravitee.management.model.configuration.dictionary.DictionaryEntity;
import io.gravitee.management.model.configuration.dictionary.UpdateDictionaryEntity;
import io.gravitee.management.services.dictionary.model.DynamicProperty;
import io.gravitee.management.services.dictionary.provider.Provider;
import io.vertx.core.Handler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Collection;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author David BRASSELY (david.brassely at graviteesource.com)
 * @author GraviteeSource Team
 */
public class DictionaryRefresher implements Handler {

    private final Logger logger = LoggerFactory.getLogger(DictionaryRefresher.class);

    private DictionaryEntity dictionary;
    private Provider provider;
    private io.gravitee.management.service.configuration.dictionary.DictionaryService dictionaryService;

    public DictionaryRefresher(final DictionaryEntity dictionary) {
        this.dictionary = dictionary;
    }

    @Override
    public void handle(Long event) {
        logger.debug("Running dictionary refresher task for {}", dictionary);

        provider.get()
                .whenComplete((dynamicProperties, throwable) -> {
                    if (throwable != null) {
                        logger.error("[{}] Unexpected error while getting dictionary's properties from provider: {}",
                                dictionary.getId(), provider.name(),
                                throwable);
                    } else if (dynamicProperties != null) {
                        updateDictionary(dynamicProperties);
                    }
                });
    }

    private void updateDictionary(Collection dynProperties) {
        Map properties = dynProperties
                .stream()
                .collect(Collectors.toMap(
                        Property::getKey,
                        dynamicProperty -> (dynamicProperty.getValue() == null) ? "" : dynamicProperty.getValue()));

        // Compare properties with latest values
        if (! properties.equals(dictionary.getProperties())) {
            dictionary.setProperties(properties);

            try {
                dictionary = dictionaryService.update(dictionary.getId(), convert(dictionary));
                dictionaryService.deploy(dictionary.getId());
            } catch (Exception ex) {
                logger.error("Unexpected error while updating and deploying the dictionary", ex);
            }
        }
    }

    private UpdateDictionaryEntity convert(DictionaryEntity dictionaryEntity) {
        UpdateDictionaryEntity dictionary = new UpdateDictionaryEntity();

        dictionary.setName(dictionaryEntity.getName());
        dictionary.setDescription(dictionaryEntity.getDescription());
        dictionary.setType(dictionaryEntity.getType());
        dictionary.setProperties(dictionaryEntity.getProperties());
        dictionary.setProvider(dictionaryEntity.getProvider());
        dictionary.setTrigger(dictionaryEntity.getTrigger());

        return dictionary;
    }

    public void setDictionaryService(io.gravitee.management.service.configuration.dictionary.DictionaryService dictionaryService) {
        this.dictionaryService = dictionaryService;
    }

    public void setProvider(Provider provider) {
        this.provider = provider;
    }
}