com.qa.framework.cache.AdminCache Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of smart-ui-framework Show documentation

Support web and moblie automaton based on selenium and appium

There is a newer version: 3.8

package com.qa.framework.cache;

import com.qa.framework.library.admin.AdminAccount;
import com.qa.framework.library.admin.BaseAdminBean;
import com.qa.framework.library.admin.XmlToBean;
import org.apache.log4j.Logger;

import java.util.ArrayList;
import java.util.List;

/**
 * The type Admin cache.
 */
public class AdminCache {
    /**
     * The constant logger.
     */
    protected static Logger logger = Logger.getLogger(AdminCache.class);
    private static ThreadLocal accountThreadLocal = new ThreadLocal();
    private static List currentUsingAdmin = new ArrayList<>();
    private static List allAdmin = new ArrayList<>();

    static {
        initList();
    }

    /**
     * Init list.
     *
     * @param adminAccount the admin account
     */
    public static void initList(AdminAccount adminAccount) {
        allAdmin.add(adminAccount);
    }

    /**
     * Init list.
     */
    public static void initList() {
        BaseAdminBean baseAdminBean = XmlToBean.getBaseAdminBean();
        for (int i = 0; i < baseAdminBean.getAdminList().size(); i++) {
            AdminAccount adminAccount = baseAdminBean.getAdminList().get(i);
            initList(adminAccount);
        }
    }

    private static AdminAccount getUnUsedAdmin() {
        for (AdminAccount adminAccount : allAdmin) {
            if (!currentUsingAdmin.contains(adminAccount)) {
                return adminAccount;
            }
        }
        throw new RuntimeException("对不起, 无可用的后台账号");
    }

    /**
     * Gets admin.
     *
     * @return the admin
     */
    public static AdminAccount getAdmin() {
        if (currentUsingAdmin.size() == allAdmin.size()) {
            throw new RuntimeException("对不起, 无可用的后台账号");
        }
        AdminAccount adminAccount = accountThreadLocal.get();
        if (adminAccount != null) {
            logger.info("AdminCache use exist " + adminAccount.getName());
            return adminAccount;
        } else {
            adminAccount = getUnUsedAdmin();
            logger.info("AdminCache use " + adminAccount.getName());
            accountThreadLocal.set(adminAccount);
            currentUsingAdmin.add(adminAccount);
            return adminAccount;
        }
    }

    /**
     * Recover admin.
     */
    public static void recoverAdmin() {
        AdminAccount adminAccount = accountThreadLocal.get();
        if (adminAccount != null) {
            logger.info("AdminCache recover account " + adminAccount.getName());
            accountThreadLocal.set(null);
            currentUsingAdmin.remove(adminAccount);
        }
    }


}