update auth.go, database.go, docs.go, forum.go, go.mod, go.sum, main.go, post_schema.go, role.go, server.go, service.go, swagger.json, swagger.yaml, thread_schema.go

pkg/repository/database.go

@@ -0,0 +1,46 @@
+package repository
+
+import (
+	"esforum"
+	"fmt"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+type Config struct {
+	Host     string
+	Port     string
+	Username string
+	Password string
+	DBName   string
+}
+
+func NewDB(cfg Config) (*gorm.DB, error) {
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		cfg.Username,
+		cfg.Password,
+		cfg.Host,
+		cfg.Port,
+		cfg.DBName)
+
+	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
+		Logger: logger.Default.LogMode(logger.Silent),
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	err = db.Set("gorm:table_options", "CHARSET=utf8mb4").AutoMigrate(&esforum.Post{})
+	if err != nil {
+		return nil, err
+	}
+
+	err = db.Set("gorm:table_options", "CHARSET=utf8mb4").AutoMigrate(&esforum.Thread{})
+	if err != nil {
+		return nil, err
+	}
+
+	return db, nil
+}

pkg/repository/post_schema.go

@@ -0,0 +1,71 @@
+package repository
+
+import (
+	"esforum"
+	"fmt"
+
+	"gorm.io/gorm"
+)
+
+type PostSchema struct {
+	db *gorm.DB
+}
+
+func NewPostSchema(db *gorm.DB) *PostSchema {
+	return &PostSchema{db: db}
+}
+
+func (r PostSchema) CreatePost(title string, content string, thread int64) (*esforum.Post, error) {
+	post := &esforum.Post{
+		Title:   title,
+		Content: content,
+		Thread:  thread,
+	}
+	err := r.db.Create(post).Error
+	return post, err
+}
+
+func (r PostSchema) GetPost(id int64) (*esforum.Post, error) {
+	post := esforum.Post{}
+	err := r.db.Find(&post, "id = ?", id).Error
+	if post.Id == -1 {
+		err = fmt.Errorf("unknown thread %d", id)
+	}
+	return &post, err
+}
+
+func (r PostSchema) DeletePost(id int64) error {
+	post := esforum.Post{}
+	err := r.db.Find(&post, "id = ?", id).Error
+	if post.Id == -1 {
+		err = fmt.Errorf("unknown thread %d", id)
+	}
+	r.db.Delete(post)
+	return err
+}
+
+func (r PostSchema) FindPostsByThread(thread int64, start, limit int) []int64 {
+	var result []esforum.Post
+	r.db.Find(&result, "thread = ?", thread).Limit(limit).Offset(start).Find(&result)
+	var ans = make([]int64, len(result))
+	for i, res := range result {
+		ans[i] = res.Id
+	}
+	return ans
+}
+
+func (r PostSchema) GetPosts(start, limit int) []int64 {
+	var result []esforum.Post
+	r.db.Select("id").Limit(limit).Offset(start).Find(&result)
+	var ans = make([]int64, len(result))
+	for i, res := range result {
+		ans[i] = res.Id
+	}
+	return ans
+}
+
+func (r PostSchema) GetPostsCount() int64 {
+	var count int64 = 0
+	r.db.Model(&esforum.Post{}).Count(&count)
+	return count
+}

pkg/repository/thread_schema.go

@@ -0,0 +1,68 @@
+package repository
+
+import (
+	"esforum"
+	"fmt"
+
+	"gorm.io/gorm"
+)
+
+type ThreadSchema struct {
+	db *gorm.DB
+}
+
+func NewThreadSchema(db *gorm.DB) *ThreadSchema {
+	return &ThreadSchema{db: db}
+}
+
+func (r ThreadSchema) CreateThread(title string) (*esforum.Thread, error) {
+	thread := &esforum.Thread{
+		Title: title,
+	}
+	err := r.db.Create(thread).Error
+	return thread, err
+}
+
+func (r ThreadSchema) GetThread(id int64) (*esforum.Thread, error) {
+	thread := esforum.Thread{}
+	err := r.db.Find(&thread, "id = ?", id).Error
+	if thread.Id == -1 {
+		err = fmt.Errorf("unknown thread %d", id)
+	}
+	return &thread, err
+}
+
+func (r ThreadSchema) FindThread(title string) (*esforum.Thread, error) {
+	thread := esforum.Thread{}
+	err := r.db.Find(&thread, "title = ?", title).Error
+	if thread.Id == -1 {
+		err = fmt.Errorf("unknown thread %s", title)
+	}
+	return &thread, err
+}
+
+func (r ThreadSchema) DeleteThread(id int64) error {
+	thread := esforum.Thread{}
+	err := r.db.Find(&thread, "id = ?", id).Error
+	if thread.Id == -1 {
+		err = fmt.Errorf("unknown thread %d", id)
+	}
+	r.db.Delete(thread)
+	return err
+}
+
+func (r ThreadSchema) GetThreads(start, limit int) []int64 {
+	var result []esforum.Thread
+	r.db.Select("id").Limit(limit).Offset(start).Find(&result)
+	var ans = make([]int64, len(result))
+	for i, res := range result {
+		ans[i] = res.Id
+	}
+	return ans
+}
+
+func (r ThreadSchema) GetThreadsCount() int64 {
+	var count int64 = 0
+	r.db.Model(&esforum.Thread{}).Count(&count)
+	return count
+}