You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@linkis.apache.org by pe...@apache.org on 2022/01/30 07:24:01 UTC
[incubator-linkis] 04/07: define metadata utils class
This is an automated email from the ASF dual-hosted git repository.
peacewong pushed a commit to branch dev-1.1.0-datasource
in repository https://gitbox.apache.org/repos/asf/incubator-linkis.git
commit 5f4c28fc8aa0537e408dfde1d8d56700c7ada679
Author: xiaojie19852006 <xi...@163.com>
AuthorDate: Sun Jan 30 13:57:13 2022 +0800
define metadata utils class
---
.../server/utils/MetadataUtils.java | 157 +++++++++++++++++++++
1 file changed, 157 insertions(+)
diff --git a/linkis-public-enhancements/linkis-datasource/linkis-metadata-manager/server/src/main/java/org/apache/linkis/metadatamanager/server/utils/MetadataUtils.java b/linkis-public-enhancements/linkis-datasource/linkis-metadata-manager/server/src/main/java/org/apache/linkis/metadatamanager/server/utils/MetadataUtils.java
new file mode 100644
index 0000000..51d56a2
--- /dev/null
+++ b/linkis-public-enhancements/linkis-datasource/linkis-metadata-manager/server/src/main/java/org/apache/linkis/metadatamanager/server/utils/MetadataUtils.java
@@ -0,0 +1,157 @@
+/*
+ * 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.linkis.metadatamanager.server.utils;
+
+import org.apache.commons.io.IOUtils;
+import org.apache.commons.lang.StringUtils;
+import org.apache.linkis.metadatamanager.common.exception.MetaRuntimeException;
+import org.apache.linkis.metadatamanager.common.service.MetadataService;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.File;
+import java.io.IOException;
+import java.lang.reflect.Constructor;
+import java.lang.reflect.Modifier;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.*;
+import java.util.function.Function;
+import java.util.jar.JarEntry;
+import java.util.jar.JarFile;
+import java.util.stream.Collectors;
+
+/**
+ * Metadata Utils
+ */
+public class MetadataUtils {
+
+ private static final String JAR_SUF_NAME = ".jar";
+
+ private static final String CLASS_SUF_NAME = ".class";
+
+ private static final Logger LOG = LoggerFactory.getLogger(MetadataUtils.class);
+
+ public static MetadataService loadMetaService(Class<? extends MetadataService> metaServiceClass,
+ ClassLoader metaServiceClassLoader){
+ ClassLoader storeClassLoader = Thread.currentThread().getContextClassLoader();
+ Thread.currentThread().setContextClassLoader(metaServiceClassLoader);
+ try{
+ final Constructor<?>[] constructors = metaServiceClass.getConstructors();
+ if (constructors.length <= 0){
+ throw new MetaRuntimeException("No public constructor in meta service class: [" + metaServiceClass.getName() + "]", null);
+ }
+ List<Constructor<?>> acceptConstructor = Arrays.stream(constructors)
+ .filter( constructor -> constructor.getParameterCount() == 0).collect(Collectors.toList());
+ if (acceptConstructor.size() > 0){
+ //Choose the first one
+ Constructor<?> constructor = acceptConstructor.get(0);
+ try {
+ return (MetadataService)constructor.newInstance();
+ } catch (Exception e) {
+ throw new MetaRuntimeException("Unable to construct meta service class: [" + metaServiceClass.getName() + "]", e);
+ }
+ }else{
+ throw new MetaRuntimeException("Illegal arguments in constructor of meta service class: [" + metaServiceClass.getName() + "]", null);
+ }
+ }finally{
+ Thread.currentThread().setContextClassLoader(storeClassLoader);
+ }
+ }
+ /**
+ * Search meta service class from classloader
+ * @param serviceClassLoader service class loader
+ * @return collect
+ */
+ public static String[] searchMetaServiceClassInLoader(URLClassLoader serviceClassLoader){
+ URL[] urlsOfClassLoader = serviceClassLoader.getURLs();
+ List<String> classNameList = new ArrayList<>();
+ for( URL url : urlsOfClassLoader ){
+ String pathForUrl = url.getPath();
+ List<String> searchResult = searchMetaServiceClassFormURI(pathForUrl, className -> isSubMetaServiceClass(className, serviceClassLoader));
+ if (Objects.nonNull(searchResult)){
+ classNameList.addAll(searchResult);
+ }
+ }
+ return classNameList.toArray(new String[]{});
+ }
+
+ public static Class<? extends MetadataService> loadMetaServiceClass(ClassLoader classLoader, String className,
+ boolean initialize, String notFoundMessage){
+ //Try to load use expectClassName
+ try {
+ return Class.forName(className, initialize, classLoader).asSubclass(MetadataService.class);
+ }catch(ClassNotFoundException ne){
+ LOG.warn(notFoundMessage, ne);
+ }
+ return null;
+ }
+
+ private static List<String> searchMetaServiceClassFormURI(String url, Function<String, Boolean> acceptedFunction){
+ List<String> classNameList = new ArrayList<>();
+ if(url.endsWith(CLASS_SUF_NAME)){
+ String className = url.substring(0, url.lastIndexOf(CLASS_SUF_NAME));
+ int splitIndex = className.lastIndexOf(IOUtils.DIR_SEPARATOR);
+ if(splitIndex >= 0){
+ className = className.substring(splitIndex);
+ }
+ if (acceptedFunction.apply(className)){
+ classNameList.add(className);
+ }
+ }else if(url.endsWith(JAR_SUF_NAME)){
+ try {
+ JarFile jarFile = new JarFile(new File(url));
+ Enumeration<JarEntry> en = jarFile.entries();
+ while(en.hasMoreElements()){
+ String name = en.nextElement().getName();
+ if(name.endsWith(CLASS_SUF_NAME)){
+ String className = name.substring(0, name.lastIndexOf(CLASS_SUF_NAME));
+ //If the splicer is different in WINDOWS system?
+ className = className.replaceAll(String.valueOf(IOUtils.DIR_SEPARATOR_UNIX), ".");
+ if(acceptedFunction.apply(className)){
+ classNameList.add(className);
+ }
+ }
+ }
+ } catch (IOException e) {
+ //Trace
+ LOG.trace("Fail to parse jar file:[" + url + "] in service classpath", e);
+ return classNameList;
+ }
+ }
+ return classNameList;
+ }
+
+ private static boolean isSubMetaServiceClass(String className, ClassLoader serviceClassLoader){
+ if(StringUtils.isEmpty(className)){
+ return false;
+ }
+ Class<?> clazz;
+ try{
+ clazz = Class.forName(className, false, serviceClassLoader);
+ //Skip interface and abstract class
+ if (Modifier.isAbstract(clazz.getModifiers()) || Modifier.isInterface(clazz.getModifiers())) {
+ return false;
+ }
+ } catch (Throwable t){
+ LOG.trace("Class: {} can not be found", className, t);
+ return false;
+ }
+ return MetadataService.class.isAssignableFrom(clazz);
+ }
+}
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@linkis.apache.org
For additional commands, e-mail: commits-help@linkis.apache.org