Skip to content

Commit

Permalink
feat: implemented night batch (#16)
Browse files Browse the repository at this point in the history
  • Loading branch information
kokoichi206 committed Nov 16, 2021
1 parent 7ad89d5 commit 2663ef6
Show file tree
Hide file tree
Showing 4 changed files with 76 additions and 1 deletion.
66 changes: 66 additions & 0 deletions db/data/update_blog.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
package data

import (
"fmt"

models "web/db/my_models"

"github.com/volatiletech/sqlboiler/v4/boil"
"github.com/volatiletech/sqlboiler/v4/queries/qm"
)

func UpdateBlogs() {
DB, err := DbInit()
if err != nil {
fmt.Println("connection error")
}
defer DB.Close()

if err := DB.Ping(); err != nil {
fmt.Println("PingError")
}

groups := []string{"nogizaka", "sakurazaka", "hinatazaka"}
for _, group := range groups {
blogs := LoadBlogInfoFile(group)
for _, blog := range blogs {

name := blog.Name
// 名前が blog に使われているものと違う場合、ここで修正を加える
if blog.Name == "saitoasuka" {
name = "saitouasuka"
}

mId, err := FindMemberByNameEn(name)
if err != nil {
fmt.Println(blog.Name + "が DB に見つかりませんでした")
continue
}

g, err := FindBlogByMemberId(mId)
if err != nil {
fmt.Printf("member_id %d の blog が見つかりませんでした?", mId)
fmt.Println(err)
continue
}

// Update information
g.LastUpdatedAt = blog.LastUpdatedAt
fmt.Println(g)
if _, err = g.Update(Ctx, DB, boil.Infer()); err != nil {
fmt.Printf("member_id %d の blog の更新に失敗しました", mId)
fmt.Println(err)
}

}
}
}

func FindBlogByMemberId(id int) (*models.Blog, error) {

g, err := models.Blogs(
qm.Where("member_id = ?", id),
).One(Ctx, DB)

return g, err
}
2 changes: 1 addition & 1 deletion db/data/utils.go
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ func LoadFormationFile(group string) []*Formation {

func LoadBlogInfoFile(group string) []*Blog {
// 実行ファイルからの相対パスなのでこの方式にしてる
raw, err := ioutil.ReadFile("./db/data/blogs/" + group + ".json")
raw, err := ioutil.ReadFile("/home/ubuntu/work/go/web/db/data/blogs/" + group + ".json")
if err != nil {
fmt.Println(err.Error())
os.Exit(1)
Expand Down
Binary file added night_batch
Binary file not shown.
9 changes: 9 additions & 0 deletions night_batch.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
package main

import (
"web/db/data"
)

func main() {
data.UpdateBlogs()
}

0 comments on commit 2663ef6

Please sign in to comment.