feat: added GORM's CreateConstraints function and Constraint struct

Author: ralvarezdev

gorm/constraint.go

@@ -0,0 +1,49 @@
+package gorm
+
+import (
+	godatabases "github.com/ralvarezdev/go-databases"
+	"gorm.io/gorm"
+)
+
+type (
+	// Constraint struct
+	Constraint struct {
+		model interface{}
+		field string
+	}
+)
+
+// NewConstraint creates a new constraint
+func NewConstraint(model interface{}, field string) *Constraint {
+	return &Constraint{
+		model: model,
+		field: field,
+	}
+}
+
+// CreateConstraint creates a new constraint
+func CreateConstraint(database *gorm.DB, constraint *Constraint) error {
+	// Check if the database or the constraint is nil
+	if database == nil {
+		return godatabases.ErrNilDatabase
+	}
+	if constraint == nil {
+		return ErrNilConstraint
+	}
+
+	// Create the constraint
+	return database.Migrator().CreateConstraint(
+		constraint.model,
+		constraint.field,
+	)
+}
+
+// CreateConstraints creates new constraints
+func CreateConstraints(database *gorm.DB, constraints []*Constraint) error {
+	for _, constraint := range constraints {
+		if err := CreateConstraint(database, constraint); err != nil {
+			return err
+		}
+	}
+	return nil
+}

gorm/errors.go

@@ -5,5 +5,6 @@ import (
 )
 
 var (
-	ErrNilJoinField = errors.New("join field cannot be nil")
+	ErrNilJoinField  = errors.New("join field cannot be nil")
+	ErrNilConstraint = errors.New("constraint cannot be nil")
 )

gorm/join_table.go

@@ -27,7 +27,7 @@ func NewJoinField(
 	}
 }
 
-// SetupJoinTable setups the join table between two models
+// SetupJoinTable setups the join table
 func SetupJoinTable(
 	database *gorm.DB,
 	joinField *JoinField,
@@ -47,17 +47,13 @@ func SetupJoinTable(
 	)
 }
 
-// SetupJoinTables setups the join tables between multiple models
+// SetupJoinTables setups the join tables
 func SetupJoinTables(
 	database *gorm.DB,
 	joinFields []*JoinField,
-) error {
+) (err error) {
 	for _, joinField := range joinFields {
-		err := SetupJoinTable(
-			database,
-			joinField,
-		)
-		if err != nil {
+		if err = SetupJoinTable(database, joinField); err != nil {
 			return err
 		}
 	}