Извлечь изображение из сервлета в flyingsaucer

Я использую flyingsaucer для рендеринга документа xhtml в pdf через сервлет, который возвращает сгенерированный документ PDF. Документ xhtml имеет изображение, которое запрашивается у другого сервлета. Сервлет образа проверяет, кто вошел в систему, прежде чем возвращать соответствующее изображение. В приведенном ниже коде показано, как запрашивается изображение:

<img height="140" width="140" src="http://localhost:8080/myapp/servlet/DisplayPic" />

Моя проблема заключается в том, что http-запрос для изображения относится к рендерингу PDF, а не к зарегистрированному пользователю, поэтому сервлет изображения не знает, кто вошел в систему, и поэтому желаемое изображение не возвращается.

В настоящее время я использую следующий код для рендеринга документа xhtml:

ITextRenderer renderer = new ITextRenderer();
renderer.setDocumentFromString(xhtmlDocumentAsString);
renderer.layout();
os = response.getOutputStream();
renderer.createPDF(os);

Мне нужно либо поддерживать сеанс пользователя, когда запрашивается сервлет изображения, либо предоставить рендереру изображение, используемое для этого конкретного элемента xhtml. Я думаю, что последнее можно сделать с помощью ReplacedElementFactory, но я не смог выкопать код примера, который может мне помочь.

Ответ 1

У меня сейчас это очень хорошо. Вот код.

В моем документе xhtml у меня есть:

<div class="profile_picture" style="display:block;width:140px;height:140px;" />

(Я использую элемент div вместо img, поскольку factory используется только для элементов уровня блока)

Я переношу свой документ, используя:

ITextRenderer renderer = new ITextRenderer();
renderer.getSharedContext().setReplacedElementFactory(new ProfileImageReplacedElementFactory(renderer.getSharedContext().getReplacedElementFactory()));
renderer.setDocumentFromString(xhtmlDocumentAsString);
renderer.layout();
os = response.getOutputStream();
renderer.createPDF(os);

И у меня есть свой ReplacedElementFactory, как показано ниже:

public class ProfileImageReplacedElementFactory implements ReplacedElementFactory {

    private final ReplacedElementFactory superFactory;

    public ProfileImageReplacedElementFactory(ReplacedElementFactory superFactory) {
        this.superFactory = superFactory;
    }

    @Override
    public ReplacedElement createReplacedElement(LayoutContext layoutContext, BlockBox blockBox,
            UserAgentCallback userAgentCallback, int cssWidth, int cssHeight) {

        Element element = blockBox.getElement();
        if (element == null) {
            return null;
        }

        String nodeName = element.getNodeName();
        String className = element.getAttribute("class");
        if ("div".equals(nodeName) && className.contains("profile_picture")) {

            InputStream input = null;
            try {
                input = ...;
                byte[] bytes = IOUtils.toByteArray(input);
                Image image = Image.getInstance(bytes);
                FSImage fsImage = new ITextFSImage(image);

                if (fsImage != null) {
                    if ((cssWidth != -1) || (cssHeight != -1)) {
                        fsImage.scale(cssWidth, cssHeight);
                    }
                    return new ITextImageElement(fsImage);
                }
            } catch (IOException e) {
                getLogger().error(ExceptionUtils.getStackTrace(e));
            } catch (BadElementException e) {
                getLogger().error(ExceptionUtils.getStackTrace(e));
            } finally {
                IOUtils.closeQuietly(input);
            }
        }

        return superFactory.createReplacedElement(layoutContext, blockBox, userAgentCallback, cssWidth, cssHeight);
    }

    @Override
    public void reset() {
        superFactory.reset();
    }

    @Override
    public void remove(Element e) {
        superFactory.remove(e);
    }

    @Override
    public void setFormSubmissionListener(FormSubmissionListener listener) {
        superFactory.setFormSubmissionListener(listener);
    }
}