自定义倒计时按钮

自定义倒计时按钮

1. 效果
   

2. 代码：

package com.dylan.frame.ui;

import android.content.Context;
import android.os.Handler;
import android.os.Message;
import android.util.AttributeSet;
import android.view.View;
import android.widget.Button;

import java.util.Timer;
import java.util.TimerTask;

/**
 * 自定义倒计时按钮
 * <p/>
 *
 * @author Dylan
 *         <p/>
 *         [佛祖保佑 永无BUG]
 *         Created by Dylan on 2015/11/5 0005.
 */
public class CountdownButton extends Button implements View.OnClickListener {
    private long lenght = 60 * 1000;//默认倒计时时间；
    private long time;//倒计时时长
    private Timer timer;//开始执行倒计时
    private TimerTask timerTask;//每次倒计时执行的任务
    private String beforeText = "点击获取验证码";
    private String afterText = "秒后重新获取";
    private OnClickListener onClickListener;//按钮点击事件
    /**
     * 更新显示的文本
     */
    private Handler handler = new Handler() {
        @Override
        public void handleMessage(Message msg) {
            super.handleMessage(msg);
            CountdownButton.this.setText(time / 1000 + afterText);
            time -= 1000;
            if (time < 0) {
                CountdownButton.this.setEnabled(true);

                CountdownButton.this.setText(beforeText);

                clearTimer();
            }
        }
    };


    public CountdownButton(Context context) {
        super(context);
        this.setText(beforeText);
        setOnClickListener(this);
    }

    public CountdownButton(Context context, AttributeSet attrs) {
        super(context, attrs);
        setOnClickListener(this);
    }

    public CountdownButton(Context context, AttributeSet attrs, int defStyleAttr) {
        super(context, attrs, defStyleAttr);
        setOnClickListener(this);
    }

    /**
     * 清除倒计时
     */
    private void clearTimer() {
        if (timerTask != null) {
            timerTask.cancel();
            timerTask = null;
        }
        if (timer != null) {
            timer.cancel();
            timer = null;
        }
    }

    /**
     * 设置倒计时时长
     *
     * @param lenght 默认毫秒
     */
    public void setLenght(long lenght) {
        this.lenght = lenght;
    }

    /**
     * 设置未点击时显示的文字
     *
     * @param beforeText
     */
    public void setBeforeText(String beforeText) {
        this.beforeText = beforeText;
    }

    /**
     * 设置未点击后显示的文字
     *
     * @param beforeText
     */
    public void setAfterText(String afterText) {
        this.afterText = afterText;
    }

    /**
     * 点击按钮后的操作
     *
     * @param v
     */
    @Override
    public void onClick(View v) {
        if (onClickListener != null) {
            onClickListener.onClick(v);
        }
        initTimer();
        this.setText(time / 1000 + afterText);
        this.setEnabled(false);
        timer.schedule(timerTask, 0, 1000);
    }

    /**
     * 初始化时间
     */
    private void initTimer() {
        time = lenght;
        timer = new Timer();
        timerTask = new TimerTask() {
            @Override
            public void run() {
                handler.sendEmptyMessage(1);
            }
        };
    }

    /**
     * 设置监听按钮点击事件
     *
     * @param onclickListener
     */
    @Override
    public void setOnClickListener(OnClickListener onclickListener) {
        if (onclickListener instanceof CountdownButton) {
            super.setOnClickListener(onclickListener);

        } else {
            this.onClickListener = onclickListener;
        }

    }
}