com.tesora.dve.db.mysql.portal.MSPAuthenticateV10Message.java Source code

Introduction

Here is the source code for com.tesora.dve.db.mysql.portal.MSPAuthenticateV10Message.java

Source

package com.tesora.dve.db.mysql.portal;

/*
 * #%L
 * Tesora Inc.
 * Database Virtualization Engine
 * %%
 * Copyright (C) 2011 - 2014 Tesora Inc.
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

import io.netty.channel.ChannelHandlerContext;
import io.netty.util.CharsetUtil;

import java.util.concurrent.ExecutorService;

import com.tesora.dve.charset.NativeCharSet;
import com.tesora.dve.charset.MysqlNativeCharSetCatalog;
import com.tesora.dve.db.DBEmptyTextResultConsumer;
import com.tesora.dve.db.mysql.libmy.MyErrorResponse;
import com.tesora.dve.db.mysql.libmy.MyMessage;
import com.tesora.dve.db.mysql.libmy.MyOKResponse;
import com.tesora.dve.db.mysql.portal.protocol.MSPAuthenticateV10MessageMessage;
import com.tesora.dve.db.mysql.portal.protocol.MSPMessage;
import com.tesora.dve.exceptions.PEException;
import com.tesora.dve.server.connectionmanager.SSConnection;
import com.tesora.dve.server.connectionmanager.messages.ExecuteRequestExecutor;
import com.tesora.dve.worker.UserCredentials;

public class MSPAuthenticateV10Message extends MSPActionBase {

    @Override
    public void execute(ExecutorService clientExecutorService, ChannelHandlerContext ctx, SSConnection ssCon,
            MSPMessage protocolMessage) throws PEException {

        MSPAuthenticateV10MessageMessage authMessage = castProtocolMessage(MSPAuthenticateV10MessageMessage.class,
                protocolMessage);

        ssCon.setClientCapabilities(authMessage.getClientCapabilities());

        byte clientCharsetId = authMessage.getCharsetID();
        String username = authMessage.getUsername();
        String password = authMessage.getPassword();

        // Login in the SSConnection
        MyMessage mysqlResp;
        try {
            UserCredentials userCred = new UserCredentials(username, password, false);
            ssCon.startConnection(userCred);
            String initialDB = authMessage.getInitialDatabase();
            if (!"".equals(authMessage.getInitialDatabase())) {
                final DBEmptyTextResultConsumer resultConsumer = new DBEmptyTextResultConsumer();
                byte[] query = ("USE " + initialDB).getBytes(CharsetUtil.UTF_8);
                ExecuteRequestExecutor.execute(ssCon, resultConsumer, query);
            }

            NativeCharSet cliendCharSet = MysqlNativeCharSetCatalog.DEFAULT_CATALOG
                    .findCharSetByCollationId(clientCharsetId);
            if (cliendCharSet != null) {
                mysqlResp = new MyOKResponse();
                ssCon.setClientCharSet(cliendCharSet);
            } else {
                mysqlResp = new MyErrorResponse(
                        new PEException("Unsupported character set specified (id=" + clientCharsetId + ")"));
            }
        } catch (PEException e) {
            mysqlResp = new MyErrorResponse(e.rootCause());
        } catch (Throwable t) {
            mysqlResp = new MyErrorResponse(new Exception(t.getMessage()));
        }

        ctx.writeAndFlush(mysqlResp);
    }

    @Override
    public byte getMysqlMessageType() {
        return (byte) 0xc0;
    }

    private static final class InstanceHolder {
        static final MSPAuthenticateV10Message INSTANCE = new MSPAuthenticateV10Message();
    }

    public static MSPAuthenticateV10Message getInstance() {
        return InstanceHolder.INSTANCE;
    }
}