summaryrefslogtreecommitdiffstats
path: root/auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/NsAttrib.java
blob: eafbe909bcd42220b67482627abc95c1657ce045 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/**
 * ============LICENSE_START====================================================
 * org.onap.aaf
 * ===========================================================================
 * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
 * ===========================================================================
 * Licensed 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.
 * ============LICENSE_END====================================================
 *
 */

package org.onap.aaf.auth.helpers;

import java.util.ArrayList;
import java.util.List;
import java.util.SortedMap;
import java.util.TreeMap;

import org.onap.aaf.misc.env.Env;
import org.onap.aaf.misc.env.TimeTaken;
import org.onap.aaf.misc.env.Trans;

import com.datastax.driver.core.ResultSet;
import com.datastax.driver.core.Row;
import com.datastax.driver.core.Session;
import com.datastax.driver.core.SimpleStatement;
import com.datastax.driver.core.Statement;

public class NsAttrib  {
	public static final List<NsAttrib> data = new ArrayList<>();
    public static final SortedMap<String,List<NsAttrib>> byKey = new TreeMap<>();
    public static final SortedMap<String,List<NsAttrib>> byNS = new TreeMap<>();

	public final String ns;
	public final String key;
	public final String value;
	public static Creator<NsAttrib> v2_0_11 = new Creator<NsAttrib>() {
		@Override
		public NsAttrib create(Row row) {
			return new NsAttrib(row.getString(0), row.getString(1), row.getString(2));
		}

		@Override
		public String select() {
			return "select ns,key,value from authz.ns_attrib";
		}
	};
	
	public NsAttrib(String ns, String key, String value) {
		this.ns = ns;
		this.key = key;
		this.value = value;
	}
	
	public static void load(Trans trans, Session session, Creator<NsAttrib> creator ) {
		trans.info().log( "query: " + creator.select() );
        ResultSet results;
        TimeTaken tt = trans.start("Load NsAttributes", Env.REMOTE);
		try {
	        Statement stmt = new SimpleStatement(creator.select());
	        results = session.execute(stmt);
        } finally {
        	tt.done();
        }
		int count = 0;
        tt = trans.start("Process NsAttributes", Env.SUB);

        try {
        	for(Row row : results.all()) {
        		++count;
	        	NsAttrib ur = creator.create(row);
	        	data.add(ur);
	        	
	        	List<NsAttrib> lna = byKey.get(ur.key);
	        	if(lna==null) {
	        		lna = new ArrayList<>();
		        	byKey.put(ur.key, lna);
	        	}
	        	lna.add(ur);
	        	
	        	lna = byNS.get(ur.ns);
	        	if(lna==null) {
	        		lna = new ArrayList<>();
		        	byNS.put(ur.ns, lna);
	        	}
	        	lna.add(ur);
        	}
        } finally {
        	tt.done();
        	trans.info().log("Found",count,"NS Attributes");
        }
	}

	public String toString() {
		return '"' + ns + "\",\"" + key + "\",\""  + value +'"';
	}

}