Browse Source

Pools limit concurrent nums

pull/33/head
Unknown 11 years ago
parent
commit
6f68620860
  1. 3
      README.md
  2. 2
      conf/app.ini
  3. 2
      gogs.go
  4. 2
      models/user.go
  5. 4
      modules/base/conf.go
  6. 6
      modules/mailer/mail.go
  7. 44
      modules/mailer/mailer.go
  8. 3
      routers/user/user.go

3
README.md

@ -5,7 +5,7 @@ Gogs(Go Git Service) is a GitHub-like clone in the Go Programming Language.
Since we choose to use pure Go implementation of Git manipulation, Gogs certainly supports **ALL platforms** that Go supports, including Linux, Max OS X, and Windows with **ZERO** dependency.
##### Current version: 0.1.0 Alpha
##### Current version: 0.1.1 Alpha
## Purpose
@ -26,6 +26,7 @@ There are some very good products in this category such as [gitlab](http://gitla
- User profile page.
- Repository viewer.
- Gravatar support.
- Mail service(register).
- Supports MySQL and PostgreSQL.
## Installation

2
conf/app.ini

@ -39,6 +39,8 @@ REGISTER_EMAIL_CONFIRM = false
[mailer]
ENABLED = false
; Buffer length of channel, keep it as it is if you don't know what it is.
SEND_BUFFER_LEN = 10
; Name displayed in mail title
SUBJECT = %(APP_NAME)s
; Mail server

2
gogs.go

@ -20,7 +20,7 @@ import (
// Test that go1.1 tag above is included in builds. main.go refers to this definition.
const go11tag = true
const APP_VER = "0.1.0.0319.1"
const APP_VER = "0.1.1.0320.1"
func init() {
base.AppVer = APP_VER

2
models/user.go

@ -52,7 +52,6 @@ type User struct {
Website string
IsActive bool
Rands string `xorm:"VARCHAR(10)"`
Expired time.Time
Created time.Time `xorm:"created"`
Updated time.Time `xorm:"updated"`
}
@ -125,7 +124,6 @@ func RegisterUser(user *User) (*User, error) {
user.LowerName = strings.ToLower(user.Name)
user.Avatar = base.EncodeMd5(user.Email)
user.AvatarEmail = user.Email
user.Expired = time.Now().Add(3 * 24 * time.Hour)
user.Rands = GetUserSalt()
if err = user.EncodePasswd(); err != nil {
return nil, err

4
modules/base/conf.go

@ -91,9 +91,11 @@ func newLogService() {
case "console":
config = fmt.Sprintf(`{"level":%s}`, level)
case "file":
logPath := Cfg.MustValue(modeSec, "FILE_NAME", "log/gogs.log")
os.MkdirAll(path.Dir(logPath), os.ModePerm)
config = fmt.Sprintf(
`{"level":%s,"filename":%s,"rotate":%v,"maxlines":%d,"maxsize",%d,"daily":%v,"maxdays":%d}`, level,
Cfg.MustValue(modeSec, "FILE_NAME", "log/gogs.log"),
logPath,
Cfg.MustBool(modeSec, "LOG_ROTATE", true),
Cfg.MustInt(modeSec, "MAX_LINES", 1000000),
1<<uint(Cfg.MustInt(modeSec, "MAX_SIZE_SHIFT", 28)),

6
modules/mailer/mail.go

@ -62,8 +62,7 @@ func SendRegisterMail(r *middleware.Render, user *models.User) {
msg := NewMailMessage([]string{user.Email}, subject, body)
msg.Info = fmt.Sprintf("UID: %d, send register mail", user.Id)
// async send mail
SendAsync(msg)
SendAsync(&msg)
}
// Send email verify active email.
@ -83,6 +82,5 @@ func SendActiveMail(r *middleware.Render, user *models.User) {
msg := NewMailMessage([]string{user.Email}, subject, body)
msg.Info = fmt.Sprintf("UID: %d, send email verify mail", user.Id)
// async send mail
SendAsync(msg)
SendAsync(&msg)
}

44
modules/mailer/mailer.go

@ -38,8 +38,34 @@ func (m Message) Content() string {
return content
}
var mailQueue chan *Message
func init() {
mailQueue = make(chan *Message, base.Cfg.MustInt("mailer", "SEND_BUFFER_LEN", 10))
go processMailQueue()
}
func processMailQueue() {
for {
select {
case msg := <-mailQueue:
num, err := Send(msg)
tos := strings.Join(msg.To, "; ")
info := ""
if err != nil {
if len(msg.Info) > 0 {
info = ", info: " + msg.Info
}
log.Error(fmt.Sprintf("Async sent email %d succeed, not send emails: %s%s err: %s", num, tos, info, err))
return
}
log.Trace(fmt.Sprintf("Async sent email %d succeed, sent emails: %s%s", num, tos, info))
}
}
}
// Direct Send mail message
func Send(msg Message) (int, error) {
func Send(msg *Message) (int, error) {
log.Trace("Sending mails to: %s", strings.Join(msg.To, "; "))
host := strings.Split(base.MailService.Host, ":")
@ -82,21 +108,9 @@ func Send(msg Message) (int, error) {
}
// Async Send mail message
func SendAsync(msg Message) {
// TODO may be need pools limit concurrent nums
func SendAsync(msg *Message) {
go func() {
num, err := Send(msg)
tos := strings.Join(msg.To, "; ")
info := ""
if err != nil {
if len(msg.Info) > 0 {
info = ", info: " + msg.Info
}
// log failed
log.Error(fmt.Sprintf("Async sent email %d succeed, not send emails: %s%s err: %s", num, tos, info, err))
return
}
log.Trace(fmt.Sprintf("Async sent email %d succeed, sent emails: %s%s", num, tos, info))
mailQueue <- msg
}()
}

3
routers/user/user.go

@ -249,6 +249,9 @@ func Activate(ctx *middleware.Context) {
user.IsActive = true
user.Rands = models.GetUserSalt()
models.UpdateUser(user)
log.Trace("%s User activated: %s", ctx.Req.RequestURI, user.LowerName)
ctx.Session.Set("userId", user.Id)
ctx.Session.Set("userName", user.Name)
ctx.Redirect("/", 302)

Loading…
Cancel
Save