kezhenxu94
3 years ago
committed by
GitHub
5 changed files with 52 additions and 7 deletions
@ -0,0 +1,50 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one |
||||
* or more contributor license agreements. See the NOTICE file |
||||
* distributed with this work for additional information |
||||
* regarding copyright ownership. The ASF licenses this file |
||||
* to you 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. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.api.controller; |
||||
|
||||
import javax.servlet.RequestDispatcher; |
||||
import javax.servlet.http.HttpServletRequest; |
||||
import org.springframework.boot.web.servlet.error.ErrorController; |
||||
import org.springframework.http.HttpStatus; |
||||
import org.springframework.stereotype.Controller; |
||||
import org.springframework.web.bind.annotation.RequestMapping; |
||||
import org.springframework.web.servlet.ModelAndView; |
||||
|
||||
@Controller |
||||
public class DsErrorController implements ErrorController { |
||||
@RequestMapping("/error") |
||||
public ModelAndView handleError(HttpServletRequest request) { |
||||
Object status = request.getAttribute(RequestDispatcher.ERROR_STATUS_CODE); |
||||
ModelAndView modelAndView = new ModelAndView(); |
||||
|
||||
if (status != null) { |
||||
Integer statusCode = Integer.valueOf(status.toString()); |
||||
|
||||
if (statusCode == HttpStatus.NOT_FOUND.value()) { |
||||
modelAndView.setStatus(HttpStatus.OK); |
||||
modelAndView.setViewName("forward:/ui/index.html"); |
||||
return modelAndView; |
||||
} |
||||
|
||||
modelAndView.setStatus(HttpStatus.valueOf(statusCode)); |
||||
} |
||||
return modelAndView; |
||||
} |
||||
} |
Loading…
Reference in new issue