com.espertech.esperio.db.core.RunnableUpsert Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of esperio-db Show documentation
EsperIO DB
There is a newer version: 9.0.0
/**************************************************************************************
 * Copyright (C) 2008 EsperTech, Inc. All rights reserved.                            *
 * http://esper.codehaus.org                                                          *
 * http://www.espertech.com                                                           *
 * ---------------------------------------------------------------------------------- *
 * The software in this package is published under the terms of the GPL license       *
 * a copy of which has been included with this distribution in the license.txt file.  *
 **************************************************************************************/
package com.espertech.esperio.db.core;

import com.espertech.esper.client.EventBean;
import com.espertech.esper.epl.db.DatabaseConfigException;
import com.espertech.esper.util.ExecutionPathDebugLog;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import java.sql.Connection;
import java.sql.SQLException;

public class RunnableUpsert implements Runnable
{
    private static Log log = LogFactory.getLog(RunnableUpsert.class);

    private final RunnableUpsertContext context;
    private final EventBean theEvent;

    public RunnableUpsert(RunnableUpsertContext context, EventBean theEvent)
    {
        this.context = context;
        this.theEvent = theEvent;
    }

    public void run()
    {
        if ((ExecutionPathDebugLog.isDebugEnabled) && (log.isDebugEnabled() && (ExecutionPathDebugLog.isTimerDebugEnabled))) {
            log.debug("Executing upsert work unit for event " + theEvent);
        }

        int retryMax = context.getRetry() == null ? 1 : context.getRetry();
        int retryCount = 0;

        while(true) {
            try {
                tryUpsert();
                break;
            }
            catch (Throwable t) {
                log.error("Error in upsert named '" + context.getName() + "' :" + t.getMessage(), t);
                retryCount++;
                if (retryCount >= retryMax) {
                    log.warn("Failed upsert named '" + context.getName() + "', retry count reached");
                    break;
                }
                if ((context.getRetryWait() != null) && (context.getRetryWait() > 0)) {
                    long interval = (long) (context.getRetryWait() * 1000);
                    log.warn("Retry upsert named '" + context.getName() + "', retry interval msec " + interval + " retry count " + retryCount + " max " + retryMax);
                    try {
                        Thread.sleep(interval);
                    }
                    catch (InterruptedException e) {
                        break;
                    }
                }
                else {
                    log.warn("Retry upsert named '" + context.getName() + "', retry count " + retryCount + " max " + retryMax);                    
                }
            }
        }
    }

    private void tryUpsert() throws DatabaseConfigException, SQLException
    {
        Connection connection = context.getConnectionFactory().getConnection();
        try {
            Object[] keys = new Object[context.getKeyGetters().length];
            for (int i = 0; i < context.getKeyGetters().length; i++) {
                keys[i] = context.getKeyGetters()[i].get(theEvent);
            }

            Object[] values = new Object[context.getValueGetters().length];
            for (int i = 0; i < context.getValueGetters().length; i++) {
                values[i] = context.getValueGetters()[i].get(theEvent);
            }

            boolean updated = context.getTable().updateValue(connection, keys, values);

            if (!updated) {
                context.getTable().insertValue(connection, keys, values);
            }
        }
        finally {
            connection.close();
        }
    }
}