blob: 833ae1b1db1a3f80fae080b5d19c21fdee70b3a2 [file] [log] [blame]
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
<html lang="en">
<head>
<title>Source code</title>
<link rel="stylesheet" type="text/css" href="../../../../../../javadoc.css" title="Style">
</head>
<body>
<div class="sourceContainer">
<pre><span class="sourceLineNo">001</span>// ***************************************************************************************************************************<a name="line.1"></a>
<span class="sourceLineNo">002</span>// * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE file *<a name="line.2"></a>
<span class="sourceLineNo">003</span>// * distributed with this work for additional information regarding copyright ownership. The ASF licenses this file *<a name="line.3"></a>
<span class="sourceLineNo">004</span>// * to you under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance *<a name="line.4"></a>
<span class="sourceLineNo">005</span>// * with the License. You may obtain a copy of the License at *<a name="line.5"></a>
<span class="sourceLineNo">006</span>// * *<a name="line.6"></a>
<span class="sourceLineNo">007</span>// * http://www.apache.org/licenses/LICENSE-2.0 *<a name="line.7"></a>
<span class="sourceLineNo">008</span>// * *<a name="line.8"></a>
<span class="sourceLineNo">009</span>// * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an *<a name="line.9"></a>
<span class="sourceLineNo">010</span>// * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the *<a name="line.10"></a>
<span class="sourceLineNo">011</span>// * specific language governing permissions and limitations under the License. *<a name="line.11"></a>
<span class="sourceLineNo">012</span>// ***************************************************************************************************************************<a name="line.12"></a>
<span class="sourceLineNo">013</span>package org.apache.juneau.rest.springboot;<a name="line.13"></a>
<span class="sourceLineNo">014</span><a name="line.14"></a>
<span class="sourceLineNo">015</span>import static org.apache.juneau.internal.CollectionUtils.*;<a name="line.15"></a>
<span class="sourceLineNo">016</span><a name="line.16"></a>
<span class="sourceLineNo">017</span>import java.util.*;<a name="line.17"></a>
<span class="sourceLineNo">018</span>import java.util.stream.*;<a name="line.18"></a>
<span class="sourceLineNo">019</span><a name="line.19"></a>
<span class="sourceLineNo">020</span>import org.apache.juneau.cp.*;<a name="line.20"></a>
<span class="sourceLineNo">021</span>import org.springframework.context.*;<a name="line.21"></a>
<span class="sourceLineNo">022</span><a name="line.22"></a>
<span class="sourceLineNo">023</span>/**<a name="line.23"></a>
<span class="sourceLineNo">024</span> * A bean store that uses Spring bean resolution to find beans if they're not already in this store.<a name="line.24"></a>
<span class="sourceLineNo">025</span> *<a name="line.25"></a>
<span class="sourceLineNo">026</span> * &lt;ul class='seealso'&gt;<a name="line.26"></a>
<span class="sourceLineNo">027</span> * &lt;li class='link'&gt;{@doc juneau-rest-server-springboot}<a name="line.27"></a>
<span class="sourceLineNo">028</span> * &lt;li class='extlink'&gt;{@source}<a name="line.28"></a>
<span class="sourceLineNo">029</span> * &lt;/ul&gt;<a name="line.29"></a>
<span class="sourceLineNo">030</span> */<a name="line.30"></a>
<span class="sourceLineNo">031</span>public class SpringBeanStore extends BeanStore {<a name="line.31"></a>
<span class="sourceLineNo">032</span><a name="line.32"></a>
<span class="sourceLineNo">033</span> private final Optional&lt;ApplicationContext&gt; appContext;<a name="line.33"></a>
<span class="sourceLineNo">034</span><a name="line.34"></a>
<span class="sourceLineNo">035</span> /**<a name="line.35"></a>
<span class="sourceLineNo">036</span> * Constructor.<a name="line.36"></a>
<span class="sourceLineNo">037</span> *<a name="line.37"></a>
<span class="sourceLineNo">038</span> * @param appContext The Spring application context used to resolve beans.<a name="line.38"></a>
<span class="sourceLineNo">039</span> * @param parent The parent REST object bean store. Can be &lt;jk&gt;null&lt;/jk&gt;.<a name="line.39"></a>
<span class="sourceLineNo">040</span> * @param resource The REST object. Can be &lt;jk&gt;null&lt;/jk&gt;.<a name="line.40"></a>
<span class="sourceLineNo">041</span> */<a name="line.41"></a>
<span class="sourceLineNo">042</span> public SpringBeanStore(Optional&lt;ApplicationContext&gt; appContext, Optional&lt;BeanStore&gt; parent, Object resource) {<a name="line.42"></a>
<span class="sourceLineNo">043</span> super(create().parent(parent.orElse(null)).outer(resource));<a name="line.43"></a>
<span class="sourceLineNo">044</span> this.appContext = appContext;<a name="line.44"></a>
<span class="sourceLineNo">045</span> }<a name="line.45"></a>
<span class="sourceLineNo">046</span><a name="line.46"></a>
<span class="sourceLineNo">047</span> @Override<a name="line.47"></a>
<span class="sourceLineNo">048</span> public &lt;T&gt; Optional&lt;T&gt; getBean(Class&lt;T&gt; c) {<a name="line.48"></a>
<span class="sourceLineNo">049</span> try {<a name="line.49"></a>
<span class="sourceLineNo">050</span> Optional&lt;T&gt; o = super.getBean(c);<a name="line.50"></a>
<span class="sourceLineNo">051</span> if (o.isPresent())<a name="line.51"></a>
<span class="sourceLineNo">052</span> return o;<a name="line.52"></a>
<span class="sourceLineNo">053</span> if (appContext.isPresent()) {<a name="line.53"></a>
<span class="sourceLineNo">054</span> return optional(appContext.get().getBeanProvider(c).getIfAvailable());<a name="line.54"></a>
<span class="sourceLineNo">055</span> }<a name="line.55"></a>
<span class="sourceLineNo">056</span> } catch (Exception e) {<a name="line.56"></a>
<span class="sourceLineNo">057</span> e.printStackTrace();<a name="line.57"></a>
<span class="sourceLineNo">058</span> }<a name="line.58"></a>
<span class="sourceLineNo">059</span> return empty();<a name="line.59"></a>
<span class="sourceLineNo">060</span> }<a name="line.60"></a>
<span class="sourceLineNo">061</span><a name="line.61"></a>
<span class="sourceLineNo">062</span> @Override<a name="line.62"></a>
<span class="sourceLineNo">063</span> public &lt;T&gt; Optional&lt;T&gt; getBean(Class&lt;T&gt; c, String name) {<a name="line.63"></a>
<span class="sourceLineNo">064</span> try {<a name="line.64"></a>
<span class="sourceLineNo">065</span> Optional&lt;T&gt; o = super.getBean(c, name);<a name="line.65"></a>
<span class="sourceLineNo">066</span> if (o.isPresent())<a name="line.66"></a>
<span class="sourceLineNo">067</span> return o;<a name="line.67"></a>
<span class="sourceLineNo">068</span> if (appContext.isPresent()) {<a name="line.68"></a>
<span class="sourceLineNo">069</span> ApplicationContext ctx = appContext.get();<a name="line.69"></a>
<span class="sourceLineNo">070</span> if (name != null)<a name="line.70"></a>
<span class="sourceLineNo">071</span> return optional(ctx.containsBean(name) ? appContext.get().getBean(name, c) : null);<a name="line.71"></a>
<span class="sourceLineNo">072</span> return optional(appContext.get().getBean(c));<a name="line.72"></a>
<span class="sourceLineNo">073</span> }<a name="line.73"></a>
<span class="sourceLineNo">074</span> } catch (Exception e) {<a name="line.74"></a>
<span class="sourceLineNo">075</span> e.printStackTrace();<a name="line.75"></a>
<span class="sourceLineNo">076</span> }<a name="line.76"></a>
<span class="sourceLineNo">077</span> return empty();<a name="line.77"></a>
<span class="sourceLineNo">078</span> }<a name="line.78"></a>
<span class="sourceLineNo">079</span><a name="line.79"></a>
<span class="sourceLineNo">080</span> @SuppressWarnings("unchecked")<a name="line.80"></a>
<span class="sourceLineNo">081</span> @Override<a name="line.81"></a>
<span class="sourceLineNo">082</span> public &lt;T&gt; Stream&lt;BeanStoreEntry&lt;T&gt;&gt; stream(Class&lt;T&gt; c) {<a name="line.82"></a>
<span class="sourceLineNo">083</span> try {<a name="line.83"></a>
<span class="sourceLineNo">084</span> Stream&lt;BeanStoreEntry&lt;T&gt;&gt; o = super.stream(c);<a name="line.84"></a>
<span class="sourceLineNo">085</span> if (appContext.isPresent())<a name="line.85"></a>
<span class="sourceLineNo">086</span> o = Stream.concat(o, appContext.get().getBeansOfType(c).entrySet().stream().map(x -&gt; BeanStoreEntry.create(c, ()-&gt;x.getValue(), x.getKey())));<a name="line.86"></a>
<span class="sourceLineNo">087</span> return o;<a name="line.87"></a>
<span class="sourceLineNo">088</span> } catch (Exception e) {<a name="line.88"></a>
<span class="sourceLineNo">089</span> e.printStackTrace();<a name="line.89"></a>
<span class="sourceLineNo">090</span> }<a name="line.90"></a>
<span class="sourceLineNo">091</span> return Collections.emptyList().stream().map(x -&gt; (BeanStoreEntry&lt;T&gt;)x);<a name="line.91"></a>
<span class="sourceLineNo">092</span> }<a name="line.92"></a>
<span class="sourceLineNo">093</span>}<a name="line.93"></a>
</pre>
</div>
</body>
</html>