Add blank plugins: email, telegram

Author: kshvakov

src/app/app.go

@@ -0,0 +1,78 @@
+package app
+
+import (
+	log "github.com/Sirupsen/logrus"
+	"github.com/jmoiron/sqlx"
+	_ "github.com/lib/pq"
+	"github.com/postgres-ci/notifier/src/common"
+	"github.com/postgres-ci/notifier/src/plugins/email"
+	"github.com/postgres-ci/notifier/src/plugins/telegram"
+
+	"os"
+	"runtime"
+	"time"
+)
+
+const (
+	MaxOpenConns    = 5
+	MaxIdleConns    = 2
+	ConnMaxLifetime = time.Hour
+)
+
+func New(config common.Config) *app {
+
+	connect, err := sqlx.Connect("postgres", config.Connect.DSN())
+
+	if err != nil {
+
+		log.Fatalf("Could not connect to database server: %v", err)
+	}
+
+	log.Debugf("Connect to postgresql server. DSN(%s)", config.Connect.DSN())
+
+	connect.SetMaxOpenConns(MaxOpenConns)
+	connect.SetMaxIdleConns(MaxIdleConns)
+	connect.SetConnMaxLifetime(ConnMaxLifetime)
+
+	app := app{
+		config:  config,
+		connect: connect,
+		plugins: []plugin{
+			email.New(config),
+			telegram.New(config.Telegram.Token, connect),
+		},
+	}
+
+	return &app
+}
+
+type plugin interface {
+	Send(common.Build) error
+}
+
+type app struct {
+	config  common.Config
+	connect *sqlx.DB
+	plugins []plugin
+	debug   bool
+}
+
+func (a *app) SetDebugMode() {
+
+	a.debug = true
+}
+
+func (a *app) Run() {
+
+	log.Info("Postgres-CI notifier started")
+	log.Debugf("Runtime version: %s. Pid: %d", runtime.Version(), os.Getpid())
+
+	if a.debug {
+
+		go a.debugInfo()
+	}
+
+	t := make(chan struct{})
+
+	<-t
+}

src/app/app_debug_info.go

@@ -0,0 +1,42 @@
+package app
+
+import (
+	log "github.com/Sirupsen/logrus"
+	_ "net/http/pprof"
+
+	"net/http"
+	"runtime"
+	"time"
+)
+
+func (a *app) debugInfo() {
+
+	go func() {
+
+		log.Info("Worker running in debug mode")
+		log.Infof("pprof: http://127.0.0.1:8081/debug/pprof")
+		log.Error(http.ListenAndServe("127.0.0.1:8081", nil))
+	}()
+
+	tick := time.Tick(time.Minute)
+
+	for {
+
+		var memStats runtime.MemStats
+
+		runtime.ReadMemStats(&memStats)
+
+		log.Debugf(
+			"gorutines: %d, num gc: %d, alloc: %d, mallocs: %d, frees: %d, heap alloc: %d, stack inuse: %d",
+			runtime.NumGoroutine(),
+			memStats.NumGC,
+			memStats.Alloc,
+			memStats.Mallocs,
+			memStats.Frees,
+			memStats.HeapAlloc,
+			memStats.StackInuse,
+		)
+
+		<-tick
+	}
+}

src/app/worker_test.go renamed to src/app/app_test.go

@@ -0,0 +1,50 @@
+package common
+
+import (
+	"encoding/json"
+	"fmt"
+	"time"
+)
+
+type Build struct {
+	ID                int32      `db:"build_id"`
+	Status            string     `db:"status"`
+	Branch            string     `db:"branch"`
+	Error             string     `db:"build_error"`
+	CreatedAt         time.Time  `db:"build_created_at"`
+	StartedAt         time.Time  `db:"build_started_at"`
+	FinishedAt        time.Time  `db:"build_finished_at"`
+	CommitSHA         string     `db:"commit_sha"`
+	CommitMessage     string     `db:"commit_message"`
+	CommittedAt       time.Time  `db:"committed_at"`
+	CommitterName     string     `db:"committer_name"`
+	CommitterEmail    string     `db:"committer_email"`
+	CommitAuthorName  string     `db:"commit_author_name"`
+	CommitAuthorEmail string     `db:"commit_author_email"`
+	SendTo            recipients `db:"send_to"`
+}
+
+type recipient struct {
+	Name   string `json:"user_name"`
+	Method string `json:"notify_method"`
+	TextID string `json:"notify_text_id"`
+	IntID  int64  `json:"notify_int_id"`
+}
+
+type recipients []recipient
+
+func (r *recipients) Scan(src interface{}) error {
+
+	var source []byte
+
+	switch src.(type) {
+	case string:
+		source = []byte(src.(string))
+	case []byte:
+		source = src.([]byte)
+	default:
+		return fmt.Errorf("Incompatible type for recipients")
+	}
+
+	return json.Unmarshal(source, r)
+}

src/common/build.go

@@ -0,0 +1,16 @@
+package email
+
+import (
+	"github.com/postgres-ci/notifier/src/common"
+)
+
+func New(config common.Config) *email {
+	return &email{}
+}
+
+type email struct {
+}
+
+func (e *email) Send(common.Build) error {
+	return nil
+}

src/plugins/email/email.go

@@ -0,0 +1,22 @@
+package telegram
+
+import (
+	"github.com/jmoiron/sqlx"
+	"github.com/postgres-ci/notifier/src/common"
+)
+
+func New(token string, connect *sqlx.DB) *bot {
+	return &bot{
+		token:   token,
+		connect: connect,
+	}
+}
+
+type bot struct {
+	token   string
+	connect *sqlx.DB
+}
+
+func (b *bot) Send(common.Build) error {
+	return nil
+}

src/plugins/telegram/bot.go

@@ -19,6 +19,12 @@
 			"revision": "ee1442bda7bd1b6a84e913bdb421cb1874ec629d",
 			"branch": "master"
 		},
+		{
+			"importpath": "github.com/tucnak/telebot",
+			"repository": "https://github.com/tucnak/telebot",
+			"revision": "68ad7fd5dcdbfdc4f6513e0195ad6edb2c3e0573",
+			"branch": "master"
+		},
 		{
 			"importpath": "gopkg.in/airbrake/gobrake.v2",
 			"repository": "https://gopkg.in/airbrake/gobrake.v2",

vendor/manifest

@@ -2,6 +2,7 @@ package main
 
 import (
 	log "github.com/Sirupsen/logrus"
+	"github.com/postgres-ci/notifier/src/app"
 	"github.com/postgres-ci/notifier/src/common"
 
 	"flag"
@@ -87,5 +88,12 @@ func main() {
 		log.SetLevel(config.LogLevel())
 	}
 
-	log.Debug(config)
+	app := app.New(config)
+
+	if debug {
+
+		app.SetDebugMode()
+	}
+
+	app.Run()
 }