org.trustedanalytics.servicebroker.hive.plans.HiveShared.java Source code

Introduction

Here is the source code for org.trustedanalytics.servicebroker.hive.plans.HiveShared.java

Source

/**
 * Copyright (c) 2015 Intel Corporation
 *
 * 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.trustedanalytics.servicebroker.hive.plans;

import org.cloudfoundry.community.servicebroker.exception.ServiceBrokerException;
import org.cloudfoundry.community.servicebroker.exception.ServiceInstanceExistsException;
import org.cloudfoundry.community.servicebroker.model.ServiceInstance;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcOperations;
import org.springframework.stereotype.Component;
import org.trustedanalytics.servicebroker.framework.service.ServicePlanDefinition;
import org.trustedanalytics.servicebroker.hive.DbNameNormalizer;
import org.trustedanalytics.servicebroker.hive.plans.binding.HiveBindingClient;

import java.util.Map;
import java.util.UUID;

@Component("shared")
class HiveShared implements ServicePlanDefinition {

    private HiveBindingClient bindingOperations;
    private JdbcOperations operations;

    @Autowired
    public HiveShared(HiveBindingClient bindingOperations, JdbcOperations operations) {
        this.bindingOperations = bindingOperations;
        this.operations = operations;
    }

    @Override
    public Map<String, Object> bind(ServiceInstance serviceInstance) throws ServiceBrokerException {
        UUID serviceId = UUID.fromString(serviceInstance.getServiceInstanceId());
        return bindingOperations.createCredentialsMap(serviceId);
    }

    @Override
    public void provision(ServiceInstance serviceInstance)
            throws ServiceInstanceExistsException, ServiceBrokerException {
        operations.execute(String.format("create database if not exists `%s`",
                DbNameNormalizer.create().normalize(serviceInstance.getServiceInstanceId())));

        if (bindingOperations.isKerberosEnabled()) {
            operations.execute(String.format("GRANT ALL ON DATABASE `%s` TO ROLE `%s`",
                    DbNameNormalizer.create().normalize(serviceInstance.getServiceInstanceId()),
                    serviceInstance.getOrganizationGuid()));
        }
    }
}