| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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. |
| */ |
| |
| package org.apache.ignite.internal.processors.cache.query.jdbc; |
| |
| import java.sql.SQLException; |
| import java.util.Collection; |
| import java.util.HashMap; |
| import java.util.LinkedList; |
| import java.util.List; |
| import java.util.Map; |
| import org.apache.ignite.Ignite; |
| import org.apache.ignite.IgniteCheckedException; |
| import org.apache.ignite.IgniteException; |
| import org.apache.ignite.IgniteLogger; |
| import org.apache.ignite.cluster.ClusterNode; |
| import org.apache.ignite.compute.ComputeJob; |
| import org.apache.ignite.compute.ComputeJobAdapter; |
| import org.apache.ignite.compute.ComputeJobResult; |
| import org.apache.ignite.compute.ComputeTaskAdapter; |
| import org.apache.ignite.internal.IgniteEx; |
| import org.apache.ignite.internal.IgniteKernal; |
| import org.apache.ignite.internal.managers.discovery.GridDiscoveryManager; |
| import org.apache.ignite.internal.processors.cache.IgniteInternalCache; |
| import org.apache.ignite.internal.processors.cache.query.GridCacheSqlIndexMetadata; |
| import org.apache.ignite.internal.processors.cache.query.GridCacheSqlMetadata; |
| import org.apache.ignite.internal.util.typedef.F; |
| import org.apache.ignite.internal.util.typedef.internal.U; |
| import org.apache.ignite.marshaller.Marshaller; |
| import org.apache.ignite.marshaller.jdk.JdkMarshaller; |
| import org.apache.ignite.resources.IgniteInstanceResource; |
| import org.apache.ignite.resources.LoggerResource; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| |
| /** |
| * Task that gets metadata for JDBC adapter. |
| */ |
| public class GridCacheQueryJdbcMetadataTask extends ComputeTaskAdapter<String, byte[]> { |
| /** */ |
| private static final long serialVersionUID = 0L; |
| |
| /** Marshaller. */ |
| private static final Marshaller MARSHALLER = new JdkMarshaller(); |
| |
| /** */ |
| @IgniteInstanceResource |
| private Ignite ignite; |
| |
| /** {@inheritDoc} */ |
| @NotNull @Override public Map<? extends ComputeJob, ClusterNode> map(List<ClusterNode> subgrid, |
| @Nullable String cacheName) { |
| Map<JdbcDriverMetadataJob, ClusterNode> map = new HashMap<>(); |
| |
| IgniteKernal kernal = (IgniteKernal)ignite; |
| |
| GridDiscoveryManager discoMgr = kernal.context().discovery(); |
| |
| for (ClusterNode n : subgrid) |
| if (discoMgr.cacheAffinityNode(n, cacheName)) { |
| map.put(new JdbcDriverMetadataJob(cacheName), n); |
| |
| break; |
| } |
| |
| return map; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public byte[] reduce(List<ComputeJobResult> results) { |
| return F.first(results).getData(); |
| } |
| |
| /** |
| * Job for JDBC adapter. |
| */ |
| private static class JdbcDriverMetadataJob extends ComputeJobAdapter { |
| /** */ |
| private static final long serialVersionUID = 0L; |
| |
| /** Cache name. */ |
| private final String cacheName; |
| |
| /** Grid instance. */ |
| @IgniteInstanceResource |
| private Ignite ignite; |
| |
| /** Logger. */ |
| @LoggerResource |
| private IgniteLogger log; |
| |
| /** |
| * @param cacheName Cache name. |
| */ |
| private JdbcDriverMetadataJob(@Nullable String cacheName) { |
| this.cacheName = cacheName; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public Object execute() { |
| byte status; |
| byte[] data; |
| |
| try { |
| IgniteInternalCache<?, ?> cache = ((IgniteEx)ignite).cachex(cacheName); |
| |
| assert cache != null; |
| |
| Collection<GridCacheSqlMetadata> metas = cache.context().queries().sqlMetadata(); |
| |
| Map<String, Map<String, Map<String, String>>> schemasMap = U.newHashMap(metas.size()); |
| |
| Collection<List<Object>> indexesInfo = new LinkedList<>(); |
| |
| for (GridCacheSqlMetadata meta : metas) { |
| String name = meta.cacheName(); |
| |
| if (name == null) |
| name = "PUBLIC"; |
| |
| Collection<String> types = meta.types(); |
| |
| Map<String, Map<String, String>> typesMap = U.newHashMap(types.size()); |
| |
| for (String type : types) { |
| typesMap.put(type.toUpperCase(), meta.fields(type)); |
| |
| for (GridCacheSqlIndexMetadata idx : meta.indexes(type)) { |
| int cnt = 0; |
| |
| for (String field : idx.fields()) { |
| indexesInfo.add(F.<Object>asList(name, type.toUpperCase(), !idx.unique(), |
| idx.name().toUpperCase(), ++cnt, field, idx.descending(field))); |
| } |
| } |
| } |
| |
| schemasMap.put(name, typesMap); |
| } |
| |
| status = 0; |
| |
| data = U.marshal(MARSHALLER, F.asList(schemasMap, indexesInfo)); |
| } |
| catch (Throwable t) { |
| U.error(log, "Failed to get metadata for JDBC.", t); |
| |
| SQLException err = new SQLException(t.getMessage()); |
| |
| status = 1; |
| |
| try { |
| data = U.marshal(MARSHALLER, err); |
| } |
| catch (IgniteCheckedException e) { |
| throw new IgniteException(e); |
| } |
| |
| if (t instanceof Error) |
| throw (Error)t; |
| } |
| |
| byte[] packet = new byte[data.length + 1]; |
| |
| packet[0] = status; |
| |
| U.arrayCopy(data, 0, packet, 1, data.length); |
| |
| return packet; |
| } |
| } |
| } |