// 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

// 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 com.cloud.utils.db;

import java.lang.reflect.Modifier;

import java.lang.reflect.ParameterizedType;

import java.lang.reflect.Type;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import javax.persistence.CollectionTable;

import javax.persistence.ElementCollection;

import javax.persistence.JoinColumn;

import com.cloud.utils.exception.CloudRuntimeException;

public class EcInfo {

protected String insertSql;

protected String selectSql;

protected String clearSql;

protected Class<?> targetClass;

protected Class<?> rawClass;

public EcInfo(Attribute attr, Attribute idAttr) {

attr.attache = this;

ElementCollection ec = attr.field.getAnnotation(ElementCollection.class);

targetClass = ec.targetClass();

Class<?> type = attr.field.getType();

if (type.isArray()) {

rawClass = null;

} else {

ParameterizedType pType = (ParameterizedType)attr.field.getGenericType();

Type rawType = pType.getRawType();

Class<?> rawClazz = (Class<?>)rawType;

try {

if (!Modifier.isAbstract(rawClazz.getModifiers()) && !rawClazz.isInterface() && rawClazz.getConstructors().length != 0 &&

rawClazz.getConstructor() != null) {

rawClass = rawClazz;

} else if (Set.class == rawClazz) {

rawClass = HashSet.class;

} else if (List.class == rawClazz) {

rawClass = ArrayList.class;

} else if (Collection.class == Collection.class) {

rawClass = ArrayList.class;

} else {

assert (false) : " We don't know how to create this calss " + rawType.toString() + " for " + attr.field.getName();

}

} catch (NoSuchMethodException e) {

throw new CloudRuntimeException("Write your own support for " + rawClazz + " defined by " + attr.field.getName());

}

}

CollectionTable ct = attr.field.getAnnotation(CollectionTable.class);

assert (ct.name().length() > 0) : "Please sepcify the table for " + attr.field.getName();

StringBuilder selectBuf = new StringBuilder("SELECT ");

StringBuilder insertBuf = new StringBuilder("INSERT INTO ");

StringBuilder clearBuf = new StringBuilder("DELETE FROM ");

clearBuf.append(ct.name()).append(" WHERE ");

selectBuf.append(attr.columnName);

selectBuf.append(" FROM ").append(ct.name()).append(", ").append(attr.table);

selectBuf.append(" WHERE ");

insertBuf.append(ct.name()).append("(");

StringBuilder valuesBuf = new StringBuilder("SELECT ");

for (JoinColumn jc : ct.joinColumns()) {

selectBuf.append(ct.name()).append(".").append(jc.name()).append("=");

if (jc.referencedColumnName().length() == 0) {

selectBuf.append(idAttr.table).append(".").append(idAttr.columnName);

valuesBuf.append(idAttr.table).append(".").append(idAttr.columnName);

clearBuf.append(ct.name()).append(".").append(jc.name()).append("=?");

} else {

selectBuf.append(attr.table).append(".").append(jc.referencedColumnName());

valuesBuf.append(attr.table).append(".").append(jc.referencedColumnName()).append(",");

}

selectBuf.append(" AND ");

insertBuf.append(jc.name()).append(", ");

valuesBuf.append(", ");

}

selectSql = selectBuf.append(idAttr.table).append(".").append(idAttr.columnName).append("=?").toString();

insertBuf.append(attr.columnName).append(") ");

valuesBuf.append("? FROM ").append(attr.table);

valuesBuf.append(" WHERE ").append(idAttr.table).append(".").append(idAttr.columnName).append("=?");

insertSql = insertBuf.append(valuesBuf).toString();

clearSql = clearBuf.toString();

}

}