auth/auth-cass/src/main/java/org/onap/aaf/auth/direct/DirectAAFLocator.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.direct;
  23
  24 import java.net.URI;
  25 import java.net.URISyntaxException;
  26 import java.util.Collections;
  27 import java.util.LinkedList;
  28 import java.util.List;
  29
  30 import org.onap.aaf.auth.dao.cass.LocateDAO;
  31 import org.onap.aaf.auth.dao.cass.LocateDAO.Data;
  32 import org.onap.aaf.auth.env.AuthzEnv;
  33 import org.onap.aaf.auth.env.AuthzTrans;
  34 import org.onap.aaf.auth.layer.Result;
  35 import org.onap.aaf.cadi.LocatorException;
  36 import org.onap.aaf.cadi.Access.Level;
  37 import org.onap.aaf.cadi.aaf.v2_0.AbsAAFLocator;
  38 import org.onap.aaf.cadi.config.Config;
  39 import org.onap.aaf.misc.env.util.Split;
  40
  41 import locate.v1_0.Endpoint;
  42
  43 public class DirectAAFLocator extends AbsAAFLocator<AuthzTrans> {
  44     private LocateDAO ldao;
  45     private int major=-1, minor=-1, patch=-1, pkg=-1;
  46     private AuthzEnv env;
  47     private final URI uri;
  48
  49     /**
  50      *
  51      * @param env
  52      * @param ldao
  53      * @param key  must be one or more of service, version, other in that order
  54      * @throws LocatorException
  55      */
  56     public DirectAAFLocator(AuthzEnv env, LocateDAO ldao, String name, String version) throws LocatorException {
  57         super(env.access(), name, 1000L /* Don't hit DB more than once a second */);
  58         this.env = env;
  59         this.ldao = ldao;
  60         if (version!=null) {
  61             try {
  62                 String[] v = Split.split('.',version);
  63                 if (v.length>0) {major = Integer.parseInt(v[0]);}
  64                 if (v.length>1) {minor = Integer.parseInt(v[1]);}
  65                 if (v.length>2) {patch = Integer.parseInt(v[2]);}
  66                 if (v.length>3) {pkg   = Integer.parseInt(v[3]);}
  67             } catch (NumberFormatException e) {
  68                 throw new LocatorException("Invalid Version String: " + version);
  69             }
  70         }
  71
  72         try {
  73             uri = new URI(access.getProperty(Config.AAF_LOCATE_URL, "localhost")+"/locate/"+name+':'+version);
  74         } catch (URISyntaxException e) {
  75             throw new LocatorException(e);
  76         }
  77         myhostname=null;
  78         myport = 0;
  79     }
  80
  81
  82     @Override
  83     public boolean refresh() {
  84         AuthzTrans trans = env.newTransNoAvg();
  85         Result<List<Data>> rl = ldao.readByName(trans, name);
  86         if (rl.isOK()) {
  87             LinkedList<EP> epl = new LinkedList<>();
  88             for (Data d : rl.value) {
  89 //                if (myhostname!=null && d.port==myport && d.hostname.equals(myhostname)) {
  90 //                    continue;
  91 //                }
  92                 if ((major<0 || major==d.major) &&
  93                    (minor<0 || minor<=d.minor) &&
  94                    (patch<0 || patch==d.patch) &&
  95                    (pkg<0   || pkg  ==d.pkg)) {
  96                     Endpoint endpoint = new Endpoint();
  97                     endpoint.setName(d.name);
  98                     endpoint.setHostname(d.hostname);
  99                     endpoint.setPort(d.port);
 100                     endpoint.setMajor(d.major);
 101                     endpoint.setMinor(d.minor);
 102                     endpoint.setPatch(d.patch);
 103                     endpoint.setPkg(d.pkg);
 104                     endpoint.setLatitude(d.latitude);
 105                     endpoint.setLongitude(d.longitude);
 106                     endpoint.setProtocol(d.protocol);
 107                     for (String s : d.subprotocol(false)) {
 108                         endpoint.getSubprotocol().add(s);
 109                     }
 110
 111                     try {
 112                         epl.add(new EP(endpoint,latitude,longitude));
 113                     } catch (URISyntaxException e) {
 114                         e.printStackTrace();
 115                     }
 116                 }
 117             }
 118             Collections.sort(epl);
 119             replace(epl);
 120             return true;
 121         } else {
 122             access.log(Level.ERROR, rl.errorString());
 123         }
 124         return false;
 125     }
 126
 127     @Override
 128     protected URI getURI() {
 129         return uri;
 130     }
 131
 132 }