The following code snippet shows an example on how to implement the call method for MySQL_Cloud Connector:
@Override
public int call(DataSession dataSession, CallAttributes callAttr) throws SDKException{
InfaUtils pInfaUtils = dataSession.getInfaUtilsHandle();
logger = pInfaUtils.getLogger();
MySQL_CloudTableDataConnection conn = (MySQL_CloudTableDataConnection) dataSession.getConnection();
Connection nativeConn = (Connection) conn.getNativeConnection();
int paramListSize = pr.getFieldList().size();
String procQuery = "{CALL " + procName + "(";
for (int i=0 ; i < paramListSize; i++) {
if(i+1==paramListSize) {
procQuery=procQuery+ "?";
break;
}
procQuery=procQuery+ "?,";
}
procQuery = procQuery + ")}";
CallableStatement callStmt = null;
int col = 0;
int rowSize = callAttr.getNumInputRows();
List<List<Object>> result = new ArrayList<List<Object>>();
String stringValue = null;
BigDecimal bdc=null;
Timestamp ts=null;
Date dt=null;
Time time=null;
double doubleValue;
float floatValue;
long longValue;
int intValue;
try {
for(int i=0; i < rowSize; i++) {
col=0;
callStmt = nativeConn.prepareCall(procQuery);
for (com.informatica.sdk.adapter.metadata.field.semantic.iface.FieldBase fieldBase : pr
.getFieldList()) {
Parameter param = (Parameter) fieldBase;
String dataType = param.getDataType();
DataAttributes dataAttr = new DataAttributes();
switch (dataType) {
case "TINYINT" :
case "SMALLINT" :
case "MEDIUMINT":
case "INT":
if(param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
stringValue = dataSession.getStringData(dataAttr);
if(stringValue==null)
{
callStmt.setNull(col + 1,Types.INTEGER);
}
else {
intValue = Integer.parseInt(stringValue);
callStmt.setInt(col + 1, intValue);
}
}
else {
if(dataType.equalsIgnoreCase("TINYINT")) {
callStmt.registerOutParameter(col+1, Types.TINYINT);
}
else if(dataType.equalsIgnoreCase("SMALLINT")) {
callStmt.registerOutParameter(col+1, Types.SMALLINT);
}
else {
callStmt.registerOutParameter(col+1, Types.INTEGER);
}
}
break;
case "BIGINT":
if(param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
stringValue = dataSession.getStringData(dataAttr);
if(stringValue==null)
{
callStmt.setNull(col + 1,Types.BIGINT);
}
else {
longValue = Long.parseLong(stringValue);
callStmt.setLong(col + 1, longValue);
}
}
else {
callStmt.registerOutParameter(col+1, Types.BIGINT);
}
break;
case "DOUBLE":
if(param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
stringValue = dataSession.getStringData(dataAttr);
if(stringValue==null)
{
callStmt.setNull(col + 1,Types.DOUBLE);
}
else {
doubleValue = Double.parseDouble(stringValue);
callStmt.setDouble(col + 1, doubleValue);
}
}
else {
callStmt.registerOutParameter(col+1, Types.BIGINT);
}
break;
case "FLOAT":
if(param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
stringValue = dataSession.getStringData(dataAttr);
if(stringValue==null)
{
callStmt.setNull(col + 1,Types.FLOAT);
}
else {
floatValue = Float.parseFloat(stringValue);
callStmt.setFloat(col + 1, floatValue);
}
}
else {
callStmt.registerOutParameter(col+1, Types.BIGINT);
}
break;
case "BIT":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
stringValue = dataSession.getStringData(dataAttr);
if(stringValue.equalsIgnoreCase("true"))
stringValue="1";
else
stringValue="0";
if(stringValue==null) {
callStmt.setNull(col+1, Types.BIT);
}
else
callStmt.setString(col+1, stringValue);
}
else {
callStmt.registerOutParameter(col+1, Types.BIT);
}
break;
case "CHAR":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
stringValue = dataSession.getStringData(dataAttr);
if(stringValue==null) {
callStmt.setNull(col+1, Types.CHAR);
}
else
callStmt.setString(col+1, stringValue);
}
else {
callStmt.registerOutParameter(col+1, Types.CHAR);
}
break;
case "VARCHAR":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
stringValue = dataSession.getStringData(dataAttr);
if(stringValue==null) {
callStmt.setNull(col+1, Types.VARCHAR);
}
else
callStmt.setString(col+1, stringValue);
}
else {
callStmt.registerOutParameter(col+1, Types.VARCHAR);
}
break;
case "DATETIME":
case "TIMESTAMP":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
ts = dataSession.getDateTimeData(dataAttr);
if(ts==null) {
callStmt.setNull(col+1, Types.TIMESTAMP);
}
else
callStmt.setTimestamp(col+1,ts);
}
else {
callStmt.registerOutParameter(col+1, Types.TIMESTAMP);
}
break;
case "DATE":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
ts = dataSession.getDateTimeData(dataAttr);
if(ts==null) {
callStmt.setNull(col+1, Types.DATE);
}
else {
dt = new Date(ts.getTime());
callStmt.setDate(col+1,dt);
}
}
else {
callStmt.registerOutParameter(col+1, Types.DATE);
}
break;
case "TIME":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
ts = dataSession.getDateTimeData(dataAttr);
if(ts==null) {
callStmt.setNull(col+1, Types.TIME);
}
else {
time = new Time(ts.getTime());
callStmt.setTime(col+1,time);
}
}
else {
callStmt.registerOutParameter(col+1, Types.TIME);
}
break;
case "DECIMAL":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
bdc=dataSession.getBigDecimalData(dataAttr);
if(bdc==null) {
callStmt.setNull(col+1, Types.DECIMAL);
}
else
callStmt.setBigDecimal(col+1,bdc);
}
else {
callStmt.registerOutParameter(col+1, Types.DECIMAL);
}
break;
case "NUMERIC":
if (param.getParameterTypeEnum()==ParameterTypeEnum.IN_TYPE || param.getParameterTypeEnum()==ParameterTypeEnum.INOUT_TYPE) {
dataAttr.setRowIndex(i);
dataAttr.setColumnIndex(col);
bdc=dataSession.getBigDecimalData(dataAttr);
if(bdc==null) {
callStmt.setNull(col+1, Types.NUMERIC);
}
else
callStmt.setBigDecimal(col+1,bdc);
}
else {
callStmt.registerOutParameter(col+1, Types.NUMERIC);
}
break;
default :
logger.logMessage(EMessageLevel.MSG_ERROR, ELogLevel.TRACE_NONE,"MySQL Stored Procedure encountered failure for the field " + param.getNativeName() + " with the value " + dataSession.getStringData(dataAttr));
logger.logMessage(EMessageLevel.MSG_ERROR, ELogLevel.TRACE_NONE,"MySQL Stored Procedure is not supported for the datatype "+dataType);
return EReturnStatus.FAILURE;
}
col++;
}
callStmt.executeQuery();
List<Object> resultRow = new ArrayList<Object>();
for (Parameter outParam : this.outputParamList) {
Object val=null;
switch(outParam.getDataType().toUpperCase()) {
case "TIMESTAMP":
val = callStmt.getTimestamp(outParam.getName());
resultRow.add(val);
break;
case "DATE":
val= callStmt.getDate(outParam.getName());
resultRow.add(val);
break;
case "TIME":
val = callStmt.getTime(outParam.getName());
resultRow.add(val);
break;
default:
val = callStmt.getObject(outParam.getName());
resultRow.add(val);
break;
}
}
result.add(resultRow);
}
setProcDataToPlatform(dataSession,result);
callAttr.setNumRowsInOutputBuffer(result.size());
if(callStmt!=null)
callStmt.close();
} catch (SQLException e) {
e.printStackTrace();
return EReturnStatus.FAILURE;
}
return EReturnStatus.NO_MORE_DATA;
}
