Map<String, TaskFactory> taskRegistry = new HashMap<String, TaskFactory>();
            taskRegistry.put(CommandTask.NAME, new TaskFactory() {
                @Override
                public Task createNewTask(TaskCallbackContext context) {
                    return new CommandTask(context);
                }
            });
            taskRegistry.put(DataCollectingTask.NAME, new TaskFactory() {
                @Override
                public Task createNewTask(TaskCallbackContext context) {
                    return new DataCollectingTask(context);
                }
            });
            taskRegistry.put(DataPushingTask.NAME, new TaskFactory() {
                @Override
                public Task createNewTask(TaskCallbackContext context) {
                    return new DataPushingTask(context);
                }
            });

            machineEngine.registerStateModelFactory("Task", new TaskStateModelFactory(zkHelixManager, taskRegistry));