001package org.apache.maven.wagon.tck.http.fixture; 002 003/* 004 * Licensed to the Apache Software Foundation (ASF) under one 005 * or more contributor license agreements. See the NOTICE file 006 * distributed with this work for additional information 007 * regarding copyright ownership. The ASF licenses this file 008 * to you under the Apache License, Version 2.0 (the 009 * "License"); you may not use this file except in compliance 010 * with the License. You may obtain a copy of the License at 011 * 012 * http://www.apache.org/licenses/LICENSE-2.0 013 * 014 * Unless required by applicable law or agreed to in writing, 015 * software distributed under the License is distributed on an 016 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 017 * KIND, either express or implied. See the License for the 018 * specific language governing permissions and limitations 019 * under the License. 020 */ 021 022import javax.servlet.ServletException; 023import javax.servlet.http.HttpServlet; 024import javax.servlet.http.HttpServletRequest; 025import javax.servlet.http.HttpServletResponse; 026import java.io.IOException; 027 028public class RedirectionServlet 029 extends HttpServlet 030{ 031 032 private static final long serialVersionUID = 1L; 033 034 private final String targetPath; 035 036 private final int code; 037 038 private final int maxRedirects; 039 040 private int redirectCount = 0; 041 042 private final String myPath; 043 044 public RedirectionServlet( final int code, final String targetPath ) 045 { 046 this.code = code; 047 this.targetPath = targetPath; 048 this.maxRedirects = 1; 049 this.myPath = null; 050 } 051 052 public RedirectionServlet( final int code, final String myPath, final String targetPath, final int maxRedirects ) 053 { 054 this.code = code; 055 this.myPath = myPath; 056 this.targetPath = targetPath; 057 this.maxRedirects = maxRedirects; 058 } 059 060 public int getRedirectCount() 061 { 062 return redirectCount; 063 } 064 065 @Override 066 protected void service( final HttpServletRequest req, final HttpServletResponse resp ) 067 throws ServletException, IOException 068 { 069 redirectCount++; 070 071 if ( myPath == null ) 072 { 073 resp.setStatus( code ); 074 resp.setHeader( "Location", targetPath ); 075 } 076 else if ( maxRedirects < 0 ) 077 { 078 resp.setStatus( code ); 079 resp.setHeader( "Location", myPath ); 080 } 081 else if ( redirectCount <= maxRedirects ) 082 { 083 resp.setStatus( code ); 084 resp.setHeader( "Location", myPath + "/" + redirectCount ); 085 } 086 else 087 { 088 resp.setStatus( code ); 089 resp.setHeader( "Location", targetPath ); 090 } 091 } 092 093}