/**
* 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.ambari.server.view;
import org.apache.ambari.server.controller.internal.AbstractResourceProvider;
import org.apache.ambari.server.controller.internal.ResourceImpl;
import org.apache.ambari.server.controller.spi.NoSuchParentResourceException;
import org.apache.ambari.server.controller.spi.NoSuchResourceException;
import org.apache.ambari.server.controller.spi.Predicate;
import org.apache.ambari.server.controller.spi.Request;
import org.apache.ambari.server.controller.spi.RequestStatus;
import org.apache.ambari.server.controller.spi.Resource;
import org.apache.ambari.server.controller.spi.ResourceAlreadyExistsException;
import org.apache.ambari.server.controller.spi.SystemException;
import org.apache.ambari.server.controller.spi.UnsupportedPropertyException;
import org.apache.ambari.server.orm.entities.ViewEntity;
import org.apache.ambari.server.orm.entities.ViewInstanceEntity;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
/**
* External view sub-resource provider.
*/
public class ViewExternalSubResourceProvider extends AbstractResourceProvider {
/**
* View external sub resource property id constants.
*/
private static final String VIEW_NAME_PROPERTY_ID = "view_name";
private static final String INSTANCE_NAME_PROPERTY_ID = "instance_name";
private static final String RESOURCE_NAME_PROPERTY_ID = "name";
/**
* The resource type.
*/
private final Resource.Type type;
/**
* The names of the external resources for the view.
*/
private final Set<String> resourceNames = new HashSet<String>();
/**
* The set of key property ids.
*/
private final Set<String> pkPropertyIds;
/**
* The associated view definition.
*/
private final ViewEntity viewDefinition;
// ----- Constructors ------------------------------------------------------
/**
* Construct a view sub-resource provider.
*
* @param type the resource type
* @param viewDefinition the associated view definition
*/
public ViewExternalSubResourceProvider(Resource.Type type, ViewEntity viewDefinition) {
super(_getPropertyIds(), _getKeyPropertyIds(type));
this.type = type;
this.pkPropertyIds = new HashSet<String>(getKeyPropertyIds().values());
this.viewDefinition = viewDefinition;
}
@Override
public RequestStatus createResources(Request request)
throws SystemException, UnsupportedPropertyException,
ResourceAlreadyExistsException, NoSuchParentResourceException {
throw new UnsupportedOperationException("Not supported!");
}
@Override
public Set<Resource> getResources(Request request, Predicate predicate)
throws SystemException, UnsupportedPropertyException,
NoSuchResourceException, NoSuchParentResourceException {
Set<Resource> resourceSet = new HashSet<Resource>();
Set<ViewInstanceEntity> instanceDefinitions = new HashSet<ViewInstanceEntity>();
Set<Map<String, Object>> propertyMaps = getPropertyMaps(predicate);
int size = propertyMaps.size();
Collection<ViewInstanceEntity> viewInstanceDefinitions = viewDefinition.getInstances();
if (size == 0) {
instanceDefinitions.addAll(viewInstanceDefinitions);
} else {
for (Map<String, Object> propertyMap : propertyMaps) {
String instanceName = (String) propertyMap.get(INSTANCE_NAME_PROPERTY_ID);
if (instanceName == null) {
instanceDefinitions.addAll(viewInstanceDefinitions);
break;
} else {
instanceDefinitions.add(viewDefinition.getInstanceDefinition(instanceName));
}
}
}
for (ViewInstanceEntity viewInstanceDefinition : instanceDefinitions) {
for (String resourceName : resourceNames) {
ResourceImpl resource = new ResourceImpl(type);
resource.setProperty(VIEW_NAME_PROPERTY_ID, viewDefinition.getName());
resource.setProperty(INSTANCE_NAME_PROPERTY_ID, viewInstanceDefinition.getName());
resource.setProperty("name", resourceName);
resourceSet.add(resource);
}
}
return resourceSet;
}
@Override
public RequestStatus updateResources(Request request, Predicate predicate)
throws SystemException, UnsupportedPropertyException,
NoSuchResourceException, NoSuchParentResourceException {
throw new UnsupportedOperationException("Not supported!");
}
@Override
public RequestStatus deleteResources(Predicate predicate)
throws SystemException, UnsupportedPropertyException,
NoSuchResourceException, NoSuchParentResourceException {
throw new UnsupportedOperationException("Not supported!");
}
@Override
protected Set<String> getPKPropertyIds() {
return pkPropertyIds;
}
@Override
public Set<String> checkPropertyIds(Set<String> propertyIds) {
return Collections.emptySet();
}
// ----- helper methods ----------------------------------------------------
/**
* Register an external sub-resource name.
*
* @param resourceName the resource name
*/
public void addResourceName(String resourceName) {
resourceNames.add(resourceName);
}
// get the key property ids for the resource
private static Map<Resource.Type, String> _getKeyPropertyIds(Resource.Type type) {
Map<Resource.Type, String> keyPropertyIds = new HashMap<Resource.Type, String>();
keyPropertyIds.put(Resource.Type.View, VIEW_NAME_PROPERTY_ID);
keyPropertyIds.put(Resource.Type.ViewInstance, INSTANCE_NAME_PROPERTY_ID);
keyPropertyIds.put(type, RESOURCE_NAME_PROPERTY_ID);
return keyPropertyIds;
}
// get the property ids for the resource
private static Set<String> _getPropertyIds() {
Set<String> propertyIds = new HashSet<String>();
propertyIds.add(INSTANCE_NAME_PROPERTY_ID);
propertyIds.add(VIEW_NAME_PROPERTY_ID);
propertyIds.add(RESOURCE_NAME_PROPERTY_ID);
return propertyIds;
}
}