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