| <!DOCTYPE HTML> |
| <html lang="en"> |
| <head> |
| <title>Source code</title> |
| <link rel="stylesheet" type="text/css" href="../../../../../../../stylesheet.css" title="Style"> |
| </head> |
| <body> |
| <main role="main"> |
| <div class="sourceContainer"> |
| <pre><span class="sourceLineNo">001</span><a id="line.1">package org.apache.archiva.redback.keys.jpa;</a> |
| <span class="sourceLineNo">002</span><a id="line.2"></a> |
| <span class="sourceLineNo">003</span><a id="line.3">/*</a> |
| <span class="sourceLineNo">004</span><a id="line.4"> * Copyright 2001-2016 The Apache Software Foundation.</a> |
| <span class="sourceLineNo">005</span><a id="line.5"> *</a> |
| <span class="sourceLineNo">006</span><a id="line.6"> * Licensed under the Apache License, Version 2.0 (the "License");</a> |
| <span class="sourceLineNo">007</span><a id="line.7"> * you may not use this file except in compliance with the License.</a> |
| <span class="sourceLineNo">008</span><a id="line.8"> * You may obtain a copy of the License at</a> |
| <span class="sourceLineNo">009</span><a id="line.9"> *</a> |
| <span class="sourceLineNo">010</span><a id="line.10"> * http://www.apache.org/licenses/LICENSE-2.0</a> |
| <span class="sourceLineNo">011</span><a id="line.11"> *</a> |
| <span class="sourceLineNo">012</span><a id="line.12"> * Unless required by applicable law or agreed to in writing, software</a> |
| <span class="sourceLineNo">013</span><a id="line.13"> * distributed under the License is distributed on an "AS IS" BASIS,</a> |
| <span class="sourceLineNo">014</span><a id="line.14"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</a> |
| <span class="sourceLineNo">015</span><a id="line.15"> * See the License for the specific language governing permissions and</a> |
| <span class="sourceLineNo">016</span><a id="line.16"> * limitations under the License.</a> |
| <span class="sourceLineNo">017</span><a id="line.17"> */</a> |
| <span class="sourceLineNo">018</span><a id="line.18"></a> |
| <span class="sourceLineNo">019</span><a id="line.19">import org.apache.archiva.redback.keys.AbstractKeyManager;</a> |
| <span class="sourceLineNo">020</span><a id="line.20">import org.apache.archiva.redback.keys.AuthenticationKey;</a> |
| <span class="sourceLineNo">021</span><a id="line.21">import org.apache.archiva.redback.keys.KeyManagerException;</a> |
| <span class="sourceLineNo">022</span><a id="line.22">import org.apache.archiva.redback.keys.KeyNotFoundException;</a> |
| <span class="sourceLineNo">023</span><a id="line.23">import org.apache.archiva.redback.keys.jpa.model.JpaAuthenticationKey;</a> |
| <span class="sourceLineNo">024</span><a id="line.24">import org.apache.commons.lang3.StringUtils;</a> |
| <span class="sourceLineNo">025</span><a id="line.25">import org.springframework.stereotype.Service;</a> |
| <span class="sourceLineNo">026</span><a id="line.26"></a> |
| <span class="sourceLineNo">027</span><a id="line.27">import javax.persistence.*;</a> |
| <span class="sourceLineNo">028</span><a id="line.28">import javax.transaction.Transactional;</a> |
| <span class="sourceLineNo">029</span><a id="line.29">import java.util.Calendar;</a> |
| <span class="sourceLineNo">030</span><a id="line.30">import java.util.List;</a> |
| <span class="sourceLineNo">031</span><a id="line.31">import java.util.concurrent.atomic.AtomicBoolean;</a> |
| <span class="sourceLineNo">032</span><a id="line.32"></a> |
| <span class="sourceLineNo">033</span><a id="line.33">/**</a> |
| <span class="sourceLineNo">034</span><a id="line.34"> * Key Manager Implementation for JPA.</a> |
| <span class="sourceLineNo">035</span><a id="line.35"> *</a> |
| <span class="sourceLineNo">036</span><a id="line.36"> * Uses an injected Entity Manager.</a> |
| <span class="sourceLineNo">037</span><a id="line.37"> *</a> |
| <span class="sourceLineNo">038</span><a id="line.38"> * @author <a href="mailto:martin_s@apache.org">Martin Stockhammer</a></a> |
| <span class="sourceLineNo">039</span><a id="line.39"> */</a> |
| <span class="sourceLineNo">040</span><a id="line.40">@Service( "keyManager#jpa" )</a> |
| <span class="sourceLineNo">041</span><a id="line.41">public class JpaKeyManager extends AbstractKeyManager {</a> |
| <span class="sourceLineNo">042</span><a id="line.42"></a> |
| <span class="sourceLineNo">043</span><a id="line.43"> @PersistenceContext(unitName = "redback-jpa")</a> |
| <span class="sourceLineNo">044</span><a id="line.44"> EntityManager em;</a> |
| <span class="sourceLineNo">045</span><a id="line.45"></a> |
| <span class="sourceLineNo">046</span><a id="line.46"> // JpaUserManager is a singleton and initialization should be thread safe</a> |
| <span class="sourceLineNo">047</span><a id="line.47"> private AtomicBoolean initialized = new AtomicBoolean(false);</a> |
| <span class="sourceLineNo">048</span><a id="line.48"></a> |
| <span class="sourceLineNo">049</span><a id="line.49"> public void setEntityManager(EntityManager em) {</a> |
| <span class="sourceLineNo">050</span><a id="line.50"> this.em = em;</a> |
| <span class="sourceLineNo">051</span><a id="line.51"> }</a> |
| <span class="sourceLineNo">052</span><a id="line.52"></a> |
| <span class="sourceLineNo">053</span><a id="line.53"> private EntityManager getEm() {</a> |
| <span class="sourceLineNo">054</span><a id="line.54"> if (initialized.compareAndSet(false,true)) {</a> |
| <span class="sourceLineNo">055</span><a id="line.55"> Query q = em.createQuery("SELECT COUNT(u.key) FROM JpaAuthenticationKey u");</a> |
| <span class="sourceLineNo">056</span><a id="line.56"> boolean dbInit = q.getFirstResult()==0;</a> |
| <span class="sourceLineNo">057</span><a id="line.57"> }</a> |
| <span class="sourceLineNo">058</span><a id="line.58"> return em;</a> |
| <span class="sourceLineNo">059</span><a id="line.59"> }</a> |
| <span class="sourceLineNo">060</span><a id="line.60"></a> |
| <span class="sourceLineNo">061</span><a id="line.61"> public String getId()</a> |
| <span class="sourceLineNo">062</span><a id="line.62"> {</a> |
| <span class="sourceLineNo">063</span><a id="line.63"> return "JPA Key Manager - " + this.getClass().getName();</a> |
| <span class="sourceLineNo">064</span><a id="line.64"> }</a> |
| <span class="sourceLineNo">065</span><a id="line.65"></a> |
| <span class="sourceLineNo">066</span><a id="line.66"></a> |
| <span class="sourceLineNo">067</span><a id="line.67"> @Override</a> |
| <span class="sourceLineNo">068</span><a id="line.68"> @Transactional</a> |
| <span class="sourceLineNo">069</span><a id="line.69"> public AuthenticationKey createKey( String principal, String purpose, int expirationMinutes )</a> |
| <span class="sourceLineNo">070</span><a id="line.70"> throws KeyManagerException</a> |
| <span class="sourceLineNo">071</span><a id="line.71"> {</a> |
| <span class="sourceLineNo">072</span><a id="line.72"> JpaAuthenticationKey authkey = new JpaAuthenticationKey();</a> |
| <span class="sourceLineNo">073</span><a id="line.73"> authkey.setKey( super.generateUUID() );</a> |
| <span class="sourceLineNo">074</span><a id="line.74"> authkey.setForPrincipal( principal );</a> |
| <span class="sourceLineNo">075</span><a id="line.75"> authkey.setPurpose( purpose );</a> |
| <span class="sourceLineNo">076</span><a id="line.76"></a> |
| <span class="sourceLineNo">077</span><a id="line.77"> Calendar now = getNowGMT();</a> |
| <span class="sourceLineNo">078</span><a id="line.78"> authkey.setDateCreated( now.getTime() );</a> |
| <span class="sourceLineNo">079</span><a id="line.79"></a> |
| <span class="sourceLineNo">080</span><a id="line.80"> if ( expirationMinutes >= 0 )</a> |
| <span class="sourceLineNo">081</span><a id="line.81"> {</a> |
| <span class="sourceLineNo">082</span><a id="line.82"> Calendar expiration = getNowGMT();</a> |
| <span class="sourceLineNo">083</span><a id="line.83"> expiration.add( Calendar.MINUTE, expirationMinutes );</a> |
| <span class="sourceLineNo">084</span><a id="line.84"> authkey.setDateExpires( expiration.getTime() );</a> |
| <span class="sourceLineNo">085</span><a id="line.85"> }</a> |
| <span class="sourceLineNo">086</span><a id="line.86"></a> |
| <span class="sourceLineNo">087</span><a id="line.87"> return addKey( authkey );</a> |
| <span class="sourceLineNo">088</span><a id="line.88"> }</a> |
| <span class="sourceLineNo">089</span><a id="line.89"></a> |
| <span class="sourceLineNo">090</span><a id="line.90"> @Transactional</a> |
| <span class="sourceLineNo">091</span><a id="line.91"> @Override</a> |
| <span class="sourceLineNo">092</span><a id="line.92"> public AuthenticationKey addKey(AuthenticationKey key) {</a> |
| <span class="sourceLineNo">093</span><a id="line.93"> final EntityManager em = getEm();</a> |
| <span class="sourceLineNo">094</span><a id="line.94"> AuthenticationKey mergedKey = em.merge((JpaAuthenticationKey)key);</a> |
| <span class="sourceLineNo">095</span><a id="line.95"> return mergedKey;</a> |
| <span class="sourceLineNo">096</span><a id="line.96"> }</a> |
| <span class="sourceLineNo">097</span><a id="line.97"></a> |
| <span class="sourceLineNo">098</span><a id="line.98"> @Transactional</a> |
| <span class="sourceLineNo">099</span><a id="line.99"> @Override</a> |
| <span class="sourceLineNo">100</span><a id="line.100"> public void eraseDatabase()</a> |
| <span class="sourceLineNo">101</span><a id="line.101"> {</a> |
| <span class="sourceLineNo">102</span><a id="line.102"> final EntityManager em = getEm();</a> |
| <span class="sourceLineNo">103</span><a id="line.103"> Query q = em.createQuery("DELETE FROM JpaAuthenticationKey k");</a> |
| <span class="sourceLineNo">104</span><a id="line.104"> q.executeUpdate();</a> |
| <span class="sourceLineNo">105</span><a id="line.105"> }</a> |
| <span class="sourceLineNo">106</span><a id="line.106"></a> |
| <span class="sourceLineNo">107</span><a id="line.107"> @Transactional</a> |
| <span class="sourceLineNo">108</span><a id="line.108"> @Override</a> |
| <span class="sourceLineNo">109</span><a id="line.109"> public AuthenticationKey findKey(final String key) throws KeyNotFoundException, KeyManagerException {</a> |
| <span class="sourceLineNo">110</span><a id="line.110"> final EntityManager em = getEm();</a> |
| <span class="sourceLineNo">111</span><a id="line.111"> if ( StringUtils.isEmpty( key ) )</a> |
| <span class="sourceLineNo">112</span><a id="line.112"> {</a> |
| <span class="sourceLineNo">113</span><a id="line.113"> throw new KeyNotFoundException( "Empty key not found." );</a> |
| <span class="sourceLineNo">114</span><a id="line.114"> }</a> |
| <span class="sourceLineNo">115</span><a id="line.115"></a> |
| <span class="sourceLineNo">116</span><a id="line.116"> try</a> |
| <span class="sourceLineNo">117</span><a id="line.117"> {</a> |
| <span class="sourceLineNo">118</span><a id="line.118"> TypedQuery<JpaAuthenticationKey> q =</a> |
| <span class="sourceLineNo">119</span><a id="line.119"> em.createQuery("SELECT k FROM JpaAuthenticationKey k WHERE k.key = :key",JpaAuthenticationKey.class);</a> |
| <span class="sourceLineNo">120</span><a id="line.120"> q.setParameter("key",key);</a> |
| <span class="sourceLineNo">121</span><a id="line.121"> JpaAuthenticationKey authkey = q.getSingleResult();</a> |
| <span class="sourceLineNo">122</span><a id="line.122"> if ( authkey == null )</a> |
| <span class="sourceLineNo">123</span><a id="line.123"> {</a> |
| <span class="sourceLineNo">124</span><a id="line.124"> throw new KeyNotFoundException( "Key [" + key + "] not found." );</a> |
| <span class="sourceLineNo">125</span><a id="line.125"> }</a> |
| <span class="sourceLineNo">126</span><a id="line.126"> assertNotExpired( authkey );</a> |
| <span class="sourceLineNo">127</span><a id="line.127"></a> |
| <span class="sourceLineNo">128</span><a id="line.128"> return authkey;</a> |
| <span class="sourceLineNo">129</span><a id="line.129"> } catch (NoResultException ex) {</a> |
| <span class="sourceLineNo">130</span><a id="line.130"> throw new KeyNotFoundException("Key [" + key + "] not found.");</a> |
| <span class="sourceLineNo">131</span><a id="line.131"> } catch (KeyNotFoundException ex) {</a> |
| <span class="sourceLineNo">132</span><a id="line.132"> throw ex;</a> |
| <span class="sourceLineNo">133</span><a id="line.133"> } catch (Throwable ex) {</a> |
| <span class="sourceLineNo">134</span><a id="line.134"> log.error("Error while trying to retrieve JpaAuthenticationKey {}", key);</a> |
| <span class="sourceLineNo">135</span><a id="line.135"> throw new KeyManagerException("Error while retrieving key "+key+": "+ex.getMessage(), ex);</a> |
| <span class="sourceLineNo">136</span><a id="line.136"> }</a> |
| <span class="sourceLineNo">137</span><a id="line.137"> }</a> |
| <span class="sourceLineNo">138</span><a id="line.138"></a> |
| <span class="sourceLineNo">139</span><a id="line.139"> @Transactional</a> |
| <span class="sourceLineNo">140</span><a id="line.140"> @Override</a> |
| <span class="sourceLineNo">141</span><a id="line.141"> protected void assertNotExpired(AuthenticationKey authkey) throws KeyManagerException {</a> |
| <span class="sourceLineNo">142</span><a id="line.142"> super.assertNotExpired(authkey);</a> |
| <span class="sourceLineNo">143</span><a id="line.143"> }</a> |
| <span class="sourceLineNo">144</span><a id="line.144"></a> |
| <span class="sourceLineNo">145</span><a id="line.145"> @Transactional</a> |
| <span class="sourceLineNo">146</span><a id="line.146"> @Override</a> |
| <span class="sourceLineNo">147</span><a id="line.147"> public void deleteKey(AuthenticationKey key) throws KeyManagerException {</a> |
| <span class="sourceLineNo">148</span><a id="line.148"> final EntityManager em = getEm();</a> |
| <span class="sourceLineNo">149</span><a id="line.149"> em.remove((JpaAuthenticationKey)key);</a> |
| <span class="sourceLineNo">150</span><a id="line.150"> }</a> |
| <span class="sourceLineNo">151</span><a id="line.151"></a> |
| <span class="sourceLineNo">152</span><a id="line.152"> @Transactional</a> |
| <span class="sourceLineNo">153</span><a id="line.153"> @Override</a> |
| <span class="sourceLineNo">154</span><a id="line.154"> public void deleteKey(String key) throws KeyManagerException {</a> |
| <span class="sourceLineNo">155</span><a id="line.155"> try {</a> |
| <span class="sourceLineNo">156</span><a id="line.156"> JpaAuthenticationKey foundKey = (JpaAuthenticationKey)findKey(key);</a> |
| <span class="sourceLineNo">157</span><a id="line.157"> em.remove(foundKey);</a> |
| <span class="sourceLineNo">158</span><a id="line.158"> } catch (KeyNotFoundException ex) {</a> |
| <span class="sourceLineNo">159</span><a id="line.159"> // Ignore</a> |
| <span class="sourceLineNo">160</span><a id="line.160"> } catch (Exception ex) {</a> |
| <span class="sourceLineNo">161</span><a id="line.161"> log.error("Error occured while trying to find key {}: {}", key, ex.getMessage());</a> |
| <span class="sourceLineNo">162</span><a id="line.162"> throw new KeyManagerException("Error while retrieving key "+key, ex);</a> |
| <span class="sourceLineNo">163</span><a id="line.163"> }</a> |
| <span class="sourceLineNo">164</span><a id="line.164"> }</a> |
| <span class="sourceLineNo">165</span><a id="line.165"></a> |
| <span class="sourceLineNo">166</span><a id="line.166"> @Override</a> |
| <span class="sourceLineNo">167</span><a id="line.167"> public List<AuthenticationKey> getAllKeys() {</a> |
| <span class="sourceLineNo">168</span><a id="line.168"> final EntityManager em = getEm();</a> |
| <span class="sourceLineNo">169</span><a id="line.169"> TypedQuery<AuthenticationKey> q= em.createQuery("SELECT x from JpaAuthenticationKey x", AuthenticationKey.class);</a> |
| <span class="sourceLineNo">170</span><a id="line.170"> return q.getResultList();</a> |
| <span class="sourceLineNo">171</span><a id="line.171"> }</a> |
| <span class="sourceLineNo">172</span><a id="line.172"></a> |
| <span class="sourceLineNo">173</span><a id="line.173"></a> |
| <span class="sourceLineNo">174</span><a id="line.174"></a> |
| <span class="sourceLineNo">175</span><a id="line.175"></a> |
| <span class="sourceLineNo">176</span><a id="line.176">}</a> |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| </pre> |
| </div> |
| </main> |
| </body> |
| </html> |
