com.nabla.dc.server.handler.company.ImportAccountListHandler.java Source code

Introduction

Here is the source code for com.nabla.dc.server.handler.company.ImportAccountListHandler.java

Source

/**
* Copyright 2012 nabla
*
* 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 com.nabla.dc.server.handler.company;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.google.inject.Inject;
import com.nabla.dc.server.ImportErrorManager;
import com.nabla.dc.shared.command.company.AddAccount;
import com.nabla.dc.shared.command.company.ImportAccountList;
import com.nabla.dc.shared.model.company.IImportAccounts;
import com.nabla.wapp.server.auth.IUserSessionContext;
import com.nabla.wapp.server.basic.general.UserPreference;
import com.nabla.wapp.server.database.BatchInsertStatement;
import com.nabla.wapp.server.database.ConnectionTransactionGuard;
import com.nabla.wapp.server.database.Database;
import com.nabla.wapp.server.database.IDatabase;
import com.nabla.wapp.server.database.IReadWriteDatabase;
import com.nabla.wapp.server.database.LockTableGuard;
import com.nabla.wapp.server.database.SqlInsert;
import com.nabla.wapp.server.database.StatementFormat;
import com.nabla.wapp.server.dispatch.AbstractHandler;
import com.nabla.wapp.server.general.Util;
import com.nabla.wapp.server.json.JsonResponse;
import com.nabla.wapp.shared.database.SqlInsertOptions;
import com.nabla.wapp.shared.dispatch.DispatchException;
import com.nabla.wapp.shared.dispatch.InternalErrorException;
import com.nabla.wapp.shared.dispatch.StringResult;
import com.nabla.wapp.shared.general.CommonServerErrors;

public class ImportAccountListHandler extends AbstractHandler<ImportAccountList, StringResult> {

    private static final Log log = LogFactory.getLog(ImportAccountListHandler.class);
    private final IDatabase writeDb;

    @Inject
    public ImportAccountListHandler(@IReadWriteDatabase final IDatabase writeDb) {
        super(true);
        this.writeDb = writeDb;
    }

    @Override
    public StringResult execute(final ImportAccountList cmd, final IUserSessionContext ctx)
            throws DispatchException, SQLException {
        final ImportErrorManager errors = new ImportErrorManager(writeDb, cmd.getBatchId());
        try {
            final JsonResponse json = new JsonResponse();
            json.put(IImportAccounts.SUCCESS, add(cmd, errors, ctx));
            return json.toStringResult();
        } catch (Throwable e) {
            if (log.isErrorEnabled())
                log.error("failed to parse accounts from CSV data", e);
            throw new InternalErrorException(Util.formatInternalErrorDescription(e));
        } finally {
            errors.close();
        }
    }

    private boolean add(final ImportAccountList cmd, final ImportErrorManager errors, final IUserSessionContext ctx)
            throws DispatchException, SQLException {
        final PreparedStatement stmtFile = StatementFormat.prepare(ctx.getReadConnection(),
                "SELECT content, userSessionId FROM import_data WHERE id=?;", cmd.getBatchId());
        try {
            final ResultSet rs = stmtFile.executeQuery();
            try {
                if (!rs.next()) {
                    errors.add(CommonServerErrors.NO_DATA);
                    return false;
                }
                if (!ctx.getSessionId().equals(rs.getString("userSessionId"))) {
                    if (log.isTraceEnabled())
                        log.trace("invalid user session ID");
                    errors.add(CommonServerErrors.ACCESS_DENIED);
                    return false;
                }
                final Connection conn = ctx.getWriteConnection();
                final LockTableGuard lock = new LockTableGuard(conn, "account WRITE");
                try {
                    final ConnectionTransactionGuard guard = new ConnectionTransactionGuard(conn);
                    try {
                        SqlInsertOptions option = cmd.getOverwrite();
                        if (option == SqlInsertOptions.REPLACE) {
                            Database.executeUpdate(ctx.getWriteConnection(),
                                    "UPDATE account SET uname=NULL WHERE company_id=?;", cmd.getCompanyId());
                            option = SqlInsertOptions.OVERWRITE;
                        }
                        final SqlInsert<AddAccount> sql = new SqlInsert<AddAccount>(AddAccount.class, option);
                        final BatchInsertStatement<AddAccount> stmt = sql
                                .prepareBatchStatement(ctx.getWriteConnection());
                        try {
                            final AccountCsvReader csv = new AccountCsvReader(rs.getCharacterStream("content"),
                                    conn, cmd.getCompanyId(), errors);
                            try {
                                if (!csv.read(cmd.isRowHeader(), stmt))
                                    return false;
                            } finally {
                                csv.close();
                            }
                            stmt.execute();
                        } finally {
                            stmt.close();
                        }
                        UserPreference.save(ctx, null, IImportAccounts.PREFERENCE_GROUP, IImportAccounts.ROW_HEADER,
                                cmd.isRowHeader());
                        UserPreference.save(ctx, null, IImportAccounts.PREFERENCE_GROUP, IImportAccounts.OVERWRITE,
                                cmd.getOverwrite());
                        guard.setSuccess();
                        return true;
                    } finally {
                        guard.close();
                    }
                } finally {
                    lock.close();
                }
            } finally {
                rs.close();
            }
        } finally {
            stmtFile.close();
        }
    }

}