当前位置导航:炫浪网>>网络学院>>编程开发>>JAVA教程>>Java进阶

学习自定义SMTPAppender中的源码


  package piv.zheng.log4j.test;
  
  import org.apache.log4j.AppenderSkeleton;
  import org.apache.log4j.Layout;
  import org.apache.log4j.helpers.LogLog;
  import org.apache.log4j.spi.LoggingEvent;
  import org.apache.log4j.spi.ErrorCode;
  import java.util.Properties;
  import java.util.Date;
  
  import javax.mail.Session;
  import javax.mail.Transport;
  import javax.mail.Message;
  import javax.mail.MessagingException;
  import javax.mail.internet.MimeMessage;
  import javax.mail.Multipart;
  import javax.mail.internet.MimeMultipart;
  import javax.mail.internet.MimeBodyPart;
  import javax.mail.internet.InternetAddress;
  import javax.mail.internet.AddressException;
  
  public class SMTPAppender extends AppenderSkeleton {
  private String to; //收件人
  private String from; // 发件人
  private String subject; //邮件标题
  private String smtpHost; //邮件服务器
  private String auth; //身份验证标识
  private String username; //用户
  private String password; //密码
  private boolean locationInfo = false;
  
  protected Session ses;
  protected Message msg;
  protected Properties props;
  
  public SMTPAppender() {}
  
  public SMTPAppender(Layout layout) {
  
  this.layout = layout;
  
  }
  
  //应用设置
  public void activateOptions() {
  props = new Properties(System.getProperties());
  if (smtpHost != null) {
  props.put("mail.smtp.host", smtpHost);
  }
  if (auth != null) {
  props.put("mail.smtp.auth", auth);
  }
  if (username != null) {
  props.put("username", username);
  }
  if (password != null) {
  props.put("password", password);
  }
  ses = Session.getInstance(props); //创建session
  
  msg = new MimeMessage(ses); //创建message
  
  try {
  //为message设置发件人
  if (from != null) {
  msg.setFrom(getAddress(from));
  } else {
  msg.setFrom();
  }
  
  //为message设置收件人
  if (to != null) {
  msg.setRecipients(Message.RecipientType.TO, parseAddress(to));
  }
  
  //为message设置标题
  if (subject != null) {
  msg.setSubject(subject);
  }
  } catch (MessagingException e) {
  LogLog.error("Could not activate SMTPAppender options.", e);
  }
  }
  
  //重写执行方法,当执行日志输出时会被调用
  public void append(LoggingEvent event) {
  //检查message及layout
  if (!checkEntryConditions()) {
  return;
  }
  
  event.getThreadName();
  event.getNDC();
  if (locationInfo) {
  event.getLocationInformation();
  }
  
  //发送邮件
  sendBuffer(event);
  }
  
  protected boolean checkEntryConditions() {
  if (this.msg == null) {
  errorHandler.error("Message object not configured.");
  return false;
  }
  
  if (this.layout == null) {
  errorHandler.error("No layout set for appender named [" + name + "].");
  return false;
  }
  return true;
  }
  
  synchronized public void close() {
  this.closed = true;
  }
  
  InternetAddress getAddress(String addressStr) {
  try {
  return new InternetAddress(addressStr);
  } catch (AddressException e) {
  errorHandler.error("Could not parse address [" + addressStr + "].", e, ErrorCode.ADDRESS_PARSE_FAILURE);
  return null;
  }
  }
  
  InternetAddress[] parseAddress(String addressStr) {
  try {
  return InternetAddress.parse(addressStr, true);
  } catch (AddressException e) {
  errorHandler.error("Could not parse address [" + addressStr + "].", e, ErrorCode.ADDRESS_PARSE_FAILURE);
  return null;
  }
  }
  
  public boolean requiresLayout() {
  return true;
  }
  
  //发送邮件
  protected void sendBuffer(LoggingEvent event) {
  try {
  //将日志信息设置到message中
  MimeBodyPart part = new MimeBodyPart();
  
  StringBuffer sbuf = new StringBuffer();
  String t = layout.getHeader();
  if (t != null) {
  sbuf.append(t);
  }
  sbuf.append(layout.format(event));
  if (layout.ignoresThrowable()) {
  String[] s = event.getThrowableStrRep();
  if (s != null) {
  for (int j = 0; j < s.length; j++) {
  sbuf.append(s[j]);
  }
  }
  }
  t = layout.getFooter();
  if (t != null) {
  sbuf.append(t);
  }
  part.setContent(sbuf.toString(), layout.getContentType());
  
  Multipart mp = new MimeMultipart();
  mp.addBodyPart(part);
  
  msg.setContent(mp); //为message设置内容,即日志信息
  msg.setSentDate(new Date()); //为message设置发送时间
  
  //发送邮件
  Transport tr = ses.getTransport("smtp");
  tr.connect(smtpHost, username, password);
  tr.sendMessage(msg, msg.getRecipients(Message.RecipientType.TO));
  tr.close();
  } catch (Exception e) {
  LogLog.error("Error occured while sending e-mail notification.", e);
  }
  }
  
  public void setTo(String to) {
  this.to = to;
  }
  
  public String getTo() {
  return to;
  }
  
  public void setFrom(String from) {
  this.from = from;
  }
  
  public String getFrom() {
  return from;
  }
  
  public String getSubject() {
  return subject;
  }
  
  public void setSubject(String subject) {
  this.subject = subject;
  }
  
  public void setSMTPHost(String smtpHost) {
  this.smtpHost = smtpHost;
  }
  
  public String getSMTPHost() {
  return smtpHost;
  }
  
  public void setAuth(String auth) {
  this.auth = auth;
  }
  
  public String getAuth() {
  return auth;
  }
  
  public void setUsername(String username) {
  this.username = username;
  }
  
  public String getUsername() {
  return username;
  }
  
  public void setPassword(String password) {
  this.password = password;
  }
  
  public String getPassword() {
  return password;
  }
  
  public void setLocationInfo(boolean locationInfo) {
  this.locationInfo = locationInfo;
  }
  
  public boolean getLocationInfo() {
  return locationInfo;
  }
  }
相关内容
赞助商链接