添加版權信息生成的Java代碼-Jcodemodel

Question

我正在使用JCodeModel生成Java源代碼。 我將版權信息添加到生成的代碼中。 目前可以嗎？

我嘗試在JDefinedClass中使用javadoc() ，它僅在類定義上方添加信息。

Answer 1

您可以創建一個寫入版權標頭的CodeWriter 。 該CodeWriter可以委派給另一個代碼，即通常傳遞給CodeModel#build方法的代碼。

一個完整的例子：

import java.io.IOException;
import java.io.OutputStream;

import com.sun.codemodel.CodeWriter;
import com.sun.codemodel.JCodeModel;
import com.sun.codemodel.JPackage;
import com.sun.codemodel.writer.SingleStreamCodeWriter;

public class HeaderInCodeModel
{
    public static void main(String[] args) throws Exception
    {
        JCodeModel codeModel = new JCodeModel();
        codeModel._class("com.example.Example");

        CodeWriter codeWriter = new SingleStreamCodeWriter(System.out);

        String header = "// Copyright 2017 - example.com\n";
        CodeWriter codeWriterWithHeader = 
            createCodeWriterWithHeader(header, codeWriter);
        codeModel.build(codeWriterWithHeader);
    }    

    private static CodeWriter createCodeWriterWithHeader(
        String header, CodeWriter delegate)
    {
        CodeWriter codeWriter = new CodeWriter()
        {
            @Override
            public OutputStream openBinary(JPackage pkg, String fileName)
                throws IOException
            {
                OutputStream result = delegate.openBinary(pkg, fileName);
                if (header != null)
                {
                    result.write(header.getBytes());
                }
                return result;
            }

            @Override
            public void close() throws IOException
            {
                delegate.close();
            }
        };
        return codeWriter;
    }

}

結果類將是

// Copyright 2017 - example.com

package com.example;


public class Example {


}

Answer 2

com.sun.codemodel.writer.PrologCodeWriter正是您要找的東西