org.nutz.spring.boot.dao.SpringResourceLoaction Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nutz-spring-boot-starter Show documentation

spring boot nutz starter

There is a newer version: 3.4.1

Show newest version

package org.nutz.spring.boot.dao;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.regex.Pattern;

import org.nutz.lang.Lang;
import org.nutz.resource.NutResource;
import org.nutz.resource.impl.ResourceLocation;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.io.Resource;

/**
 * 
 * @author Kerbores([email protected])
 *
 */
public class SpringResourceLoaction extends ResourceLocation implements ApplicationContextAware {

	protected ApplicationContext applicationContext;

	@Override
	public String id() {
		return "spring";
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.nutz.resource.impl.ResourceLocation#equals(java.lang.Object)
	 */
	@Override
	public boolean equals(Object obj) {
		return super.equals(obj);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.nutz.resource.impl.ResourceLocation#hashCode()
	 */
	@Override
	public int hashCode() {
		return super.hashCode();
	}

	@Override
	public void scan(String base, Pattern pattern, List list) {
		base = pattern.matcher(base).find() ? "classpath*:" + base : "classpath*:" + base + "/**";
		try {
			Resource[] tmp = applicationContext.getResources(base);
			for (Resource resource : tmp) {
				
				if (resource.getFilename() == null || !resource.isReadable() || !pattern.matcher(resource.getFilename()).find())
					continue;
				SpringResource sr = new SpringResource();
				sr.resource = resource;
				sr.setName(resource.getFilename());
				sr.setSource("spring");
				list.add(sr);
			}
		} catch (IOException e) {
			throw Lang.wrapThrow(e);
		}
	}

	@Override
	public void setApplicationContext(ApplicationContext applicationContext) {
		this.applicationContext = applicationContext;
	}

	public class SpringResource extends NutResource {

		protected Resource resource;

		@Override
		public InputStream getInputStream() throws IOException {
			return resource.getInputStream();
		}

		/*
		 * (non-Javadoc)
		 * 
		 * @see org.nutz.resource.NutResource#equals(java.lang.Object)
		 */
		@Override
		public boolean equals(Object obj) {
			return super.equals(obj);
		}

		/*
		 * (non-Javadoc)
		 * 
		 * @see org.nutz.resource.NutResource#hashCode()
		 */
		@Override
		public int hashCode() {
			return super.hashCode();
		}

	}
}