2020import java .util .Collections ;
2121import java .util .List ;
2222
23- import org .springframework .beans .factory .DisposableBean ;
24- import org .springframework .beans .factory .FactoryBean ;
25- import org .springframework .beans .factory .InitializingBean ;
23+ import org .springframework .beans .factory .config .AbstractFactoryBean ;
2624import org .springframework .dao .DataAccessException ;
2725import org .springframework .dao .support .PersistenceExceptionTranslator ;
2826import org .springframework .util .CollectionUtils ;
4038 * @author Christoph Strobl
4139 * @since 1.7
4240 */
43- public class MongoClientFactoryBean implements FactoryBean <Mongo >, InitializingBean , DisposableBean ,
44- PersistenceExceptionTranslator {
41+ public class MongoClientFactoryBean extends AbstractFactoryBean <Mongo > implements PersistenceExceptionTranslator {
4542
46- private MongoClient mongo ;
43+ private static final PersistenceExceptionTranslator DEFAULT_EXCEPTION_TRANSLATOR = new MongoExceptionTranslator () ;
4744
4845 private MongoClientOptions mongoClientOptions ;
49-
5046 private String host ;
5147 private Integer port ;
5248 private List <ServerAddress > replicaSetSeeds ;
5349 private List <MongoCredential > credentials ;
5450
55- private PersistenceExceptionTranslator exceptionTranslator = new MongoExceptionTranslator () ;
51+ private PersistenceExceptionTranslator exceptionTranslator = DEFAULT_EXCEPTION_TRANSLATOR ;
5652
5753 /**
5854 * Set the {@link MongoClientOptions} to be used when creating {@link MongoClient}.
@@ -72,52 +68,40 @@ public void setCredentials(MongoCredential[] credentials) {
7268 this .credentials = filterNonNullElementsAsList (credentials );
7369 }
7470
71+ /**
72+ * Set the list of {@link ServerAddress} to build up a replica set for.
73+ *
74+ * @param replicaSetSeeds can be {@literal null}.
75+ */
7576 public void setReplicaSetSeeds (ServerAddress [] replicaSetSeeds ) {
7677 this .replicaSetSeeds = filterNonNullElementsAsList (replicaSetSeeds );
7778 }
7879
7980 /**
80- * @param elements the elements to filter <T>
81- * @return a new unmodifiable {@link List#} from the given elements without nulls
81+ * Configures the host to connect to.
82+ *
83+ * @param host
8284 */
83- private <T > List <T > filterNonNullElementsAsList (T [] elements ) {
84-
85- if (elements == null ) {
86- return Collections .emptyList ();
87- }
88-
89- List <T > candidateElements = new ArrayList <T >();
90-
91- for (T element : elements ) {
92- if (element != null ) {
93- candidateElements .add (element );
94- }
95- }
96-
97- return Collections .unmodifiableList (candidateElements );
98- }
99-
10085 public void setHost (String host ) {
10186 this .host = host ;
10287 }
10388
89+ /**
90+ * Configures the port to connect to.
91+ *
92+ * @param port
93+ */
10494 public void setPort (int port ) {
10595 this .port = port ;
10696 }
10797
10898 /**
99+ * Configures the {@link PersistenceExceptionTranslator} to use.
100+ *
109101 * @param exceptionTranslator
110102 */
111103 public void setExceptionTranslator (PersistenceExceptionTranslator exceptionTranslator ) {
112- this .exceptionTranslator = exceptionTranslator ;
113- }
114-
115- /*
116- * (non-Javadoc)
117- * @see org.springframework.beans.factory.FactoryBean#getObject()
118- */
119- public Mongo getObject () throws Exception {
120- return mongo ;
104+ this .exceptionTranslator = exceptionTranslator == null ? DEFAULT_EXCEPTION_TRANSLATOR : exceptionTranslator ;
121105 }
122106
123107 /*
@@ -128,14 +112,6 @@ public Class<? extends Mongo> getObjectType() {
128112 return Mongo .class ;
129113 }
130114
131- /*
132- * (non-Javadoc)
133- * @see org.springframework.beans.factory.FactoryBean#isSingleton()
134- */
135- public boolean isSingleton () {
136- return true ;
137- }
138-
139115 /*
140116 * (non-Javadoc)
141117 * @see org.springframework.dao.support.PersistenceExceptionTranslator#translateExceptionIfPossible(java.lang.RuntimeException)
@@ -146,18 +122,29 @@ public DataAccessException translateExceptionIfPossible(RuntimeException ex) {
146122
147123 /*
148124 * (non-Javadoc)
149- * @see org.springframework.beans.factory.InitializingBean#afterPropertiesSet ()
125+ * @see org.springframework.beans.factory.config.AbstractFactoryBean#createInstance ()
150126 */
151- public void afterPropertiesSet () throws Exception {
127+ @ Override
128+ protected Mongo createInstance () throws Exception {
152129
153130 if (mongoClientOptions == null ) {
154131 mongoClientOptions = MongoClientOptions .builder ().build ();
155132 }
133+
156134 if (credentials == null ) {
157135 credentials = Collections .emptyList ();
158136 }
159137
160- this .mongo = createMongoClient ();
138+ return createMongoClient ();
139+ }
140+
141+ /*
142+ * (non-Javadoc)
143+ * @see org.springframework.beans.factory.config.AbstractFactoryBean#destroyInstance(java.lang.Object)
144+ */
145+ @ Override
146+ protected void destroyInstance (Mongo instance ) throws Exception {
147+ instance .close ();
161148 }
162149
163150 private MongoClient createMongoClient () throws UnknownHostException {
@@ -172,15 +159,31 @@ private MongoClient createMongoClient() throws UnknownHostException {
172159 private ServerAddress createConfiguredOrDefaultServerAddress () throws UnknownHostException {
173160
174161 ServerAddress defaultAddress = new ServerAddress ();
162+
175163 return new ServerAddress (StringUtils .hasText (host ) ? host : defaultAddress .getHost (),
176164 port != null ? port .intValue () : defaultAddress .getPort ());
177165 }
178166
179- /*
180- * (non-Javadoc)
181- * @see org.springframework.beans.factory.DisposableBean#destroy()
167+ /**
168+ * Returns the given array as {@link List} with all {@literal null} elements removed.
169+ *
170+ * @param elements the elements to filter <T>, can be {@literal null}.
171+ * @return a new unmodifiable {@link List#} from the given elements without {@literal null}s.
182172 */
183- public void destroy () throws Exception {
184- this .mongo .close ();
173+ private static <T > List <T > filterNonNullElementsAsList (T [] elements ) {
174+
175+ if (elements == null ) {
176+ return Collections .emptyList ();
177+ }
178+
179+ List <T > candidateElements = new ArrayList <T >();
180+
181+ for (T element : elements ) {
182+ if (element != null ) {
183+ candidateElements .add (element );
184+ }
185+ }
186+
187+ return Collections .unmodifiableList (candidateElements );
185188 }
186189}
0 commit comments