示例#1
0
文件: main.go 项目: kevgathuku/mailer
func main() {
	err := godotenv.Load()
	if err != nil {
		log.Println("Error loading .env file")
	}
	// Create a new instance of Frodo
	App := Frodo.New()

	// Add the root route
	App.Get("/", func(w http.ResponseWriter, r *Frodo.Request) {
		w.Write([]byte("Hello World!!!"))
	})

	App.Post("/", &controller.Home{}, Frodo.Use{
		Method: "Index",
	})

	// Specify PORT as an environment variable
	// Necessary for running on Heroku
	port := "3102"
	if os.Getenv("PORT") != "" {
		port = os.Getenv("PORT")
	}
	App.ServeOnPort(port)
}
示例#2
0
func init() {
	godotenv.Load()
}
示例#3
0
文件: home.go 项目: kevgathuku/mailer
// Index defines the Post request handler
func (h *Home) Index(w http.ResponseWriter, r *Frodo.Request) {
	err := godotenv.Load()
	if err != nil {
		log.Println("Error loading .env file")
	}
	// Decode the JSON from the POST request body
	decoder := json.NewDecoder(r.Body)
	// Initialize a new newsletter struct
	var t newsletter

	decodeErr := decoder.Decode(&t)
	if decodeErr != nil {
		panic(decodeErr)
	}

	recipients := t.Subscribers
	sender := os.Getenv("SENDER_ADDRESS")
	workerCount := 8
	debug := true

	mailer := mail.NewMailer(
		os.Getenv("SMTP_USERNAME"),
		os.Getenv("SMTP_PASSWORD"),
		os.Getenv("SMTP_HOST"),
		os.Getenv("SMTP_PORT"),
	)

	jobs := make(chan string, len(recipients))
	success := make(chan *email.Email)
	fail := make(chan error)
	htmlContent := t.Content
	subject := t.Title

	// Start workers
	for i := 0; i < workerCount; i++ {
		go func() {
			for recipient := range jobs {
				sendMail(sender, recipient, subject, htmlContent, &mailer, debug, success, fail)
			}
		}()
	}

	// Send jobs to workers
	for _, recipient := range recipients {
		jobs <- recipient
	}
	close(jobs)

	for i := 0; i < len(recipients); i++ {
		select {
		case msg := <-success:
			if debug {
				fmt.Printf("\rEmailed recipient %d of %d...", i+1, len(recipients))
			} else {
				bytes, err := msg.Bytes()
				if err != nil {
					fmt.Printf("Error parsing email: %v", err)
				}
				fmt.Printf("%s\n\n\n", string(bytes))
			}
		case err := <-fail:
			fmt.Fprintln(os.Stderr, "\nError sending email:", err.Error())
			os.Exit(2)
		}
	}
	fmt.Println()

	w.Write([]byte("Newsletter Successfully Sent!"))
}