com.yahoo.glimmer.web.DocObjectView.java Source code

Introduction

Here is the source code for com.yahoo.glimmer.web.DocObjectView.java

Source

package com.yahoo.glimmer.web;

/*
 * Copyright (c) 2012 Yahoo! Inc. All rights reserved.
 * 
 *  Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
 *  Unless required by applicable law or agreed to in writing, software distributed under the License is 
 *  distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and limitations under the License.
 *  See accompanying LICENSE file.
 */

import java.io.PrintWriter;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.web.servlet.View;

public class DocObjectView implements View {
    private boolean includeUri;

    @Override
    public String getContentType() {
        // Setting charset= means that response.getWriter() will return a Writer
        // that writes in that charset.
        return "text/plain; charset=UTF-8";
    }

    @Override
    public void render(Map<String, ?> model, HttpServletRequest request, HttpServletResponse response)
            throws Exception {
        Object object = model.get(QueryController.OBJECT_KEY);
        if (object == null) {
            throw new IllegalArgumentException("Model does not contain an object!");
        }

        response.setContentType(getContentType());
        PrintWriter writer = response.getWriter();

        // Return URIs and docs separated by tab
        if (object instanceof QueryResult) {
            QueryResult result = (QueryResult) object;
            for (QueryResultItem item : result.getResultItems()) {
                if (includeUri) {
                    writer.println(item.getSubject());
                }
                writer.println(item.toString());
            }
        } else {
            writer.print(object.toString());
        }
    }

    public void setIncludeUri(boolean includeUri) {
        this.includeUri = includeUri;
    }
}