Init stable version

Author: numq

.gitignore

@@ -0,0 +1,108 @@
+### Go ###
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go debug -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+### Go Patch ###
+/vendor/
+/Godeps/
+
+### Goland ###
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio, WebStorm, Rider and Goland
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/usage.statistics.xml
+.idea/**/dictionaries
+.idea/**/shelf
+
+# AWS User-specific
+.idea/**/aws.xml
+
+# Generated files
+.idea/**/contentModel.xml
+
+# Sensitive or high-churn files
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+
+# Gradle
+.idea/**/gradle.xml
+.idea/**/libraries
+
+# Gradle and Maven with auto-import
+# When using Gradle or Maven with auto-import, you should exclude module files,
+# since they will be recreated, and may cause churn.  Uncomment if using
+# auto-import.
+# .idea/artifacts
+# .idea/compiler.xml
+# .idea/jarRepositories.xml
+# .idea/modules.xml
+# .idea/*.iml
+# .idea/modules
+# *.iml
+# *.ipr
+
+# CMake
+cmake-build-*/
+
+# Mongo Explorer plugin
+.idea/**/mongoSettings.xml
+
+# File-based project format
+*.iws
+
+# IntelliJ
+out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Cursive Clojure plugin
+.idea/replstate.xml
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+
+# Editor-based Rest Client
+.idea/httpRequests
+
+# Ignores the whole .idea folder and all .iml files
+.idea/
+
+# Android studio 3.1+ serialized cache file
+.idea/caches/build_file_checksums.ser
+
+### Redis ###
+# Ignore redis binary dump (dump.rdb) files
+
+*.rdb
+
+# GRPC
+
+*.pb.go

Dockerfile

@@ -0,0 +1,9 @@
+FROM golang:1.17
+
+WORKDIR /go/src/app
+COPY . .
+
+RUN go get -d -v ./...
+RUN go install -v ./...
+
+CMD ["protoc -I . proto/*.proto --go_out=plugins=grpc:.","app"]

config/config.go

@@ -0,0 +1,44 @@
+package config
+
+import (
+	"gopkg.in/yaml.v2"
+	"log"
+	"os"
+)
+
+type Config struct {
+	Database struct {
+		DbUri       string `yaml:"uri"`
+		DbName      string `yaml:"name"`
+		CollBoards  string `yaml:"collection_boards"`
+		CollThreads string `yaml:"collection_threads"`
+		CollPosts   string `yaml:"collection_posts"`
+	} `yaml:"database"`
+	Server struct {
+		ServerHost string `yaml:"host"`
+		ServerPort string `yaml:"port"`
+	} `yaml:"server"`
+}
+
+func NewConfig() *Config {
+
+	file, err := os.Open("config/config.yaml")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer func(file *os.File) {
+		err := file.Close()
+		if err != nil {
+			log.Fatal(err)
+		}
+	}(file)
+
+	var cfg Config
+	decoder := yaml.NewDecoder(file)
+	err = decoder.Decode(&cfg)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	return &cfg
+}

config/config.yaml

@@ -0,0 +1,9 @@
+database:
+  uri: mongodb://localhost:27017
+  name: imageboard
+  collection_boards: boards
+  collection_threads: threads
+  collection_posts: posts
+server:
+  host: localhost
+  port: 8080

data/db/database.go

@@ -0,0 +1,56 @@
+package db
+
+import (
+	"context"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"grpcImageboard/config"
+	"log"
+	"time"
+)
+
+type Database struct {
+	Boards  *mongo.Collection
+	Threads *mongo.Collection
+	Posts   *mongo.Collection
+}
+
+func NewDatabase(cfg config.Config, db *mongo.Database) Database {
+	return Database{
+		Boards:  db.Collection(cfg.Database.CollBoards),
+		Threads: db.Collection(cfg.Database.CollThreads),
+		Posts:   db.Collection(cfg.Database.CollPosts),
+	}
+}
+
+func Open(uri string) (*mongo.Client, context.Context, context.CancelFunc) {
+	return connect(uri)
+}
+
+func Close(client *mongo.Client, ctx context.Context, cancel context.CancelFunc) {
+	disconnect(client, ctx, cancel)
+}
+
+func connect(uri string) (*mongo.Client, context.Context, context.CancelFunc) {
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri).SetDirect(true).SetReplicaSet("rs0"))
+	if err != nil {
+		log.Println(err)
+	} else {
+		log.Println("Connected to DB")
+	}
+	return client, ctx, cancel
+}
+
+func disconnect(client *mongo.Client, ctx context.Context, cancel context.CancelFunc) {
+
+	defer cancel()
+
+	defer func() {
+		if err := client.Disconnect(ctx); err != nil {
+			log.Println(err)
+		} else {
+			log.Println("Disconnected from DB")
+		}
+	}()
+}

