miaoyachun

/**
 * 重載合并屬性實現(xiàn)
 * 先加載file properties， 然后并入ZK配置中心讀取的properties
 *
 * @return 合并后的屬性集合
 * @throws IOException 異常
 */
@Override
protected Properties mergeProperties() throws IOException {
    Properties result = new Properties();
    // 加載父類的配置
    Properties mergeProperties = super.mergeProperties();
    result.putAll(mergeProperties);
    // 加載從zk中讀取到的配置
    Map<String, String> configs = loadZkConfigs();
    result.putAll(configs);
    return result;
}

/**
 * 重載處理屬性實現(xiàn)
 * 根據(jù)選項, 決定是否將合并后的props寫入系統(tǒng)屬性, Spring boot需要
 *
 * @param beanFactoryToProcess
 * @param props                合并后的屬性
 * @throws BeansException
 */
@Override
protected void processProperties(ConfigurableListableBeanFactory beanFactoryToProcess, Properties props) throws BeansException {
    // 原有邏輯
     super.processProperties(beanFactoryToProcess, props);
    // 寫入到系統(tǒng)屬性
    if (writePropsToSystem) {
       // write all properties to system for spring boot
        Enumeration<?> propertyNames = props.propertyNames();
        while (propertyNames.hasMoreElements()) {
             String propertyName = (String) propertyNames.nextElement();
             String propertyValue = props.getProperty(propertyName);
             System.setProperty(propertyName, propertyValue);
        }
    }
}

@Override
public Object postProcessBeforeInitialization(Object bean, String beanName)
      throws BeansException {
   ConfigurationProperties annotation = AnnotationUtils
         .findAnnotation(bean.getClass(), ConfigurationProperties.class);
   if (annotation != null) {
      postProcessBeforeInitialization(bean, beanName, annotation);
 }
   annotation = this.beans.findFactoryAnnotation(beanName,
 ConfigurationProperties.class);
   if (annotation != null) {
      postProcessBeforeInitialization(bean, beanName, annotation);
 }
   return bean;
}

解決思路有兩個:

• 思路1: 想辦法對靜態(tài)變量注入, 也就是繞過Spring只能運行非靜態(tài)變量才能注入依賴的壁壘
• 思路2: 想辦法@AfterClass改造為非靜態(tài)

  • 實現(xiàn)Junit RunListener, 覆蓋testRunFinished方法, 這里去實現(xiàn)類似@AfterClass的功能, 這個方法是非靜態(tài)的
  • 不要用Junit, 改用TestNG, TestNG里的AfterClass是非靜態(tài)的
  • 用Spring的TestExecutionListeners, 實現(xiàn)個Listener, 里面也有個類似非靜態(tài)的AfterClass的實現(xiàn), 覆蓋實現(xiàn)就行

思路2的幾個方法都可以實現(xiàn), 但是單元測試Runner需要用

@RunWith(Theories.class)

而且改用TestNG工程浩大, 只能放棄掉這個思路

繼續(xù)走思路1, 只能去繞過Spring的依賴注入的static壁壘了, 具體代碼如下:

 @Autowired
 private Destination dfsOperationQueue;
 private static Destination dfsOperationQueueStatic; // static version
 @Autowired
 private MessageQueueAPI messageQueueAPI;
 private static MessageQueueAPI messageQueueAPIStatic; // static version


 @PostConstruct
 public void init() {
        dfsOperationQueueStatic = this.dfsOperationQueue;
        messageQueueAPIStatic = this.messageQueueAPI;
 }

 @AfterClass
 public static void afterClass() {
        MessageVO messageVO = messageQueueAPIStatic.removeDestination(dfsOperationQueueStatic);
        System.out.println(messageVO);
 }

其實就是用了@PostConstruct 來個偷梁換柱而已, 多聲明個靜態(tài)成員指向非靜態(tài)對象, 兩者其實是一個對象