/*
* 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.sling.cassandra.resource.provider.util;
import org.apache.commons.codec.binary.Base64;
import me.prettyprint.cassandra.model.CqlQuery;
import me.prettyprint.cassandra.model.CqlRows;
import me.prettyprint.cassandra.serializers.StringSerializer;
import me.prettyprint.hector.api.Keyspace;
import me.prettyprint.hector.api.beans.HColumn;
import me.prettyprint.hector.api.beans.Row;
import me.prettyprint.hector.api.query.QueryResult;
import org.apache.felix.scr.annotations.Reference;
import org.apache.sling.api.resource.ResourceResolver;
import org.apache.sling.cassandra.resource.provider.CassandraResourceProvider;
import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class CassandraResourceProviderUtil {
private static Logger LOGGER = LoggerFactory.getLogger(CassandraResourceProviderUtil.class);
public ResourceResolver getResourceResolver() {
return resourceResolver;
}
@Reference
private ResourceResolver resourceResolver;
public static QueryResult<CqlRows<String, String, String>> executeQuery(String query, Keyspace keyspace, StringSerializer se) {
// path, resourceType,resourceSuperType,metadata
CqlQuery<String, String, String> cqlQuery = new CqlQuery<String, String, String>(keyspace, se, se, se);
cqlQuery.setQuery(query);
return cqlQuery.execute();
}
public static String getRemainingPath(String path) {
if (path.startsWith("/") && path.split("/").length > 4) {
return path.substring(new StringBuilder("/").
append(path.split("/")[1]).
append("/").
append(path.split("/")[2]).
append("/").
append(path.split("/")[3]).
toString().length() + 1,
path.length());
} else {
return null;
}
}
public static String getColumnFamilySector(String path) {
if (path.startsWith("/") && path.split("/").length > 4) {
return path.split("/")[3];
} else {
return null;
}
}
public static void logResults(QueryResult<CqlRows<String, String, String>> result) {
LOGGER.info("############# RESULT ################");
for (Row<String, String, String> row : ((CqlRows<String, String, String>) result.get()).getList()) {
for (HColumn column : row.getColumnSlice().getColumns()) {
LOGGER.info(column.getValue().toString());
}
LOGGER.info("------------------------------------------------------------------------------------");
}
LOGGER.info("############# RESULT ################");
}
public static boolean recordExists(QueryResult<CqlRows<String, String, String>> result,String key) {
boolean exists = false;
for (Row<String, String, String> row : (result.get()).getList()) {
for (HColumn column : row.getColumnSlice().getColumns()) {
if(column.getName().equals(key) && column.getValue() != null) {
exists = true;
break;
}
}
}
return exists;
}
public static boolean isResourceExists(CassandraResourceProvider resourceProvider, Keyspace keyspace, String path) throws Exception {
String cql = resourceProvider.getCassandraMapperMap().get(CassandraResourceProviderUtil.getColumnFamilySector(path)).getCQL(CassandraResourceProviderUtil.getColumnFamilySector(path), CassandraResourceProviderUtil.getRemainingPath(path));
QueryResult<CqlRows<String, String, String>> result = CassandraResourceProviderUtil.executeQuery(cql, keyspace, new StringSerializer());
return recordExists(result,"policy");
}
public static String getrowID(String path) throws NoSuchAlgorithmException, UnsupportedEncodingException {
MessageDigest md = MessageDigest.getInstance("SHA1");
String rowID = new String(Base64.encodeBase64(md.digest(getRemainingPath(path).getBytes("UTF-8"))));
return rowID;
}
public static String getParentPath(String path) {
if(path.lastIndexOf("/") == 0){
return "/";
}
return path.substring(0,path.lastIndexOf("/"));
}
public static String getNameFromPath(String path){
return path.substring(path.lastIndexOf("/")+1);
}
public static QueryResult<CqlRows<String, String, String>> getAllNodes(Keyspace keyspace, String cf) throws Exception {
String cql = "select * from " + cf;
return CassandraResourceProviderUtil.executeQuery(cql, keyspace, new StringSerializer());
}
public static boolean isAnImmediateChild(String path,String s){
if(s.length() > path.length()) {
return !s.substring(path.length() + 1).contains("/");
}
return false;
}
// row.getColumnSlice().getColumnByName("metadata").getValue()
}