Examples with Packet com.sun.messaging.jmq.io.Packet used on opensource projects

Example 11 with Packet

use of com.sun.messaging.jmq.io.Packet in project openmq by eclipse-ee4j.

the class ClusterMessageInfo method getGPacket.

public GPacket getGPacket() throws Exception {
    assert (ref != null);
    assert (consumers != null);
    GPacket gp = GPacket.getInstance();
    gp.setType(ProtocolGlobals.G_MESSAGE_DATA);
    gp.putProp("D", Boolean.valueOf(sendMessageDeliveredAck));
    gp.putProp("C", Integer.valueOf(consumers.size()));
    if (Globals.getDestinationList().isPartitionMode()) {
        gp.putProp("partitionID", Long.valueOf(ref.getPartitionedStore().getPartitionID().longValue()));
    }
    c.marshalBrokerAddress(c.getSelfAddress(), gp);
    ByteArrayOutputStream bos = new ByteArrayOutputStream();
    DataOutputStream dos = new DataOutputStream(bos);
    Packet roPkt = null;
    try {
        for (int i = 0; i < consumers.size(); i++) {
            ConsumerUID intid = consumers.get(i).getConsumerUID();
            ClusterConsumerInfo.writeConsumerUID(intid, dos);
            gp.putProp(PROP_PREFIX_CUID_DCT + intid.longValue(), deliveryCnts.get(i));
        }
        if (redelivered) {
            gp.putProp(PROP_REDELIVERED, Boolean.valueOf(redelivered));
        }
        roPkt = ref.getPacket();
        if (roPkt == null) {
            throw new BrokerException(Globals.getBrokerResources().getKString(BrokerResources.X_NULL_PACKET_FROM_REF, ref.toString()));
        }
        roPkt.generateTimestamp(false);
        roPkt.generateSequenceNumber(false);
        roPkt.writePacket(dos);
        dos.flush();
        bos.flush();
    } catch (Exception e) {
        String emsg = Globals.getBrokerResources().getKString(BrokerResources.X_EXCEPTION_WRITE_PKT_ON_SEND_MSG_REMOTE, ref.toString(), e.getMessage());
        if (e instanceof BrokerException) {
            logger.log(Logger.WARNING, emsg);
            throw e;
        }
        logger.logStack(Logger.WARNING, emsg, e);
        throw e;
    }
    byte[] buf = bos.toByteArray();
    gp.setPayload(ByteBuffer.wrap(buf));
    return gp;
}

Example 12 with Packet

use of com.sun.messaging.jmq.io.Packet in project openmq by eclipse-ee4j.

the class DestinationList method loadTakeoverMsgs.

