aboutsummaryrefslogtreecommitdiffstats
path: root/appc-dispatcher/appc-workflow-management/appc-workflow-management-core/src/main/java/org/openecomp/appc/workflow/impl/WorkflowResolverDataReader.java
blob: 3ffeddcdfe9804fa102466b25e24b032b023eea8 (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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*-
 * ============LICENSE_START=======================================================
 * openECOMP : APP-C
 * ================================================================================
 * Copyright (C) 2017 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.openecomp.appc.workflow.impl;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;

import org.openecomp.appc.dao.util.DBUtils;
import org.openecomp.appc.rankingframework.AbstractRankedAttributesResolverFactory;
import org.openecomp.appc.rankingframework.ConfigurationEntry;
import org.openecomp.appc.rankingframework.ConfigurationSet;
import org.openecomp.appc.rankingframework.RankedAttributesResolver;

class WorkflowResolverDataReader {

    private static final String QUERY_STMT = "SELECT action,api_version,vnf_type,vnf_version,dg_name,dg_version,dg_module FROM VNF_DG_MAPPING";

    private static final Collection<String> ATTRIBUTE_NAMES = Arrays.asList("action","api_version", "vnf_type", "vnf_version");

    private static class ConfigurationSetAdaptor implements ConfigurationSet<WorkflowKey> {

        private final ResultSet resultSet;

        private class ResultSetIterator implements Iterator<ConfigurationEntry<WorkflowKey>>, ConfigurationEntry<WorkflowKey> {
            @Override
            public boolean hasNext() {
                try {
                    return resultSet.next();
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }

            @Override
            public ConfigurationEntry<WorkflowKey> next() {
                return this;
            }

            @Override
            public void remove() {
                throw new UnsupportedOperationException();
            }

            @Override
            public Object getAttributeValue(String name) {
                try {
                    return resultSet.getObject(name);
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }

            @Override
            public WorkflowKey getResult() {
                try {
                    return new WorkflowKey(resultSet.getString("dg_name"), resultSet.getString("dg_version"), resultSet.getString("dg_module"));
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }
        }

        ConfigurationSetAdaptor(ResultSet resultSet) {
            this.resultSet = resultSet;
        }

        @Override
        public Iterable<ConfigurationEntry<WorkflowKey>> getEntries() {
            return new Iterable<ConfigurationEntry<WorkflowKey>>() {

                @Override
                public Iterator<ConfigurationEntry<WorkflowKey>> iterator() {
                    return new ResultSetIterator();
                }
            };
        }

        @Override
        public Collection<String> getRankedAttributeNames() {
            return ATTRIBUTE_NAMES;
        }
    }

    RankedAttributesResolver<WorkflowKey> read() {
        try {
            try (Connection conn = DBUtils.getConnection("sdnctl")) {
                try (PreparedStatement stmt = conn.prepareStatement(QUERY_STMT)) {
                    try (ResultSet res = stmt.executeQuery()) {
                        if (res.next()) {
                            res.beforeFirst();
                            ConfigurationSet<WorkflowKey> resolverConfig = new ConfigurationSetAdaptor(res);
                            return AbstractRankedAttributesResolverFactory.getInstance().create(resolverConfig);
                        } else {
                            // TODO: Return empty object
                            throw new IllegalStateException();
                        }
                    }
                }
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
    }
}