1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one or more
3 * contributor license agreements. See the NOTICE file distributed with
4 * this work for additional information regarding copyright ownership.
5 * The ASF licenses this file to You under the Apache License, Version 2.0
6 * (the "License"); you may not use this file except in compliance with
7 * the License. You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 */
17
18 package org.apache.shardingsphere.proxy.frontend.postgresql.command;
19
20 import com.google.common.base.Preconditions;
21 import org.apache.shardingsphere.proxy.frontend.postgresql.command.query.extended.Portal;
22
23 import java.sql.SQLException;
24 import java.util.LinkedHashMap;
25 import java.util.Map;
26
27 /**
28 * PostgreSQL portal context.
29 */
30 public final class PortalContext {
31
32 private final Map<String, Portal> portals = new LinkedHashMap<>();
33
34 /**
35 * Add portal.
36 *
37 * @param portal portal name
38 * @throws SQLException SQL exception
39 */
40 public void add(final Portal portal) throws SQLException {
41 boolean isNamedPortal = !portal.getName().isEmpty();
42 Preconditions.checkState(!isNamedPortal || !portals.containsKey(portal.getName()), "Named portal `%s` must be explicitly closed", portal.getName());
43 Portal previousPortal = portals.put(portal.getName(), portal);
44 if (null != previousPortal) {
45 previousPortal.close();
46 }
47 }
48
49 /**
50 * Get portal.
51 *
52 * @param portalName portal name
53 * @return portal
54 */
55 public Portal get(final String portalName) {
56 return portals.get(portalName);
57 }
58
59 /**
60 * Close portal.
61 *
62 * @param portalName portal name
63 * @throws SQLException SQL exception
64 */
65 public void close(final String portalName) throws SQLException {
66 Portal result = portals.remove(portalName);
67 if (null != result) {
68 result.close();
69 }
70 }
71
72 /**
73 * Close all portals.
74 *
75 * @throws SQLException SQL exception
76 */
77 public void closeAll() throws SQLException {
78 for (Portal portal : portals.values()) {
79 portal.close();
80 }
81 portals.clear();
82 }
83 }