public static synchronized void loadTakeoverMsgs(PartitionedStore storep, Map<String, String> msgs, List txns, Map txacks) throws BrokerException {
    DestinationList dl = destinationListList.get(storep);
    Map m = new HashMap();
    Logger logger = Globals.getLogger();
    Map ackLookup = new HashMap();
    // ok create a hashtable for looking up txns
    if (txacks != null) {
        Iterator itr = txacks.entrySet().iterator();
        while (itr.hasNext()) {
            Map.Entry entry = (Map.Entry) itr.next();
            TransactionUID tuid = (TransactionUID) entry.getKey();
            List l = (List) entry.getValue();
            Iterator litr = l.iterator();
            while (litr.hasNext()) {
                TransactionAcknowledgement ta = (TransactionAcknowledgement) litr.next();
                String key = ta.getSysMessageID() + ":" + ta.getStoredConsumerUID();
                ackLookup.put(key, tuid);
            }
        }
    }
    // Alright ...
    // all acks fail once takeover begins
    // we expect all transactions to rollback
    // here is the logic:
    // - load all messages
    // - remove any messages in open transactions
    // - requeue all messages
    // - resort (w/ load comparator)
    // 
    // 
    // OK, first get msgs and sort by destination
    HashMap openMessages = new HashMap();
    Iterator itr = msgs.entrySet().iterator();
    while (itr.hasNext()) {
        Map.Entry me = (Map.Entry) itr.next();
        String msgID = (String) me.getKey();
        String dst = (String) me.getValue();
        DestinationUID dUID = new DestinationUID(dst);
        Packet p = null;
        try {
            p = storep.getMessage(dUID, msgID);
        } catch (BrokerException ex) {
            Throwable cause = ex.getCause();
            if (cause instanceof InvalidPacketException) {
                String[] args = { msgID, dst, cause.toString() };
                String emsg = Globals.getBrokerResources().getKString(BrokerResources.X_MSG_CORRUPTED_IN_STORE, args);
                logger.logStack(Logger.ERROR, emsg, ex);
                handleInvalidPacket(msgID, dst, emsg, (InvalidPacketException) cause, storep);
                itr.remove();
                continue;
            }
            // Check if dst even exists!
            if (ex.getStatusCode() == Status.NOT_FOUND) {
                Destination[] ds = getDestination(storep, dUID);
                Destination d = ds[0];
                if (d == null) {
                    String[] args = { msgID, dst, Globals.getBrokerResources().getString(BrokerResources.E_DESTINATION_NOT_FOUND_IN_STORE, dst) };
                    logger.log(Logger.ERROR, BrokerResources.W_CAN_NOT_LOAD_MSG, args, ex);
                }
            }
            throw ex;
        }
        dUID = DestinationUID.getUID(p.getDestination(), p.getIsQueue());
        PacketReference pr = PacketReference.createReference(storep, p, dUID, null);
        // mark already stored and make packet a SoftReference to
        // prevent running out of memory if dest has lots of msgs
        pr.setLoaded();
        logger.log(Logger.DEBUG, "Loading message " + pr.getSysMessageID() + " on " + pr.getDestinationUID());
        // check transactions
        TransactionUID tid = pr.getTransactionID();
        if (tid != null) {
            // see if in transaction list
            if (txns.contains(tid)) {
                // open transaction
                TransactionState ts = dl.getTransactionList().retrieveState(pr.getTransactionID());
                if (ts != null && ts.getState() != TransactionState.ROLLEDBACK && ts.getState() != TransactionState.COMMITTED) {
                    // in transaction ...
                    logger.log(Logger.DEBUG, "Processing open transacted message " + pr.getSysMessageID() + " on " + tid + "[" + TransactionState.toString(ts.getState()) + "]");
                    openMessages.put(pr.getSysMessageID(), tid);
                } else if (ts != null && ts.getState() == TransactionState.ROLLEDBACK) {
                    pr.destroy();
                    continue;
                } else {
                }
            }
        }
        dl.packetlistAdd(pr.getSysMessageID(), pr.getDestinationUID(), null);
        Set l = null;
        if ((l = (Set) m.get(dUID)) == null) {
            l = new TreeSet(new RefCompare());
            m.put(dUID, l);
        }
        l.add(pr);
    }
    // OK, handle determining how to queue the messages
    Map<PacketReference, MessageDeliveryTimeInfo> deliveryDelays = new HashMap<>();
    // first add all messages
    Iterator dsts = m.entrySet().iterator();
    while (dsts.hasNext()) {
        Map.Entry entry = (Map.Entry) dsts.next();
        DestinationUID dst = (DestinationUID) entry.getKey();
        Set l = (Set) entry.getValue();
        Destination[] ds = getDestination(storep, dst);
        Destination d = ds[0];
        if (d == null) {
            // create it
            String destinationName = dst.getName();
            try {
                ds = getDestination(storep, destinationName, (dst.isQueue() ? DestType.DEST_TYPE_QUEUE : DestType.DEST_TYPE_TOPIC), true, true);
                d = ds[0];
            } catch (IOException ex) {
                throw new BrokerException(Globals.getBrokerResources().getKString(BrokerResources.X_CANT_LOAD_DEST, destinationName));
            }
        } else {
            synchronized (d) {
                if (d.isLoaded()) {
                    // Destination has already been loaded so just called
                    // initialize() to update the size and bytes variables
                    d.initialize();
                }
                d.load(l);
            }
        }
        logger.log(Logger.INFO, BrokerResources.I_LOADING_DST, d.getName(), String.valueOf(l.size()));
        MessageDeliveryTimeTimer dt = d.deliveryTimeTimer;
        if (dt == null && !d.isDMQ()) {
            if (!d.isValid()) {
                String emsg = Globals.getBrokerResources().getKString(BrokerResources.W_UNABLE_LOAD_TAKEOVER_MSGS_TO_DESTROYED_DST, d.getDestinationUID());
                logger.log(Logger.WARNING, emsg);
                continue;
            }
            String emsg = Globals.getBrokerResources().getKString(BrokerResources.W_UNABLE_LOAD_TAKEOVER_MSGS_NO_DST_DELIVERY_TIMER, d.getDestinationUID() + "[" + d.isValid() + "]");
            logger.log(Logger.WARNING, emsg);
            continue;
        }
        // now we're sorted, process
        Iterator litr = l.iterator();
        try {
            MessageDeliveryTimeInfo di = null;
            while (litr.hasNext()) {
                PacketReference pr = (PacketReference) litr.next();
                di = pr.getDeliveryTimeInfo();
                if (di != null) {
                    dt.removeMessage(di);
                }
                try {
                    // ok allow overrun
                    boolean el = d.destMessages.getEnforceLimits();
                    d.destMessages.enforceLimits(false);
                    if (DEBUG) {
                        logger.log(logger.INFO, "Put message " + pr + "[" + di + "] to destination " + d);
                    }
                    pr.lock();
                    d.acquireQueueRemoteLock();
                    try {
                        d.putMessage(pr, AddReason.LOADED, true);
                    } finally {
                        d.clearQueueRemoteLock();
                    }
                    // turn off overrun
                    d.destMessages.enforceLimits(el);
                } catch (IllegalStateException | OutOfLimitsException ex) {
                    // thats ok, we already exists
                    String[] args = { pr.getSysMessageID().toString(), pr.getDestinationUID().toString(), ex.getMessage() };
                    logger.logStack(Logger.WARNING, BrokerResources.W_CAN_NOT_LOAD_MSG, args, ex);
                    continue;
                } finally {
                    if (di != null && !di.isDeliveryDue()) {
                        dt.addMessage(di);
                        deliveryDelays.put(pr, di);
                    }
                }
            }
            // then resort the destination
            d.sort(new RefCompare());
        } catch (Exception ex) {
        }
    }
    // now route
    dsts = m.entrySet().iterator();
    while (dsts.hasNext()) {
        Map.Entry entry = (Map.Entry) dsts.next();
        DestinationUID dst = (DestinationUID) entry.getKey();
        Set l = (Set) entry.getValue();
        Destination d = dl.getDestination(dst);
        // now we're sorted, process
        Iterator litr = l.iterator();
        try {
            while (litr.hasNext()) {
                PacketReference pr = (PacketReference) litr.next();
                if (DEBUG) {
                    logger.log(logger.INFO, "Process takeover message " + pr + "[" + pr.getDeliveryTimeInfo() + "] for destination " + d);
                }
                TransactionUID tuid = (TransactionUID) openMessages.get(pr.getSysMessageID());
                if (tuid != null) {
                    dl.getTransactionList().addMessage(tuid, pr.getSysMessageID(), true);
                    pr.unlock();
                    continue;
                }
                ConsumerUID[] consumers = storep.getConsumerUIDs(dst, pr.getSysMessageID());
                if (consumers == null) {
                    consumers = new ConsumerUID[0];
                }
                if (consumers.length == 0 && storep.hasMessageBeenAcked(dst, pr.getSysMessageID())) {
                    logger.log(Logger.INFO, Globals.getBrokerResources().getString(BrokerResources.W_TAKEOVER_MSG_ALREADY_ACKED, pr.getSysMessageID()));
                    d.unputMessage(pr, RemoveReason.ACKNOWLEDGED);
                    pr.destroy();
                    pr.unlock();
                    continue;
                }
                if (consumers.length > 0) {
                    pr.setStoredWithInterest(true);
                } else {
                    pr.setStoredWithInterest(false);
                }
                int[] states = null;
                if (consumers.length == 0 && deliveryDelays.get(pr) == null) {
                    // message
                    try {
                        consumers = d.routeLoadedTransactionMessage(pr);
                    } catch (Exception ex) {
                        logger.logStack(Logger.WARNING, Globals.getBrokerResources().getKString(BrokerResources.W_EXCEPTION_ROUTE_LOADED_MSG, pr.getSysMessageID(), ex.getMessage()), ex);
                    }
                    states = new int[consumers.length];
                    for (int i = 0; i < states.length; i++) {
                        states[i] = PartitionedStore.INTEREST_STATE_ROUTED;
                    }
                    try {
                        storep.storeInterestStates(d.getDestinationUID(), pr.getSysMessageID(), consumers, states, true, null);
                        pr.setStoredWithInterest(true);
                    } catch (Exception ex) {
                        // message already routed
                        StringBuilder debuf = new StringBuilder();
                        for (int i = 0; i < consumers.length; i++) {
                            if (i > 0) {
                                debuf.append(", ");
                            }
                            debuf.append(consumers[i]);
                        }
                        logger.log(logger.WARNING, BrokerResources.W_TAKEOVER_MSG_ALREADY_ROUTED, pr.getSysMessageID(), debuf.toString(), ex);
                    }
                } else if (consumers.length > 0) {
                    states = new int[consumers.length];
                    for (int i = 0; i < consumers.length; i++) {
                        states[i] = storep.getInterestState(dst, pr.getSysMessageID(), consumers[i]);
                    }
                }
                pr.update(consumers, states, false);
                // OK deal w/ transsactions
                // LKS - XXX
                ExpirationInfo ei = pr.getExpireInfo();
                if (ei != null && d.expireReaper != null) {
                    d.expireReaper.addExpiringMessage(ei);
                }
                List<ConsumerUID> consumerList = new ArrayList(Arrays.asList(consumers));
                // OK ... see if we are in txn
                Iterator citr = consumerList.iterator();
                while (citr.hasNext()) {
                    logger.log(Logger.DEBUG, " Message " + pr.getSysMessageID() + " has " + consumerList.size() + " consumers ");
                    ConsumerUID cuid = (ConsumerUID) citr.next();
                    String key = pr.getSysMessageID() + ":" + cuid;
                    TransactionList tl = dl.getTransactionList();
                    TransactionUID tid = (TransactionUID) ackLookup.get(key);
                    if (DEBUG) {
                        logger.log(logger.INFO, "loadTakeoverMsgs: lookup " + key + " found tid=" + tid);
                    }
                    if (tid != null) {
                        boolean remote = false;
                        TransactionState ts = tl.retrieveState(tid);
                        if (ts == null) {
                            ts = tl.getRemoteTransactionState(tid);
                            remote = true;
                        }
                        if (DEBUG) {
                            logger.log(logger.INFO, "tid=" + tid + " has state=" + TransactionState.toString(ts.getState()));
                        }
                        if (ts != null && ts.getState() != TransactionState.ROLLEDBACK && ts.getState() != TransactionState.COMMITTED) {
                            // in transaction ...
                            if (DEBUG) {
                                logger.log(Logger.INFO, "loadTakeoverMsgs: Open transaction ack [" + key + "]" + (remote ? "remote" : "") + ", TUID=" + tid);
                            }
                            if (!remote) {
                                try {
                                    tl.addAcknowledgement(tid, pr.getSysMessageID(), cuid, cuid, true, false);
                                } catch (TransactionAckExistException e) {
                                    // can happen if takeover tid's remote txn after restart
                                    // then txn ack would have already been loaded
                                    logger.log(Logger.INFO, Globals.getBrokerResources().getKString(BrokerResources.I_TAKINGOVER_TXN_ACK_ALREADY_EXIST, "[" + pr.getSysMessageID() + "]" + cuid + ":" + cuid, tid + "[" + TransactionState.toString(ts.getState()) + "]"));
                                }
                                tl.addOrphanAck(tid, pr.getSysMessageID(), cuid);
                            }
                            citr.remove();
                            logger.log(Logger.INFO, "Processing open ack " + pr.getSysMessageID() + ":" + cuid + " on " + tid);
                            continue;
                        } else if (ts != null && ts.getState() == TransactionState.COMMITTED) {
                            logger.log(Logger.INFO, "Processing committed ack " + pr.getSysMessageID() + ":" + cuid + " on " + tid);
                            if (pr.acknowledged(cuid, cuid, false, true)) {
                                d.unputMessage(pr, RemoveReason.ACKNOWLEDGED);
                                pr.destroy();
                                continue;
                            }
                            citr.remove();
                            continue;
                        }
                    }
                }
                // route msgs not in transaction
                if (DEBUG) {
                    StringBuilder buf = new StringBuilder();
                    ConsumerUID cid = null;
                    for (int j = 0; j < consumerList.size(); j++) {
                        cid = consumerList.get(j);
                        buf.append(cid);
                        buf.append(' ');
                    }
                    if (deliveryDelays.get(pr) == null) {
                        logger.log(Logger.INFO, "non-transacted: Routing Message " + pr.getSysMessageID() + " to " + consumerList.size() + " consumers:" + buf.toString());
                    } else {
                        logger.log(Logger.INFO, "non-transacted: deliver time not arrived for message " + pr.getSysMessageID());
                    }
                }
                pr.unlock();
                if (deliveryDelays.get(pr) == null) {
                    if (DEBUG) {
                        logger.log(logger.INFO, "Route takeover message " + pr + "[" + pr.getDeliveryTimeInfo() + "] for destination " + d + " to consumers " + consumerList);
                    }
                    if (pr.getDeliveryTimeInfo() != null) {
                        d.forwardDeliveryDelayedMessage(new HashSet<>(consumerList), pr);
                    } else {
                        d.routeLoadedMessage(pr, consumerList);
                    }
                } else {
                    MessageDeliveryTimeInfo di = pr.getDeliveryTimeInfo();
                    di.setDeliveryReady();
                }
                if (d.destReaper != null) {
                    d.destReaper.cancel();
                    d.destReaper = null;
                }
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }
}

Example 13 with Packet

use of com.sun.messaging.jmq.io.Packet in project openmq by eclipse-ee4j.

the class DestinationList method remoteCheckTakeoverMsgs.

public static void remoteCheckTakeoverMsgs(Map<String, String> msgs, String brokerid, PartitionedStore ps) throws BrokerException {
    Set destroyConns = new HashSet();
    Map<String, String> badsysids = null;
    Iterator<Map.Entry<String, String>> itr = msgs.entrySet().iterator();
    Map.Entry<String, String> me = null;
    String sysidstr = null, duidstr = null;
    SysMessageID sysid = null;
    DestinationUID duid = null;
    while (itr.hasNext()) {
        me = itr.next();
        sysidstr = me.getKey();
        duidstr = me.getValue();
        try {
            sysid = SysMessageID.get(sysidstr);
        } catch (InvalidSysMessageIDException e) {
            Globals.getLogger().logStack(Logger.ERROR, e.getMessage(), e);
            if (!Globals.getStore().getStoreType().equals(Store.JDBC_STORE_TYPE)) {
                throw e;
            }
            duid = new DestinationUID(duidstr);
            Globals.getLogger().log(Logger.WARNING, Globals.getBrokerResources().getKString(BrokerResources.X_TAKEOVER_MSGID_CORRUPT_TRY_REPAIR, sysidstr, duidstr));
            try {
                Packet p = null;
                try {
                    p = ps.getMessage(duid, sysidstr);
                } catch (BrokerException ee) {
                    Throwable cause = ee.getCause();
                    String[] args = { sysidstr, "[?]", duidstr, cause.toString() };
                    String emsg = Globals.getBrokerResources().getKString(BrokerResources.X_REPAIR_CORRUPTED_MSGID_IN_STORE, args);
                    Globals.getLogger().logStack(Logger.ERROR, emsg, ee);
                    if (cause instanceof InvalidPacketException) {
                        handleInvalidPacket(sysidstr, duidstr, emsg, (InvalidPacketException) cause, ps);
                        itr.remove();
                        continue;
                    } else {
                        throw ee;
                    }
                }
                sysid = p.getSysMessageID();
                String realsysidstr = sysid.getUniqueName();
                String[] args3 = { sysidstr, realsysidstr, duidstr };
                Globals.getLogger().log(Logger.INFO, Globals.getBrokerResources().getKString(BrokerResources.X_REPAIR_CORRUPTED_MSGID_TO, args3));
                ps.repairCorruptedSysMessageID(sysid, sysidstr, duidstr, true);
                if (badsysids == null) {
                    badsysids = new HashMap<>();
                }
                badsysids.put(sysidstr, realsysidstr);
            } catch (BrokerException ee) {
                Globals.getLogger().logStack(Logger.ERROR, e.getMessage(), ee);
                throw e;
            }
        }
        PacketReference ref = get(null, sysid);
        if (ref == null) {
            continue;
        }
        Iterator cnitr = ref.getRemoteConsumerUIDs().values().iterator();
        while (cnitr.hasNext()) {
            destroyConns.add(cnitr.next());
        }
    }
    if (badsysids != null) {
        itr = badsysids.entrySet().iterator();
        String v = null;
        while (itr.hasNext()) {
            me = itr.next();
            v = msgs.remove(me.getKey());
            msgs.put(me.getValue(), v);
        }
    }
    destroyConnections(destroyConns, GoodbyeReason.BKR_IN_TAKEOVER, GoodbyeReason.toString(GoodbyeReason.BKR_IN_TAKEOVER) + ":" + brokerid);
}

Example 14 with Packet

use of com.sun.messaging.jmq.io.Packet in project openmq by eclipse-ee4j.

the class JDBCStore method updateTransactionStateWithWorkInternal.

public void updateTransactionStateWithWorkInternal(TransactionUID tid, TransactionState ts, TransactionWork txnwork, long storeSessionID, boolean sync) throws BrokerException {
    if (tid == null || ts == null || txnwork == null) {
        throw new NullPointerException();
    }
    if (DEBUG) {
        logger.log(Logger.INFO, "JDBCStore.updateTransactionStateInternal(" + tid + ", " + ts + ", " + ", " + txnwork + ", " + sync + ")");
    }
    Connection conn = null;
    Exception myex = null;
    try {
        conn = dbmgr.getConnection(false);
        Util.RetryStrategy retry = null;
        do {
            boolean inside = false;
            try {
                Iterator<TransactionWorkMessage> itr1 = txnwork.getSentMessages().iterator();
                while (itr1.hasNext()) {
                    TransactionWorkMessage txnmsg = itr1.next();
                    Packet m = txnmsg.getMessage();
                    if (m == null) {
                        continue;
                    }
                    inside = true;
                    daoFactory.getMessageDAO().insert(conn, txnmsg.getDestUID(), m, null, null, storeSessionID, m.getTimestamp(), true, false);
                    inside = false;
                }
                if (FI.FAULT_INJECTION) {
                    try {
                        FI.checkFaultAndThrowBrokerException(FaultInjection.FAULT_TXN_PERSIST_WORK_1_5, null);
                    } catch (BrokerException e) {
                        FI.unsetFault(FI.FAULT_TXN_PERSIST_WORK_1_5);
                        throw e;
                    }
                }
                List<TransactionWorkMessageAck> txnacks = txnwork.getMessageAcknowledgments();
                if (txnacks != null) {
                    Iterator<TransactionWorkMessageAck> itr2 = txnacks.iterator();
                    while (itr2.hasNext()) {
                        TransactionWorkMessageAck txnack = itr2.next();
                        TransactionAcknowledgement ta = txnack.getTransactionAcknowledgement();
                        if (ta != null) {
                            inside = true;
                            daoFactory.getConsumerStateDAO().updateTransaction(conn, ta.getSysMessageID(), ta.getStoredConsumerUID(), tid);
                            inside = false;
                        }
                    }
                }
                inside = true;
                daoFactory.getTransactionDAO().updateTransactionState(conn, tid, ts, false);
                inside = false;
                conn.commit();
                return;
            } catch (Exception e) {
                if (!inside) {
                    try {
                        conn.rollback();
                    } catch (SQLException rbe) {
                        logger.logStack(Logger.WARNING, BrokerResources.X_DB_ROLLBACK_FAILED, rbe);
                    }
                }
                if (retry == null) {
                    retry = new Util.RetryStrategy();
                }
                try {
                    retry.assertShouldRetry(e, conn);
                } catch (RetrySQLRecoverableException ee) {
                    try {
                        Util.close(null, null, conn, ee);
                        conn = dbmgr.getConnection(false);
                    } catch (Exception eee) {
                        logger.logStack(Logger.WARNING, eee.getMessage(), eee);
                        conn = null;
                        if (e instanceof BrokerException) {
                            throw (BrokerException) e;
                        }
                        throw new BrokerException(e.getMessage(), e);
                    }
                }
            }
        } while (true);
    } catch (BrokerException e) {
        myex = e;
        throw e;
    } finally {
        Util.close(null, null, conn, myex);
    }
}

Example 15 with Packet

use of com.sun.messaging.jmq.io.Packet in project openmq by eclipse-ee4j.

the class UpgradeHAStore method doUpgradeMsg.

private void doUpgradeMsg(Connection conn) throws BrokerException {
    MessageDAO msgDAO = dbMgr.getDAOFactory().getMessageDAO();
    String oldmsgtbl = MessageDAO.TABLE_NAME_PREFIX + "S" + brokerID;
    HashMap msgToDst = new HashMap();
    // SQL to select all messages from Non-HA table
    String getAllMsgFromOldSQL = new StringBuilder(128).append("SELECT ").append(MessageDAO.ID_COLUMN).append(", ").append(MessageDAO.MESSAGE_COLUMN).append(", ").append(MessageDAO.DESTINATION_ID_COLUMN).append(", ").append(MessageDAO.STORE_SESSION_ID_COLUMN).append(", ").append(MessageDAO.CREATED_TS_COLUMN).append(" FROM ").append(oldmsgtbl).toString();
    Statement stmt = null;
    ResultSet rs = null;
    String msgID = null;
    Packet msg = null;
    Exception myex = null;
    try {
        stmt = conn.createStatement();
        rs = dbMgr.executeQueryStatement(stmt, getAllMsgFromOldSQL);
        while (rs.next()) {
            msgID = rs.getString(1);
            msg = new Packet(false);
            msg.generateTimestamp(false);
            msg.generateSequenceNumber(false);
            InputStream is = null;
            Blob blob = rs.getBlob(2);
            is = blob.getBinaryStream();
            msg.readPacket(is);
            is.close();
            String dstID = rs.getString(3);
            long sessionID = rs.getLong(4);
            long createdTS = rs.getLong(5);
            try {
                msgDAO.insert(conn, dstID, msg, null, null, sessionID, createdTS, true, false);
            } catch (BrokerException be) {
                // If msg exist, just logged and continue
                if (be.getStatusCode() == Status.CONFLICT) {
                    logger.log(Logger.WARNING, be.getMessage() + ": Ignore");
                } else {
                    throw be;
                }
            }
            msgToDst.put(msgID, dstID);
        }
    } catch (Exception e) {
        myex = e;
        String errorMsg = br.getKString(BrokerResources.X_JDBC_UPGRADE_MESSAGES_FAILED, (msg == null ? msgID : msg.getSysMessageID().toString()));
        logger.logStack(Logger.ERROR, errorMsg, e);
        throw new BrokerException(errorMsg, e);
    } finally {
        Util.close(rs, stmt, null, myex);
    }
    // upgrade interest list
    ConsumerStateDAO stateDAO = dbMgr.getDAOFactory().getConsumerStateDAO();
    String oldstatetbl = ConsumerStateDAO.TABLE_NAME_PREFIX + "S" + brokerID;
    // SQL to select all interest states from Non-HA table
    String getAllStateFromOldSQL = new StringBuilder(128).append("SELECT ").append(ConsumerStateDAO.MESSAGE_ID_COLUMN).append(", ").append(ConsumerStateDAO.CONSUMER_ID_COLUMN).append(", ").append(ConsumerStateDAO.STATE_COLUMN).append(", ").append(ConsumerStateDAO.TRANSACTION_ID_COLUMN).append(", ").append(ConsumerStateDAO.CREATED_TS_COLUMN).append(" FROM ").append(oldstatetbl).append(" WHERE ").append(TINTSTATE_CSTATE).append(" <> ").append(PartitionedStore.INTEREST_STATE_ACKNOWLEDGED).toString();
    String insertStateSQL = new StringBuilder(128).append("INSERT INTO ").append(stateDAO.getTableName()).append(" ( ").append(ConsumerStateDAO.MESSAGE_ID_COLUMN).append(", ").append(ConsumerStateDAO.CONSUMER_ID_COLUMN).append(", ").append(ConsumerStateDAO.STATE_COLUMN).append(", ").append(ConsumerStateDAO.TRANSACTION_ID_COLUMN).append(", ").append(ConsumerStateDAO.CREATED_TS_COLUMN).append(") VALUES ( ?, ?, ?, ?, ? )").toString();
    boolean dobatch = dbMgr.supportsBatchUpdates();
    PreparedStatement pstmt = null;
    String mid = null;
    long cuid = 0;
    try {
        pstmt = dbMgr.createPreparedStatement(conn, insertStateSQL);
        stmt = conn.createStatement();
        rs = dbMgr.executeQueryStatement(stmt, getAllStateFromOldSQL);
        while (rs.next()) {
            mid = rs.getString(1);
            cuid = rs.getLong(2);
            int state = rs.getInt(3);
            long txnID = rs.getLong(4);
            long createdTS = rs.getLong(5);
            String dst = (String) msgToDst.get(mid);
            // ignore a state whose dst or message does not exists
            if (dst == null) {
                continue;
            }
            try {
                pstmt.setString(1, mid);
                pstmt.setLong(2, cuid);
                pstmt.setInt(3, state);
                if (txnID > 0) {
                    pstmt.setLong(4, txnID);
                } else {
                    pstmt.setNull(4, Types.BIGINT);
                }
                pstmt.setLong(5, createdTS);
                if (dobatch) {
                    pstmt.addBatch();
                } else {
                    pstmt.executeUpdate();
                }
            } catch (SQLException e) {
                SQLException ex = DBManager.wrapSQLException("[" + insertStateSQL + "]", e);
                throw ex;
            }
        }
        msgToDst.clear();
        if (dobatch) {
            pstmt.executeBatch();
        }
        conn.commit();
    } catch (SQLException e) {
        myex = e;
        String errorMsg = br.getKString(BrokerResources.X_JDBC_UPGRADE_MESSAGES_FAILED, (mid == null ? "loading" : mid));
        logger.logStack(Logger.ERROR, errorMsg, e);
        throw new BrokerException(errorMsg, e);
    } finally {
        Util.close(rs, stmt, null, myex);
        Util.close(null, pstmt, null, myex);
    }
}