CassandraEpRegistrationByEndpointIdDao.java

/**
 * Copyright 2014-2016 CyberVision, Inc.
 *
 * 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.kaaproject.kaa.server.common.nosql.cassandra.dao.filter;

import com.datastax.driver.core.Statement;
import com.datastax.driver.core.querybuilder.Clause;
import com.datastax.driver.core.querybuilder.QueryBuilder;

import org.kaaproject.kaa.server.common.nosql.cassandra.dao.AbstractCassandraDao;
import org.kaaproject.kaa.server.common.nosql.cassandra.dao.model.CassandraEpRegistrationByEndpointId;
import org.kaaproject.kaa.server.common.nosql.cassandra.dao.model.CassandraModelConstants;
import org.springframework.stereotype.Repository;

import java.util.Optional;

/**
 * @author Bohdan Khablenko
 *
 * @since v0.9.0
 */
@Repository
public class CassandraEpRegistrationByEndpointIdDao
    extends AbstractCassandraDao<CassandraEpRegistrationByEndpointId, String> {

  @Override
  protected Class<CassandraEpRegistrationByEndpointId> getColumnFamilyClass() {
    return CassandraEpRegistrationByEndpointId.class;
  }

  @Override
  public String getColumnFamilyName() {
    return CassandraModelConstants.EP_REGISTRATIONS_BY_ENDPOINT_ID_COLUMN_FAMILY_NAME;
  }

  /**
   * Found credential's id by endpoint id.
   * @param endpointId is endpoint id
   * @return string id or empty <code>Optional</code> if not found
   */
  public Optional<String> getCredentialsIdByEndpointId(String endpointId) {
    Clause clause = QueryBuilder.eq(
        CassandraModelConstants.EP_REGISTRATION_BY_ENDPOINT_ID_ENDPOINT_ID_PROPERTY,
        endpointId);
    Statement statement = QueryBuilder.select().from(this.getColumnFamilyName()).where(clause);
    CassandraEpRegistrationByEndpointId result = this.findOneByStatement(statement);
    if (result != null) {
      return Optional.of(result.getCredentialsId());
    } else {
      return Optional.empty();
    }
  }
}