LoginRetryStrategy.java

package com.bao.lc.site.s3.commands;

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

import com.bao.lc.bean.IDValuePair;
import com.bao.lc.bean.ResultCode;
import com.bao.lc.httpcommand.CommandRetryStrategy;
import com.bao.lc.site.s3.params.TdPNames;

public class LoginRetryStrategy implements CommandRetryStrategy
{
	private static Log log = LogFactory.getLog(LoginRetryStrategy.class);
	
	@Override
	public boolean shouldRetry(Context context, int executionCount, IDValuePair rc)
	{
		boolean ret = false;
		
		if(rc == ResultCode.RC_TD_LOGIN_ACCOUNT_ERROR || rc == ResultCode.RC_TD_LOGIN_PASSWORD_ERROR)
		{
			ret = false;
		}
		else if(rc == ResultCode.RC_TD_LOGIN_VCODE_ERROR)
		{
			//retry with verification code
			context.remove(TdPNames._LOGIN_VOCDE);
			
			ret = true;
		}
		else if(rc == ResultCode.RC_TD_LOGIN_NO_SUBMIT_FORM_ERROR)
		{
			//start from GetLoginPage
			context.remove(TdPNames._LOGIN_PAGE_CONTENT);
			context.remove(TdPNames._LOGIN_PAGE_ENCODING);
			
			ret = true;
		}
		else if(rc == ResultCode.RC_TD_LOGIN_OTHER_MSG_ERROR)
		{
			//retry without user's input
			ret = true;
		}
		//do not retry with all the other results
		else
		{
			ret = false;
		}

		if(ret)
		{
			context.put(TdPNames._IS_FIRST_LOGIN, Boolean.FALSE);
		}
		
		log.debug("[LoginRetryStrategy]: retry=" + ret + ", rc=" + rc);
		return ret;
	}

	@Override
	public boolean shouldRetry(Context context, int executionCount, Exception e)
	{
		return false;
	}

}