Users

bigv/models.go

@@ -0,0 +1,57 @@
+package bigv
+
+import (
+	"errors"
+	"tange/common"
+
+	"golang.org/x/crypto/bcrypt"
+)
+
+type UserModel struct {
+	ID           uint   `gorm:"primary_key"`
+	Username     string `gorm:"column:username"`
+	PasswordHash string `gorm:"column:password;not null"`
+}
+
+func AutoMigrate() {
+	db := common.GetDB()
+
+	db.AutoMigrate(&UserModel{})
+}
+
+// 	err := userModel.setPassword("password0")
+func (u *UserModel) setPassword(password string) error {
+	if len(password) == 0 {
+		return errors.New("password should not be empty!")
+	}
+	bytePassword := []byte(password)
+	// Make sure the second param `bcrypt generator cost` between [4, 32)
+	passwordHash, _ := bcrypt.GenerateFromPassword(bytePassword, bcrypt.DefaultCost)
+	u.PasswordHash = string(passwordHash)
+	return nil
+}
+
+func (u *UserModel) checkPassword(password string) error {
+	bytePassword := []byte(password)
+	byteHashedPassword := []byte(u.PasswordHash)
+	return bcrypt.CompareHashAndPassword(byteHashedPassword, bytePassword)
+}
+
+func FindOneUser(condition interface{}) (UserModel, error) {
+	db := common.GetDB()
+	var model UserModel
+	err := db.Where(condition).First(&model).Error
+	return model, err
+}
+
+func SaveOne(data interface{}) error {
+	db := common.GetDB()
+	err := db.Save(data).Error
+	return err
+}
+
+func (model *UserModel) Update(data interface{}) error {
+	db := common.GetDB()
+	err := db.Model(model).Update(data).Error
+	return err
+}

cmd/initdb.go

@@ -0,0 +1,34 @@
+package cmd
+
+import (
+	"tange/bigv"
+	"tange/common"
+
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	//	"github.com/spf13/viper"
+)
+
+//var port string
+
+// serveCmd represents the serve command
+var initdbCmd = &cobra.Command{
+	Use:   "initdb",
+	Short: "Initialize database",
+	Long:  `No longer description Needed.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		initdb()
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(initdbCmd)
+
+}
+
+func initdb() {
+	log.Info("Initializing Database")
+	db := common.Init()
+	bigv.AutoMigrate()
+	defer db.Close()
+}

cmd/serve.go

@@ -3,14 +3,17 @@ package cmd
 import (
 	"fmt"
 	"net/http"
+	"tange/common"
 
+	"github.com/jinzhu/gorm"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 	//	"github.com/spf13/viper"
 )
 
 var port string
+var db *gorm.DB
 
 // serveCmd represents the serve command
 var serveCmd = &cobra.Command{
@@ -42,6 +45,7 @@ func handler(RespW http.ResponseWriter, ReqR *http.Request) {
 }
 
 func serve(port string) {
+	db = common.Init()
 	log.Info("I'll serve, RelX DUDE")
 	portString := ":" + port
 	log.Info("On port " + portString)

common/dasabase.go

@@ -0,0 +1,42 @@
+package common
+
+import (
+	"fmt"
+
+	"github.com/jinzhu/gorm"
+	_ "github.com/jinzhu/gorm/dialects/postgres"
+	"github.com/spf13/viper"
+)
+
+type Database struct {
+	*gorm.DB
+}
+
+var DB *gorm.DB
+
+func Init() *gorm.DB {
+
+	// var dbPort string
+	// var dbHost string
+	// var postgresUser string
+	// var postgresPassword string
+	dbPort := viper.GetString("db_port")
+	dbHost := viper.GetString("db_host")
+	dbName := viper.GetString("db_name")
+	postgresUser := viper.GetString("postgres_user")
+	postgresPassword := viper.GetString("postgres_password")
+
+	db, err := gorm.Open("postgres", "host="+dbHost+" port="+dbPort+" user="+postgresUser+" dbname="+dbName+" password="+postgresPassword)
+	if err != nil {
+		fmt.Println("db err: ", err)
+	}
+
+	db.DB().SetMaxIdleConns(10)
+	//db.LogMode(true)
+	DB = db
+	return DB
+}
+
+func GetDB() *gorm.DB {
+	return DB
+}