com.baomidou.framework.mail.MailHelper.java Source code

Introduction

Here is the source code for com.baomidou.framework.mail.MailHelper.java

Source

/**
 * Copyright (c) 2011-2014, hubin (jobob@qq.com).
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.baomidou.framework.mail;

import java.util.Map;

import javax.mail.internet.MimeMessage;

import org.apache.velocity.app.VelocityEngine;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.ui.Model;
import org.springframework.ui.velocity.VelocityEngineUtils;

import com.baomidou.framework.common.SwConstants;

/**
 * <p>
 * 
 * </p>
 * 
 * @author hubin
 * @Date 2016-01-15
 */
public class MailHelper {

    protected Logger logger = LoggerFactory.getLogger(MailHelper.class);

    private VelocityEngine velocityEngine;

    private JavaMailSender mailSender;

    private String charset = SwConstants.UTF_8;

    private String mailTitle;

    private String mailFrom;

    public boolean sendMail(String to, String subject, String tplName, Model model) {
        return sendMail(to, subject, tplName, model.asMap());
    }

    public boolean sendMail(String[] to, String subject, String tplName, Model model) {
        return sendMail(to, subject, tplName, model.asMap());
    }

    public boolean sendMail(String to, String subject, String tplName, Map<String, Object> data) {
        return sendMail(new String[] { to }, subject, tplName, data);
    }

    public boolean sendMail(String[] to, String subject, String tplName, Map<String, Object> data) {
        return sendMail(this.mailTitle, this.mailFrom, to, subject, tplName, data);
    }

    public boolean sendMail(String personal, String from, String to, String subject, String tplName, Model model) {
        return sendMail(personal, from, new String[] { to }, subject, tplName, model.asMap());
    }

    public boolean sendMail(String personal, String from, String[] to, String subject, String tplName,
            Model model) {
        return sendMail(personal, from, to, subject, tplName, model.asMap());
    }

    /**
     * ??
     * 
     * @param personal
     *            ??????
     * @param from
     *            ???
     * @param to
     *            ???
     * @param subject
     *            
     * @param tplName
     *            ???xxx.vm  ??/WEB-INF/views/mail/..
     * @param data
     *            ???
     * @return
     */
    public boolean sendMail(String personal, String from, String[] to, String subject, String tplName,
            Map<String, Object> data) {
        try {
            MimeMessage msg = mailSender.createMimeMessage();
            MimeMessageHelper msgHelper = new MimeMessageHelper(msg, getCharset());
            msgHelper.setFrom(from, personal);
            msgHelper.setTo(to);
            msgHelper.setSubject(subject);
            msgHelper.setText(this.getHtmltext(tplName, data), true);
            mailSender.send(msg);
            return true;
        } catch (Exception e) {
            logger.error("send mail error.", e);
        }
        return false;
    }

    /**
     * <p>
     * velocity ? html
     * </p>
     * 
     * @param tplName
     *            ???
     * @param data
     *            ?
     * @return
     */
    public String getHtmltext(String tplName, Map<String, Object> data) {
        return VelocityEngineUtils.mergeTemplateIntoString(this.velocityEngine, tplName, getCharset(), data);
    }

    public String getHtmltext(String tplName, Model model) {
        return getHtmltext(tplName, model.asMap());
    }

    public void setVelocityEngine(VelocityEngine velocityEngine) {
        this.velocityEngine = velocityEngine;
    }

    public void setMailSender(JavaMailSender mailSender) {
        this.mailSender = mailSender;
    }

    public String getCharset() {
        return charset;
    }

    public void setCharset(String charset) {
        this.charset = charset;
    }

    public String getMailTitle() {
        return mailTitle;
    }

    public void setMailTitle(String mailTitle) {
        this.mailTitle = mailTitle;
    }

    public String getMailFrom() {
        return mailFrom;
    }

    public void setMailFrom(String mailFrom) {
        this.mailFrom = mailFrom;
    }

}