-
Notifications
You must be signed in to change notification settings - Fork 0
/
application.go
92 lines (74 loc) · 2.47 KB
/
application.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package application
import (
"encoding/json"
"log"
"net/http"
"github.com/akmittal/webstats/pkg/config"
"github.com/akmittal/webstats/pkg/crawl"
"github.com/akmittal/webstats/pkg/db"
"github.com/akmittal/webstats/pkg/image"
"github.com/akmittal/webstats/pkg/site"
"github.com/go-chi/chi"
"github.com/joho/godotenv"
"github.com/akmittal/webstats/pkg/router"
"github.com/akmittal/webstats/pkg/server"
)
type Application struct {
DB *db.DB
Cfg *config.Config
Router *router.Router
Server *server.Server
}
func Get() (*Application, error) {
if err := godotenv.Load(); err != nil {
log.Println("failed to load env vars")
}
cfg := config.Get()
db, err := db.Get(cfg.GetDBConnStr())
if err != nil {
return nil, err
}
router, err := router.Get()
server := server.Get(cfg.GetAppHost(), router)
db.Client.AutoMigrate(&crawl.Stats{}, &crawl.Crawl{}, &site.Site{}, &image.Image{})
return &Application{
DB: db,
Cfg: cfg,
Router: router,
Server: server,
}, nil
}
func (a *Application) Start() error {
a.RegisterRoutes()
return a.Server.Start()
}
func (a *Application) RunCrawler() {
crawl.RunCrawler(a.DB.Client)
}
func (a *Application) RegisterRoutes() {
// a.Router.Get("/api/public", user.UserController())
a.Router.Get("/api", func(rw http.ResponseWriter, r *http.Request) {
var MaxID uint
a.DB.Client.Select("MAX(id)").Table("crawls").First(&MaxID)
var stats []crawl.Stats
a.DB.Client.Preload("Site").Preload("Images").Model(&crawl.Stats{}).Where(&crawl.Stats{CrawlerRefer: MaxID}).Find(&stats)
json.NewEncoder(rw).Encode(stats)
})
a.Router.Get("/api/stats/{type}", func(rw http.ResponseWriter, r *http.Request) {
paramaType := chi.URLParam(r, "type")
var MaxID uint
a.DB.Client.Model(crawl.Crawl{}).Select("MAX(id)").First(&MaxID)
var stats []struct {
Total int
Name string
}
if paramaType == "images" {
a.DB.Client.Model(&crawl.Stats{}).Select(" images.type as name", "count(*) as total").Where("images.type <> ''").Joins("inner join images on images.stat_refer=stats.id").Group("images.type").Find(&stats)
} else if paramaType == "ssg" {
a.DB.Client.Model(&crawl.Stats{}).Select(paramaType+" as name", "count(*) as total").Where(paramaType + " <> ''").Where(paramaType + " <> 'other'").Group(paramaType).Find(&stats)
} else {
a.DB.Client.Model(&crawl.Stats{}).Select(paramaType+" as name", "count(*) as total").Where(paramaType + " <> ''").Group(paramaType).Find(&stats)
}
json.NewEncoder(rw).Encode(stats)
})
}