auth/auth-core/src/main/java/org/onap/aaf/auth/rserv/RServlet.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.auth.rserv;
  23
  24 import java.io.IOException;
  25 import java.util.List;
  26
  27 import javax.servlet.Servlet;
  28 import javax.servlet.ServletConfig;
  29 import javax.servlet.ServletException;
  30 import javax.servlet.ServletRequest;
  31 import javax.servlet.ServletResponse;
  32 import javax.servlet.http.HttpServletRequest;
  33 import javax.servlet.http.HttpServletResponse;
  34
  35 import org.onap.aaf.misc.env.Env;
  36 import org.onap.aaf.misc.env.TimeTaken;
  37 import org.onap.aaf.misc.env.Trans;
  38
  39 public abstract class RServlet<TRANS extends Trans> implements Servlet {
  40         private Routes<TRANS> routes = new Routes<TRANS>();
  41
  42         private ServletConfig config;
  43
  44         @Override
  45         public void init(ServletConfig config) throws ServletException {
  46                 this.config = config;
  47         }
  48
  49         @Override
  50         public ServletConfig getServletConfig() {
  51                 return config;
  52         }
  53
  54         public void route(Env env, HttpMethods meth, String path, HttpCode<TRANS, ?> code, String ... moreTypes) {
  55                 Route<TRANS> r = routes.findOrCreate(meth,path);
  56                 r.add(code,moreTypes);
  57                 env.init().log(r.report(code),code);
  58         }
  59
  60         @Override
  61         public void service(ServletRequest req, ServletResponse res) throws ServletException, IOException {
  62                 HttpServletRequest request = (HttpServletRequest)req;
  63                 HttpServletResponse response = (HttpServletResponse)res;
  64
  65                 @SuppressWarnings("unchecked")
  66                 TRANS trans = (TRANS)req.getAttribute(TransFilter.TRANS_TAG);
  67                 if(trans==null) {
  68                         response.setStatus(404); // Not Found, because it didn't go through TransFilter
  69                         return;
  70                 }
  71
  72                 Route<TRANS> route;
  73                 HttpCode<TRANS,?> code=null;
  74                 String ct = req.getContentType();
  75                 TimeTaken tt = trans.start("Resolve to Code", Env.SUB);
  76                 try {
  77                         // routes have multiple code sets.  This object picks the best code set
  78                         // based on Accept or Content-Type
  79                         CodeSetter<TRANS> codesetter = new CodeSetter<TRANS>(trans,request,response);
  80                         // Find declared route
  81                         route = routes.derive(request, codesetter);
  82                         if(route==null) {
  83                                 String method = request.getMethod();
  84                                 trans.checkpoint("No Route matches "+ method + ' ' + request.getPathInfo());
  85                                 response.setStatus(404); // Not Found
  86                         } else {
  87                                 // Find best Code in Route based on "Accepts (Get) or Content-Type" (if exists)
  88                                 code = codesetter.code();// route.getCode(trans, request, response);
  89                         }
  90                 } finally {
  91                         tt.done();
  92                 }
  93
  94                 if(route!=null && code!=null) {
  95                         StringBuilder sb = new StringBuilder(72);
  96                         sb.append(route.auditText);
  97                         sb.append(',');
  98                         sb.append(code.desc());
  99                         if(ct!=null) {
 100                                 sb.append(", ContentType: ");
 101                                 sb.append(ct);
 102                         }
 103                         tt = trans.start(sb.toString(),Env.SUB);
 104                         try {
 105                                 /*obj = */
 106                                 code.handle(trans, request, response);
 107                                 response.flushBuffer();
 108                         } catch (ServletException e) {
 109                                 trans.error().log(e);
 110                                 throw e;
 111                         } catch (Exception e) {
 112                                 trans.error().log(e,request.getMethod(),request.getPathInfo());
 113                                 throw new ServletException(e);
 114                         } finally {
 115                                 tt.done();
 116                         }
 117                 }
 118         }
 119
 120         @Override
 121         public String getServletInfo() {
 122                 return "RServlet for Jetty";
 123         }
 124
 125         @Override
 126         public void destroy() {
 127         }
 128
 129         public String applicationJSON(Class<?> cls, String version) {
 130                 StringBuilder sb = new StringBuilder();
 131                 sb.append("application/");
 132                 sb.append(cls.getSimpleName());
 133                 sb.append("+json");
 134                 sb.append(";charset=utf-8");
 135                 sb.append(";version=");
 136                 sb.append(version);
 137                 return sb.toString();
 138         }
 139
 140         public String applicationXML(Class<?> cls, String version) {
 141                 StringBuilder sb = new StringBuilder();
 142                 sb.append("application/");
 143                 sb.append(cls.getSimpleName());
 144                 sb.append("+xml");
 145                 sb.append(";charset=utf-8");
 146                 sb.append(";version=");
 147                 sb.append(version);
 148                 return sb.toString();
 149         }
 150
 151         public List<RouteReport> routeReport() {
 152                 return routes.routeReport();
 153         }
 154 }