import org.mybatis.generator.internal.DefaultCommentGenerator; //导入依赖的package包/类
/**
 * Set the context under which this plugin is running.
 *
 * @param context
 *            the new context
 */
@Override
public void setContext(Context context) {
    super.setContext(context);

    // 配置插件使用的模板引擎
    PluginConfiguration cfg = PluginTools.getPluginConfiguration(context, CommentPlugin.class);

    if (cfg == null || cfg.getProperty(CommentPlugin.PRO_TEMPLATE) == null){
        if (context.getCommentGenerator() instanceof DefaultCommentGenerator){
            // 使用默认模板引擎
            commentGenerator = new TemplateCommentGenerator("default-comment.ftl", true);
        } else {
            // 用户自定义
            commentGenerator = context.getCommentGenerator();
        }
    } else {
        TemplateCommentGenerator templateCommentGenerator = new TemplateCommentGenerator(cfg.getProperty(CommentPlugin.PRO_TEMPLATE), false);

        // ITFSW 插件使用的注释生成器
        commentGenerator = templateCommentGenerator;

        // 修正系统插件
        try {
            // 先执行一次生成CommentGenerator操作，然后再替换
            context.getCommentGenerator();

            Field field = Context.class.getDeclaredField("commentGenerator");
            field.setAccessible(true);
            field.set(context, templateCommentGenerator);
        } catch (Exception e) {
            logger.error("反射异常",e);
        }
    }
}