如何根据Android Studio中的定时器设置动画时长？

import android.animation.Animator;
import android.animation.AnimatorSet;
import android.animation.ObjectAnimator;
import android.support.v7.app.AppCompatActivity;
import android.os.Bundle;
import android.view.View;
import android.widget.Button;

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

public class MainActivity extends AppCompatActivity {

    private Button animateButton;
    private View animatedView;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        animateButton = findViewById(R.id.animate_button);
        animatedView = findViewById(R.id.animated_view);

        animateButton.setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                // 创建动画对象
                ObjectAnimator scaleXAnimator = ObjectAnimator.ofFloat(animatedView, "scaleX", 1f, 2f);
                ObjectAnimator scaleYAnimator = ObjectAnimator.ofFloat(animatedView, "scaleY", 1f, 2f);

                // 设置动画时长为2秒
                int duration = 2000;
                scaleXAnimator.setDuration(duration);
                scaleYAnimator.setDuration(duration);

                // 创建动画集合
                AnimatorSet animatorSet = new AnimatorSet();
                animatorSet.playTogether(scaleXAnimator, scaleYAnimator);

                // 启动动画
                animatorSet.start();

                // 使用定时器在动画结束后恢复原始状态
                Timer timer = new Timer();
                timer.schedule(new TimerTask() {
                    @Override
                    public void run() {
                        runOnUiThread(new Runnable() {
                            @Override
                            public void run() {
                                // 恢复原始状态
                                animatedView.setScaleX(1f);
                                animatedView.setScaleY(1f);
                            }
                        });
                    }
                }, duration);
            }
        });
    }
}