com.alibaba.rocketmq.store.MessageStore Java Examples

public final static MessageStore build(MessageStorePluginContext context, MessageStore messageStore)
        throws IOException {
    String plugin = context.getBrokerConfig().getMessageStorePlugIn();
    if (plugin != null && plugin.trim().length() != 0) {
        String[] pluginClasses = plugin.split(",");
        for (int i = pluginClasses.length - 1; i >= 0; --i) {
            String pluginClass = pluginClasses[i];
            try {
                @SuppressWarnings("unchecked")
                Class<AbstractPluginMessageStore> clazz = (Class<AbstractPluginMessageStore>) Class.forName(pluginClass);
                Constructor<AbstractPluginMessageStore> construct = clazz.getConstructor(MessageStorePluginContext.class, MessageStore.class);
                AbstractPluginMessageStore pluginMessageStore = (AbstractPluginMessageStore) construct.newInstance(context, messageStore);
                messageStore = pluginMessageStore;
            } catch (Throwable e) {
                throw new RuntimeException(String.format(
                        "Initialize plugin's class %s not found!", pluginClass), e);
            }
        }
    }
    return messageStore;
}
 

@Test
public void testMessageStoreFactory() throws IOException {
    BrokerConfig brokerConfig = new BrokerConfig();
    MessageStorePluginContext context = new MessageStorePluginContext(null, null, null, brokerConfig);
    MessageStore messageStore = null;
    AbstractPluginMessageStore m = null;
    //��plugin����
    brokerConfig.setMessageStorePlugIn(null);
    messageStore = MessageStoreFactory.build(context, new MockMessageStore());
    assertTrue(messageStore instanceof MockMessageStore);

    brokerConfig.setMessageStorePlugIn("");
    messageStore = MessageStoreFactory.build(context, new MockMessageStore());
    assertTrue(messageStore instanceof MockMessageStore);

    //��plugin����
    brokerConfig.setMessageStorePlugIn("com.alibaba.rocketmq.broker.plugin.MockMessageStorePlugin1,com.alibaba.rocketmq.broker.plugin.MockMessageStorePlugin2");
    messageStore = MessageStoreFactory.build(context, new MockMessageStore());
    assertTrue(messageStore instanceof MockMessageStorePlugin1);
    m = (AbstractPluginMessageStore) messageStore;

    assertTrue(m.next instanceof MockMessageStorePlugin2);
    m = (AbstractPluginMessageStore) m.next;
    assertTrue(m.next instanceof MockMessageStore);

    //�׳��쳣����
    brokerConfig.setMessageStorePlugIn("aaaaaa");
    try {
        messageStore = MessageStoreFactory.build(context, new MockMessageStore());
        assertTrue(false);
    } catch (RuntimeException e) {
    }

}
 

public MessageStore getMessageStore() {
    return messageStore;
}
 

public void setMessageStore(MessageStore messageStore) {
    this.messageStore = messageStore;
}
 

public MessageStore getMessageStore() {
    return messageStore;
}
 

public void setMessageStore(MessageStore messageStore) {
    this.messageStore = messageStore;
}
 

@Test
public void testAbstractPluginMessageStore() throws IllegalAccessException, IllegalArgumentException,
        InvocationTargetException {


    final ThreadLocal<String> invokingMethodName = new ThreadLocal<String>();

    MessageStore messageStore = (MessageStore) Proxy.newProxyInstance(
            MessageStore.class.getClassLoader(), new Class[]{MessageStore.class},
            new InvocationHandler() {
                @Override
                public Object invoke(Object proxy, Method method, Object[] args)
                        throws Throwable {
                    assertEquals(invokingMethodName.get(), method.getName());
                    if (method.getReturnType() == int.class) {
                        return Integer.valueOf(0);
                    } else if (method.getReturnType() == long.class) {
                        return Long.valueOf(0);
                    } else if (method.getReturnType() == char.class) {
                        return Byte.valueOf((byte) 0);
                    } else if (method.getReturnType() == byte.class) {
                        return Byte.valueOf((byte) 0);
                    } else if (method.getReturnType() == boolean.class) {
                        return true;
                    } else {
                        return null;
                    }
                }
            });

    AbstractPluginMessageStore pluginMessageStore = new AbstractPluginMessageStore(null, messageStore) {

        @Override
        public boolean isOSPageCacheBusy() {
            return false;
        }
    };

    Method[] methods = MessageStore.class.getMethods();
    for (Method m : methods) {
        Class[] paramType = m.getParameterTypes();
        Object[] mockParam = new Object[paramType.length];
        for (int i = 0; i < paramType.length; ++i) {
            if (paramType[i] == int.class) {
                mockParam[i] = Integer.valueOf(0);
            } else if (paramType[i] == long.class) {
                mockParam[i] = Long.valueOf(0);
            } else if (paramType[i] == byte.class) {
                mockParam[i] = Byte.valueOf((byte)0);
            } else if (paramType[i] == char.class) {
                mockParam[i] = Byte.valueOf((byte)0);
            } else
                mockParam[i] = null;
        }
        invokingMethodName.set(m.getName());
        m.invoke(pluginMessageStore, mockParam);
    }
}
 

public MessageStore getMessageStore() {
    return messageStore;
}
 

public void setMessageStore(MessageStore messageStore) {
    this.messageStore = messageStore;
}
 

/**
 * @param context
 * @param next
 */
public MockMessageStorePlugin1(MessageStorePluginContext context, MessageStore next) {
    super(context, next);
}
 

/**
 * @param context
 * @param next
 */
public MockMessageStorePlugin2(MessageStorePluginContext context, MessageStore next) {
    super(context, next);
}