data/repository/board_repository.go

@@ -0,0 +1,152 @@
+package repository
+
+import (
+	"context"
+	"fmt"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"grpcImageboard/config"
+	"grpcImageboard/data/db"
+	"grpcImageboard/domain"
+)
+
+type boardRepository struct {
+	cfg config.Config
+	db  db.Database
+}
+
+func CreateBoardRepository(cfg config.Config, db db.Database) domain.BoardRepository {
+	return &boardRepository{cfg, db}
+}
+
+func (r *boardRepository) GetBoardsStream(ctx context.Context, changed func([]*domain.Board)) {
+	stream, err := r.db.Boards.Watch(ctx, mongo.Pipeline{})
+	if err != nil {
+		fmt.Println(err)
+	}
+	defer func(stream *mongo.ChangeStream, ctx context.Context) {
+		err := stream.Close(ctx)
+		if err != nil {
+			fmt.Println(err)
+		}
+	}(stream, ctx)
+	for {
+		select {
+		case <-ctx.Done():
+			err := stream.Close(ctx)
+			if err != nil {
+				fmt.Println(err)
+			}
+			return
+		default:
+			var data []*domain.Board
+			ok := stream.Next(ctx)
+			if !ok {
+				err := stream.Err()
+				if err != nil {
+					println(err)
+					return
+				}
+			}
+			if ok {
+				err := stream.Decode(data)
+				if err != nil {
+					println(err)
+					return
+				}
+				changed(data)
+			}
+		}
+	}
+
+	/*var result *bson.D
+	for stream.Next(ctx) {
+		if err := stream.Decode(&result); err != nil {
+			fmt.Println(err)
+		}
+		fmt.Println(result)
+
+		filter, opts := &bson.M{}, options.Find()
+		data, err := r.fetch(ctx, filter, opts)
+		if err != nil {
+			fmt.Println(err)
+		}
+		changed(data)
+	}*/
+}
+
+func (r *boardRepository) fetch(ctx context.Context, filter *bson.M, opts *options.FindOptions) ([]*domain.Board, error) {
+	cursor, err := r.db.Boards.Find(ctx, filter, opts)
+	if err != nil {
+		return nil, err
+	}
+	defer func(cursor *mongo.Cursor, ctx context.Context) {
+		err := cursor.Close(ctx)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+	}(cursor, ctx)
+	data := make([]*domain.Board, 0)
+	if err = cursor.All(ctx, &data); err != nil {
+		return nil, err
+	}
+	return data, nil
+}
+
+func (r *boardRepository) GetBoards(ctx context.Context, skip int64, limit int64) ([]*domain.Board, error) {
+	filter, opts := &bson.M{}, options.Find().SetSkip(skip).SetLimit(limit)
+	data, err := r.fetch(ctx, filter, opts)
+	if err != nil {
+		return nil, err
+	}
+	return data, nil
+}
+
+func (r *boardRepository) GetBoardById(ctx context.Context, id string) (*domain.Board, error) {
+	data := new(domain.Board)
+	if err := r.db.Boards.FindOne(ctx, bson.M{"_id": bson.M{"$eq": id}}).Decode(&data); err != nil {
+		return nil, err
+	}
+	return data, nil
+}
+
+func (r *boardRepository) CreateBoard(ctx context.Context, board *domain.Board) (string, error) {
+	board.Id = primitive.NewObjectID().Hex()
+	_, err := r.db.Boards.InsertOne(ctx, board)
+	if err != nil {
+		return "", err
+	}
+	return board.Id, nil
+}
+
+func (r *boardRepository) DeleteBoard(ctx context.Context, id string) (int64, error) {
+	session, err := r.db.Boards.Database().Client().StartSession()
+	if err != nil {
+		return 0, err
+	}
+	var count int64 = 0
+	if err = mongo.WithSession(ctx, session, func(sc mongo.SessionContext) error {
+		result, err := r.db.Boards.DeleteOne(ctx, bson.M{"_id": id})
+		if err != nil {
+			return err
+		}
+		count = result.DeletedCount
+		threadId, err := r.db.Threads.Find(ctx, bson.M{"board_id": id})
+		if err != nil {
+			return err
+		}
+		if _, err = r.db.Threads.DeleteMany(ctx, bson.M{"board_id": id}); err != nil {
+			return err
+		}
+		if _, err = r.db.Posts.DeleteMany(ctx, bson.M{"thread_id": threadId}); err != nil {
+			return err
+		}
+		return nil
+	}); err != nil {
+		return 0, err
+	}
+	return count, nil
+}