DB new user creation

Author: arnabsen1729

docker-compose.yml

@@ -32,4 +32,4 @@ services:
 
 networks:
   postgres:
-    driver: bridge
+    driver: bridge

go.mod

@@ -10,4 +10,5 @@ require (
 	github.com/gofiber/fiber/v2 v2.1.3
 	github.com/gofiber/jwt/v2 v2.0.1
 	github.com/joho/godotenv v1.3.0
+	golang.org/x/crypto v0.0.0-20201112155050-0c6587e931a9
 )

go.sum

@@ -114,6 +114,8 @@ golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPh
 golang.org/x/crypto v0.0.0-20200820211705-5c72a883971a/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
 golang.org/x/crypto v0.0.0-20201012173705-84dcc777aaee h1:4yd7jl+vXjalO5ztz6Vc1VADv+S/80LGJmyl1ROJ2AI=
 golang.org/x/crypto v0.0.0-20201012173705-84dcc777aaee/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
+golang.org/x/crypto v0.0.0-20201112155050-0c6587e931a9 h1:umElSU9WZirRdgu2yFHY0ayQkEnKiOC1TtM3fWXFnoU=
+golang.org/x/crypto v0.0.0-20201112155050-0c6587e931a9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
 golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
 golang.org/x/exp v0.0.0-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
 golang.org/x/exp v0.0.0-20200821190819-94841d0725da/go.mod h1:3jZMyOhIsHpP37uCMkUooju7aAi5cS1Q23tOzKc+0MU=

migrations/1_init.go

@@ -8,12 +8,21 @@ import (
 
 func init() {
 	migrations.MustRegisterTx(func(db migrations.DB) error {
-		fmt.Println("creating table my_table...")
-		_, err := db.Exec(`CREATE TABLE my_table()`)
+		fmt.Println("creating table users...")
+		_, err := db.Exec(`CREATE TABLE users (
+			id serial PRIMARY KEY,
+			first_name varchar(30),
+			last_name varchar(30),
+			email varchar(50) UNIQUE NOT NULL,
+			password varchar NOT NULL,
+			handle varchar(30) UNIQUE NOT NULL,
+			rating integer DEFAULT 1300,
+			createdAt timestamp
+		);`)
 		return err
 	}, func(db migrations.DB) error {
-		fmt.Println("dropping table my_table...")
-		_, err := db.Exec(`DROP TABLE my_table`)
+		fmt.Println("dropping table users...")
+		_, err := db.Exec(`DROP TABLE users`)
 		return err
 	})
 }

sample.env

@@ -4,4 +4,4 @@ DB_PASSWD=
 PGADMIN_EMAIL=
 PGADMIN_PASSWD=
 
-SECRET_KEY=
+SECRET_KEY=

server/handler/user_handler.go

@@ -7,6 +7,7 @@ import (
 	jwt "github.com/dgrijalva/jwt-go"
 	"github.com/gofiber/fiber/v2"
 	"github.com/joho/godotenv"
+	models "github.com/raydwaipayan/onlinejudge-server/server/models"
 	"github.com/raydwaipayan/onlinejudge-server/server/types"
 )
 
@@ -17,7 +18,7 @@ func Register(c *fiber.Ctx) error {
 	if err := c.BodyParser(u); err != nil {
 		return c.SendStatus(fiber.StatusBadRequest)
 	}
-
+	u.Create(models.DBConfigURL)
 	return c.SendStatus(fiber.StatusOK)
 }
 

server/models/models.go

@@ -0,0 +1,31 @@
+package models
+
+import (
+	"context"
+	"os"
+
+	pg "github.com/go-pg/pg/v10"
+	"github.com/joho/godotenv"
+)
+
+// DBConfigURL  config url of database
+var DBConfigURL *pg.DB
+
+// InitDb to initialise database
+func InitDb() {
+	godotenv.Load()
+
+	DBConfigURL = pg.Connect(&pg.Options{
+		Addr:     ":5432",
+		User:     os.Getenv("DB_USER"),
+		Password: os.Getenv("DB_PASSWD"),
+		Database: os.Getenv("DB_NAME"),
+	})
+
+	//Check if the database is running
+	ctx := context.Background()
+
+	if err := DBConfigURL.Ping(ctx); err != nil {
+		panic(err)
+	}
+}

server/server.go

@@ -4,6 +4,7 @@ import (
 	"github.com/gofiber/fiber/v2"
 	"github.com/gofiber/fiber/v2/middleware/logger"
 	"github.com/gofiber/fiber/v2/middleware/recover"
+	models "github.com/raydwaipayan/onlinejudge-server/server/models"
 	router "github.com/raydwaipayan/onlinejudge-server/server/router"
 )
 
@@ -12,5 +13,6 @@ func main() {
 	app.Use(recover.New(), logger.New())
 
 	router.SetupRoutes(app)
+	models.InitDb()
 	app.Listen(":3000")
 }

server/types/db.go

@@ -0,0 +1,42 @@
+package types
+
+import (
+	"log"
+
+	pg "github.com/go-pg/pg/v10"
+	"golang.org/x/crypto/bcrypt"
+)
+
+//Create create a new user and insert into db
+func (u *User) Create(db *pg.DB) error {
+	_, err := db.Model(u).Insert()
+	return err
+}
+
+//Update update existing user
+func (u *User) Update(db *pg.DB) error {
+	_, err := db.Model(u).Update()
+	return err
+}
+
+// CheckUserExists to check if user is present
+func (u *User) CheckUserExists(db *pg.DB) (bool, error) {
+	user := new(User)
+	err := db.Model(user).Table("users").Where("users.email = ?", u.Email).Limit(1).Select()
+	encryptionErr := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(u.Password))
+	if encryptionErr == nil {
+		u.FirstName = user.FirstName
+		u.LastName = user.LastName
+		return true, nil
+	}
+	if err != nil {
+		log.Println(err)
+	}
+	return false, err
+}
+
+//Delete delete user
+func (u *User) Delete(db *pg.DB) error {
+	_, err := db.Model(u).Delete()
+	return err
+}