mahdium/CatsOfMastodonGo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add periodic post fetching and update default Mastodon instance to mstdn.party

Browse Source
This commit is contained in:
Mohammad Mahdi
2025-05-13 22:47:52 +03:30
parent 117fe3dd34
commit f9ec882cd9
2 changed files with 46 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
cmd/main.go
View File

@@ -15,41 +15,51 @@ func main() {
// Setup AppContext
var appContext = helpers.SetupAppContext()
// Get posts
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
var posts []models.Post = nil
err, posts := appContext.PostService.GetPostsFromApi(ctx, appContext.Tag, appContext.Instance)
if err != nil {
log.Fatal(err)
ticker := time.NewTicker(1 * time.Minute)
runFetchPosts := func() {
// Get posts
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
var posts []models.Post = nil
err, posts := appContext.PostService.GetPostsFromApi(ctx, appContext.Tag, appContext.Instance)
if err != nil {
log.Fatal(err)
}
var existingPostIds = appContext.PostService.GetExistingPostIds()
var existingAccountIds = appContext.PostService.GetExistingAccountIds()
var newPosts = appContext.PostService.GetNewPosts(existingPostIds, posts)
var newAccounts = appContext.PostService.GetNewAccounts(existingAccountIds, newPosts)
// Save to database
log.Printf("Number of existing posts: %d, existing accounts: %d, new posts: %d, new accounts: %d\n", len(existingPostIds), len(existingAccountIds), len(newPosts), len(newAccounts))
// Additional logging
if newAccounts != nil {
log.Println("Number of inserted accounts: ", appContext.PostService.InsertNewAccounts(newAccounts))
} else {
log.Print(" - No new accounts inserted")
}
if newPosts != nil {
log.Print(" - Number of inserted posts: ", appContext.PostService.InsertNewPosts(newPosts))
} else {
log.Print(" - No new posts inserted")
}
}
var existingPostIds = appContext.PostService.GetExistingPostIds()
var existingAccountIds = appContext.PostService.GetExistingAccountIds()
var newPosts = appContext.PostService.GetNewPosts(existingPostIds, posts)
var newAccounts = appContext.PostService.GetNewAccounts(existingAccountIds, newPosts)
// Save to database
log.Println("Number of existing posts: ", len(existingPostIds))
log.Println("Number of existing accounts: ", len(existingAccountIds))
log.Println("Number of new posts: ", len(newPosts))
log.Println("Number of new accounts: ", len(newAccounts))
// Additional logging
if newAccounts != nil {
log.Println("Number of inserted accounts: ", appContext.PostService.InsertNewAccounts(newAccounts))
} else {
log.Println("No new accounts inserted")
}
if newPosts != nil {
log.Println("Number of inserted posts: ", appContext.PostService.InsertNewPosts(newPosts))
} else {
log.Println("No new posts inserted")
}
go func() {
for range ticker.C {
runFetchPosts()
}
}()
// https://seefnasrul.medium.com/create-your-first-go-rest-api-with-jwt-authentication-in-gin-framework-dbe5bda72817
engine := gin.Default()
r := server.SetupRouter(engine)
r.Run(":8080")
err := r.Run(":8080")
if err != nil {
log.Fatal(err)
}
}

9
internal/helpers/SetupAppContext.go
View File

@@ -14,14 +14,13 @@ func SetupAppContext() *internal.AppContext {
// Setup AppContext
instance := os.Getenv("INSTANCE")
if instance == "" {
instance = "https://haminoa.net"
instance = "https://mstdn.party"
}
tag := os.Getenv("TAG")
if tag == "" {
tag = "catsofmastodon"
}
// Setup database
db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{Logger: logger.Default.LogMode(logger.Warn)})
if err != nil {
@@ -34,10 +33,10 @@ func SetupAppContext() *internal.AppContext {
// Inititlize AppContext
var appContext = &internal.AppContext{
Db: db,
Db: db,
PostService: postService,
Instance: instance,
Tag: tag,
Instance: instance,
Tag: tag,
}
return appContext