org.fastquery.service.GenerateRepositoryImpl Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2016-2017, fastquery.org and/or its affiliates. All rights reserved.
 *
 * 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.
 * 
 * For more information, please see http://www.fastquery.org/.
 * 
 */

package org.fastquery.service;

import java.io.File;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import org.apache.log4j.Logger;
import org.apache.velocity.app.Velocity;
import org.fastquery.asm.AsmRepository;
import org.fastquery.core.GenerateRepository;
import org.fastquery.core.Repository;
import org.fastquery.core.Resource;
import org.fastquery.dsm.FastQueryJson;
import org.fastquery.mapper.QueryPool;
import org.fastquery.util.ClassUtil;
import org.fastquery.util.FastQueryJSONObject;
import org.fastquery.util.LoadPrperties;

/**
 * 这个类在osgi环境中不能调用
 * @author xixifeng ([email protected])
 */
class GenerateRepositoryImpl implements GenerateRepository {

	private static final Logger LOG = Logger.getLogger(GenerateRepositoryImpl.class);
	
	private static GenerateRepositoryImpl instance;

	private GenerateRepositoryImpl() {
		LOG.debug("GenerateRepositoryImpl 已实例化.");
		
		Resource resource = new FQueryResourceImpl();
		
		// 1). 装载配置文件
		Set fqPropertie = LoadPrperties.load(resource);
		
		// 2). 初始化velocity
		String fdir = FastQueryJSONObject.getVelocity();
		if(fdir!=null) {
			File velocity = new File(fdir);
			if( velocity.exists() && velocity.isFile()) {
				Velocity.init(FastQueryJSONObject.getVelocity());	
			} else {
				LOG.info("fastquery.json中的velocity属性指定的文件地址不存在.那行配置不是必须的,请注意:不要重复配置!");
			}	
		}
		
		List> clses = new ArrayList<>();
		
		// 3). 批量生成 Repository 的实现类
		Set basePackages = null;
		for (FastQueryJson fQueryPropertie : fqPropertie) {
			basePackages = fQueryPropertie.getBasePackages();
			for (String basePackage : basePackages) {
				List> classes = ClassUtil.getClasses(basePackage);
				clses.addAll(classes);
				// classes.forEach(this::generate)
				for (Class rcls : classes) {
					generate(rcls); // 生成
					QueryPool.put(rcls.getName(), resource);
				}
			}
		}
		
		// 3). 生成 Repository之后的检测
		AsmRepository.after(clses);
		
	}

	static GenerateRepositoryImpl getInstance() {
		if (instance == null) {
			// 如果在这儿, 可能有两个线程进入了
			synchronized (GenerateRepositoryImpl.class) {
				if (instance == null) { // 这步不能省!
					instance = new GenerateRepositoryImpl();
				}
			}
		}
		return instance;
	}
	
	@SuppressWarnings("unchecked")
	@Override
	public  T generate(Class repositoryClazz) {
		String name = repositoryClazz.getName()+ SUFFIX;
		if( FqClassLoader.findLoadedClassByName(name) == null ) {
			
			byte[] bytes = AsmRepository.generateBytes(repositoryClazz);
			
			// 把生成的文件存储起来
			 try(FileOutputStream fos = new FileOutputStream("/data/tmp/"+name+".class")) {
				fos.write(bytes);
			} catch (Exception e) {
				// TODO: handle exception
			}
			// 把生成的文件存储起来 end

			return (T) new FqClassLoader(this.getClass().getClassLoader()).defineClassByName(name, bytes, 0, bytes.length);
		}
		return null;
	}

	@SuppressWarnings("unchecked")
	@Override
	public  T getProxyRepository(Class clazz) {
		return (T) FqClassLoader.findLoadedClassByName(clazz.getName() + SUFFIX);
	}
	
}