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

前言

無論什么程序，其目的都是解決問題。而為了解決問題，我們又需要編寫特定的算法。使用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));
}

}</code></pre> 
  
以 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;
        }
    });</code></pre> 

從這邊的代碼可以看到，通過遍歷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。這樣的代碼靈活性非常高——簡單來說，傳入函數的是一個行為，而不是參數。
 
  
CloudBus的源碼分析點擊這里 ，有興趣的讀者可以看一下其實現
 
  
小結
 
  
在本片文章中，筆者和大家一起了解了Strategy在ZStack中的使用場景。通常在編程時，算法（策略）會被寫在具體方法中，這樣會導致具體方法中充斥著條件判斷語句。但是Strategy卻特意將算法與其他部分剝離開來，僅僅定義了接口，然后再以委托的方式來使用算法。然而這種做法正是讓程序更加的松耦合（因為使用委托可以方便的整體替換算法），使得整個項目更加茁壯。
 
  
 
 
  
來自：https://segmentfault.com/a/1190000013460437