Skip to content

Commit 6993baa

Browse files
Set smtp_host as env variable (#1119)
1 parent d899b71 commit 6993baa

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

cuebot/src/main/java/com/imageworks/spcue/util/CueUtil.java

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
import java.util.regex.Pattern;
3636
import javax.activation.DataHandler;
3737
import javax.activation.DataSource;
38+
import javax.annotation.PostConstruct;
3839
import javax.mail.BodyPart;
3940
import javax.mail.Message;
4041
import javax.mail.Session;
@@ -47,17 +48,24 @@
4748

4849
import org.apache.log4j.Logger;
4950
import org.springframework.core.env.Environment;
51+
import org.springframework.beans.factory.annotation.Autowired;
52+
import org.springframework.stereotype.Component;
5053

5154
import com.imageworks.spcue.LayerInterface;
5255
import com.imageworks.spcue.SpcueRuntimeException;
5356
import com.imageworks.spcue.dispatcher.Dispatcher;
5457

58+
5559
/**
5660
* CueUtil is set of common methods used throughout the application.
5761
*/
62+
@Component
5863
public final class CueUtil {
5964

6065
private static final Logger logger = Logger.getLogger(CueUtil.class);
66+
private static String smtpHost = "";
67+
@Autowired
68+
private Environment env;
6169

6270
/**
6371
* Commonly used macros for gigabyte values in KB.
@@ -88,6 +96,11 @@ public final class CueUtil {
8896
*/
8997
public static final int ONE_HOUR = 3600;
9098

99+
@PostConstruct
100+
public void init() {
101+
CueUtil.smtpHost = this.env.getRequiredProperty("smtp_host", String.class);
102+
}
103+
91104
/**
92105
* Return true if the given name is formatted as a valid
93106
* allocation name. Allocation names should be facility.unique_name.
@@ -157,7 +170,7 @@ public static int findChunk(List<Integer> dependOnFrames, int dependErFrame) {
157170
public static void sendmail(String to, String from, String subject, StringBuilder body, Map<String, byte[]> images) {
158171
try {
159172
Properties props = System.getProperties();
160-
props.put("mail.smtp.host", "smtp");
173+
props.put("mail.smtp.host", CueUtil.smtpHost);
161174
Session session = Session.getDefaultInstance(props, null);
162175
Message msg = new MimeMessage(session);
163176
msg.setFrom(new InternetAddress(from));
@@ -189,6 +202,8 @@ public static void sendmail(String to, String from, String subject, StringBuilde
189202
msg.setContent(mimeMultipart);
190203
msg.setHeader("X-Mailer", "OpenCueMailer");
191204
msg.setSentDate(new Date());
205+
Transport transport = session.getTransport("smtp");
206+
transport.connect(CueUtil.smtpHost, null, null);
192207
Transport.send(msg);
193208
}
194209
catch (Exception e) {

cuebot/src/main/resources/opencue.properties

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,3 +117,6 @@ history.archive_jobs_cutoff_hours=72
117117

118118
# Delete down hosts automatically.
119119
maintenance.auto_delete_down_hosts=false
120+
121+
# Set hostname/IP of the smtp host. Will be used for mailing
122+
smtp_host=smtp

0 commit comments

Comments
 (0)