package com.essa.framework;
/**
 * Created by weicheng on 2018/10/25.
 */

import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicReference;

public enum BrowserEnum {
    Firefox("Firefox","webdriver.gecko.driver","geckodriver.exe"),
    Chrome("Chrome","webdriver.chrome.driver","chromedriver.exe"),
    IE("IE","webdriver.ie.driver","IEDriverServer.exe");

    /**
     * 原子引用(全部)
     */
    private static AtomicReference<ConcurrentHashMap<String, BrowserEnum>> mapAllRef = new AtomicReference<>();

    private static AtomicReference<ConcurrentHashMap<String, BrowserEnum>> mapPartRef = new AtomicReference<>();

    /**
     * 根据code获取 BrowserEnum
     *
     * @param code
     * @return
     */
    public static BrowserEnum fromCode(String code) {
        buildConcurrentHashMap();
        return mapAllRef.get().get(code);
    }

    /**
     * 如果map是空,就会构建Map
     */
    private static void buildConcurrentHashMap() {
        //如果所有枚举的原子引用是空对象,就创建一个ConcurrentHashMap给原子引用
        mapAllRef.compareAndSet(null, new ConcurrentHashMap<>());
        //获取原子引用的currentHashMap对象
        ConcurrentHashMap<String, BrowserEnum> concurrentHashAllMap = mapAllRef.get();
        //如果ConcurrentHashMap里面没有一个枚举对象,需要初次化
        if (concurrentHashAllMap.isEmpty()) {
            for (BrowserEnum e : BrowserEnum.values()) {
                concurrentHashAllMap.put(e.getCode(), e);
            }
            mapAllRef.set(concurrentHashAllMap);
        }
    }

    /**
     * 枚举代码
     */
    private String code;

    /**
     * 浏览器驱动名字
     */
    private String driver;

    /**
     * exe 名字
     */
    private String exeName;

    public String getCode() {
        return code;
    }

    public void setCode(String code) {
        this.code = code;
    }

    public String getDriver() {
        return driver;
    }

    public void setDriver(String driver) {
        this.driver = driver;
    }

    public String getExeName() {
        return exeName;
    }

    public void setExeName(String exeName) {
        this.exeName = exeName;
    }

    BrowserEnum(String code,String exeName, String driver) {
        this.code = code;
        this.driver = driver;
        this.exeName = exeName;
    }

    public static List<BrowserEnum> getAllEnum() {
        buildConcurrentHashMap();
        return (List<BrowserEnum>) mapAllRef.get().values();
    }

    /**
     * 只返回常用枚举
     *
     * @return
     */
    public static List<BrowserEnum> getEnumByDefault() {
        buildConcurrentHashMap();
        return (List<BrowserEnum>) mapPartRef.get().values();
    }
}