侧边栏壁纸
博主头像
分享你我博主等级

行动起来,活在当下

  • 累计撰写 108 篇文章
  • 累计创建 13 个标签
  • 累计收到 0 条评论

目 录CONTENT

文章目录

C# 使用NVelocity 生成代码

管理员
2020-09-22 / 0 评论 / 0 点赞 / 3 阅读 / 12451 字

首先封装一下NVelocity

新建 VelocityHelper.cs

    /// <summary>
    /// NVelocity模板工具类 VelocityHelper
    /// </summary>
    public class VelocityHelper
    {
        private VelocityEngine velocity = null;
        private IContext context = null;
        /// <summary>
        /// 构造函数
        /// </summary>
        /// <param name="templatDir">模板文件夹路径</param>
        public VelocityHelper(string templatDir)
        {
            Init(templatDir);
        }
        /// <summary>
        /// 无参数构造函数
        /// </summary>
        public VelocityHelper() { }
        /// <summary>
        /// 初始话NVelocity模块
        /// </summary>
        public void Init(string templatDir)
        {
            //创建VelocityEngine实例对象
            velocity = new VelocityEngine();
            //使用设置初始化VelocityEngine
            ExtendedProperties props = new ExtendedProperties();
            props.AddProperty(RuntimeConstants.RESOURCE_LOADER, "file");
            props.AddProperty(RuntimeConstants.FILE_RESOURCE_LOADER_PATH, templatDir);
            props.AddProperty(RuntimeConstants.INPUT_ENCODING, "utf-8");
            props.AddProperty(RuntimeConstants.OUTPUT_ENCODING, "utf-8");
            //模板的缓存设置
            props.AddProperty(RuntimeConstants.FILE_RESOURCE_LOADER_CACHE, true);              //是否缓存
            props.AddProperty("file.resource.loader.modificationCheckInterval", (Int64)30);    //缓存时间(秒)
            velocity.Init(props);
            //为模板变量赋值
            context = new VelocityContext();
        }
        /// <summary>
        /// 给模板变量赋值
        /// </summary>
        /// <param name="key">模板变量</param>
        /// <param name="value">模板变量值</param>
        public void Add(string key, object value)
        {
            if (context == null)
                context = new VelocityContext();
            context.Put(key, value);
        }
        /// <summary>
        /// 显示模板
        /// </summary>
        /// <param name="templatFileName">模板文件名</param>
        public string Display(string templatFileName)
        {
            //从文件中读取模板
            Template template = velocity.GetTemplate(templatFileName);
            //合并模板
            using (StringWriter writer = new StringWriter())
            {
                template.Merge(context, writer);
                var str = writer.ToString();
                writer.Flush();
                writer.Close();
                //输出
                return str;
            }
        }
        /// <summary>
        /// 根据模板生成文件
        /// </summary>
        /// <param name="templatFileName"></param>
        /// <param name="path">保存路径</param>
        public void CreateFile(string templatFileName, string path, string fileName)
        {
            //从文件中读取模板
            Template template = velocity.GetTemplate(templatFileName);
            //合并模板
            StringWriter writer = new StringWriter();
            template.Merge(context, writer);
            if (!Directory.Exists(path))
            {
                Directory.CreateDirectory(path);
            }
            File.WriteAllText(path + "\\" + fileName, writer.ToString(), new System.Text.UTF8Encoding(false));
            writer.Write(writer);
            writer.Flush();
            writer.Close();
        }
    }


然后使用

 var tool = new Tool();
 VelocityHelper vh = new VelocityHelper();
                    vh.Init(path + "/template/");//模板路径
                    vh.Add("table", item);
                    vh.Add("tools", tool);
                    vh.Add("tableColumns", context._db.DbMaintenance.GetColumnInfosByTableName(item.Name));
                    vh.Add("namespace", "Com.Javalc.Bolg");
                    vh.CreateFile("entity.vm", "E:\\hisWork\\Com.Javalc.Bolg/Com.Javalc.Bolg.Entity/", tool.ReplaceFirst(tool.FirstUpperCase(item.Name),"Tb_").Replace("_","")+ ".cs");
                    vh.CreateFile("dao.vm", "E:\\hisWork\\Com.Javalc.Bolg/Com.Javalc.Bolg.Respository/ModelRepository/", tool.ReplaceFirst(tool.FirstUpperCase(item.Name), "Tb_").Replace("_", "") + "Repository.cs");

示例模版:

using System;
using Newtonsoft.Json;
using SqlSugar;
namespace ${namespace}.Entity
{
    #if( "${table.Description}" != "" )
 	    /// <summary>
        /// ${table.Description}
        /// </summary>
    #end
    [Serializable]
    [SugarTable("${table.Name}")]    //对应数据库的ORM_School表
    public partial class $tools.ReplaceFirst($tools.FirstUpperCase($table.Name),"Tb_")
    {
     #foreach($column in $tableColumns)
     /// <summary>
     #if( "${column.ColumnDescription}" != "" )  
     /// $tools.StrTrim($column.ColumnDescription)
     #end
     /// </summary>
        [JsonProperty("$tools.StrFirstUpperCase(${column.DbColumnName})")]
        [SugarColumn(ColumnName ="${column.DbColumnName}")]  
        public $tools.ConvertDataType($column.DataType) $tools.StrFirstUpperCase(${column.DbColumnName}) { get; set; }
        
     #end
    }
}


0

评论区