ZStack源碼剖析之設計模式鑑賞——策略模式

本文首發於泊浮目的專欄： https://segmentfault.com/blog...

前言

不管什麼程序，其目的都是解決問題。而爲了解決問題，咱們又須要編寫特定的算法。使用Strategy模式能夠總體地替換算法的實現部分。可以總體地替換算法，能讓咱們輕鬆地以不一樣的算法去解決一個問題，這種模式就是Strategy模式。

在ZStack中，Strategy模式幾乎是充斥在80%以上的代碼中的，接下來咱們就來一塊兒看看吧。

CollectionUtils

CollectionUtils 這個類是在JDK8發佈前常在ZStack中被使用的一個類，代碼以下：

package org.zstack.utils;

import org.zstack.utils.function.ForEachFunction;
import org.zstack.utils.function.Function;
import org.zstack.utils.function.ListFunction;
import org.zstack.utils.logging.CLogger;

import java.util.*;

/**
 */
public class CollectionUtils {
    private static final CLogger logger = Utils.getLogger(CollectionUtils.class);

    public static <K, V> List<K> transformToList(Collection<V> from, ListFunction<K, V> func) {
        List<K> ret = new ArrayList<K>();
        for (V v : from) {
            List<K> k = func.call(v);
            if (k == null) {
                continue;
            }
            ret.addAll(k);
        }

        return ret;
    }

    public static <K, V> List<K> transformToList(Collection<V> from, Function<K, V> func) {
        List<K> ret = new ArrayList<K>();
        for (V v : from) {
            K k = func.call(v);
            if (k == null) {
                continue;
            }
            ret.add(k);
        }

        return ret;
    }

    public static <K, V> Set<K> transformToSet(Collection<V> from, Function<K, V> func) {
        Set<K> ret = new HashSet<K>();
        for (V v : from) {
            K k = func.call(v);
            if (k == null) {
                continue;
            }
            ret.add(k);
        }

        return ret;
    }

    public static <K, V> Set<K> transformToSet(Collection<V> from, ListFunction<K, V> func) {
        Set<K> ret = new HashSet<K>();
        for (V v : from) {
            List<K> k = func.call(v);
            if (k == null) {
                continue;
            }
            ret.addAll(k);
        }

        return ret;
    }

    public static <K, V> K find(Collection<V> from, Function<K, V> func) {
        for (V v : from) {
            K k = func.call(v);
            if (k != null) {
                return k;
            }
        }

        return null;
    }

    public static <K> void forEach(Collection<K> cols, ForEachFunction<K> func) {
        for (K c : cols) {
            func.run(c);
        }
    }

    public static <K> void safeForEach(Collection<K> cols, ForEachFunction<K> func) {
        for (K c : cols) {
            try {
                func.run(c);
            } catch (Throwable t) {
                logger.warn(String.format("unhandled exception happened"), t);
            }
        }
    }

    public static <K> List<K> removeDuplicateFromList(List<K> lst) {
        return new ArrayList<K>(new LinkedHashSet<K>(lst));
    }
}

以 public static <K, V> List<K> transformToList(Collection<V> from, Function<K, V> func) 爲例，從語義上來講就是爲了from中的每一個元素調用func函數。沒錯，就像函數式編程中的map。

List<KVMHostAsyncHttpCallMsg> msgs = CollectionUtils.transformToList(hostUuids, new Function<KVMHostAsyncHttpCallMsg, String>() {
            @Override
            public KVMHostAsyncHttpCallMsg call(String huuid) {
                ScanCmd cmd = new ScanCmd();
                cmd.ip = getIpForScan(struct);
                cmd.startPort = 1;
                cmd.endPort = 65535;
                cmd.interval = struct.getInterval();
                cmd.times = struct.getMaxTimes();
                cmd.successInterval = struct.getSuccessInterval();
                cmd.successTimes = struct.getSuccessTimes();

                KVMHostAsyncHttpCallMsg msg = new KVMHostAsyncHttpCallMsg();
                msg.setHostUuid(huuid);
                msg.setPath(SCAN_HOST_PATH);
                msg.setCommandTimeout(timeoutManager.getTimeout(cmd.getClass(), TimeUnit.SECONDS.toMillis(cmd.interval *cmd.times) + TimeUnit.MINUTES.toMillis(1)));
                msg.setCommand(cmd);
                bus.makeTargetServiceIdByResourceUuid(msg, HostConstant.SERVICE_ID, huuid);

                return msg;
            }
        });

從這邊的代碼能夠看到，經過遍歷hostUuids並作了一些操做，成功的組成了一組msg。

Completion

在異步系統中，Completion是很常見的——當一個異步行爲完成時，則會調用其相應的CompletionHandle。

bus.send(amsg, new CloudBusCallBack(completion) {
            @Override
            public void run(MessageReply re) {
                if (!re.isSuccess()) {
                    completion.fail(re.getError());
                } else {
                    completion.success(re);
                }
            }
        });

以CloudBus的send調用爲例，當一個msg發送並獲得回覆後，便會執行傳進來CallBack的run。這樣的代碼靈活性很是高——簡單來講，傳入send這個函數的第二個參數是一個策略，而不是一個單純的參數。

CloudBus的源碼分析點擊這裏，有興趣的讀者能夠看一下其實現

小結

在本篇文章中，筆者和你們一塊兒瞭解了Strategy在ZStack中的使用場景。一般在編程時，算法（策略）會被寫在具體方法中，這樣會致使具體方法中充斥着條件判斷語句。可是Strategy卻特地將算法與其餘部分剝離開來，僅僅定義了接口，而後再以委託的方式來使用算法。然而這種作法正是讓程序更加的鬆耦合（由於使用委託能夠方便的總體替換算法），使得整個項目更加茁壯。