com.hp.autonomy.searchcomponents.hod.databases.DatabasesServiceImpl Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 Hewlett-Packard Development Company, L.P.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 */

package com.hp.autonomy.searchcomponents.hod.databases;

import com.hp.autonomy.hod.client.api.authentication.TokenType;
import com.hp.autonomy.hod.client.api.resource.*;
import com.hp.autonomy.hod.client.error.HodErrorException;
import com.hp.autonomy.hod.client.token.TokenProxy;
import lombok.Data;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

@Data
@Service
public class DatabasesServiceImpl implements DatabasesService {
    private static final Set CONTENT_FLAVOURS = ResourceFlavour.of(ResourceFlavour.EXPLORER, ResourceFlavour.STANDARD, ResourceFlavour.CUSTOM_FIELDS);

    private final ResourcesService resourcesService;
    private final ResourceMapper resourceMapper;

    @Autowired
    public DatabasesServiceImpl(final ResourcesService resourcesService, final ResourceMapper resourceMapper) {
        this.resourcesService = resourcesService;
        this.resourceMapper = resourceMapper;
    }

    @Override
    public Set getDatabases(final String domain) throws HodErrorException {
        return getDatabases(null, domain);
    }

    @Override
    public Set getDatabases(final TokenProxy tokenProxy, final String domain) throws HodErrorException {
        final ListResourcesRequestBuilder builder = new ListResourcesRequestBuilder()
            .setTypes(Collections.singleton(ResourceType.CONTENT));

        final Resources resources = tokenProxy == null ? resourcesService.list(builder) : resourcesService.list(tokenProxy, builder);

        final Set databases = new HashSet<>();

        // Private and public indexes can have the same name. You can't do anything with the public index in this case,
        // so we remove the public index duplicates here.
        final Collection privateResourceNames = new HashSet<>();

        final Set privateResources = new HashSet<>();

        for (final Resource resource : resources.getResources()) {
            if (CONTENT_FLAVOURS.contains(resource.getFlavour())) {
                privateResources.add(resource);
                privateResourceNames.add(resource.getResource());
            }
        }

        databases.addAll(resourceMapper.map(tokenProxy, privateResources, domain));

        final Set publicResources = new HashSet<>();

        for (final Resource resource : resources.getPublicResources()) {
            if (!privateResourceNames.contains(resource.getResource())) {
                publicResources.add(resource);
            }
        }

        databases.addAll(resourceMapper.map(tokenProxy, publicResources, ResourceIdentifier.PUBLIC_INDEXES_DOMAIN));

        return databases;
    }
}