first modify

pkg/services/companyService

@@ -0,0 +1,96 @@
+package services
+
+import (
+	"context"
+	"errors"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"pkg.jojmatic.com/root/rpaas-core-service/pkg/logging"
+	"pkg.jojmatic.com/root/rpaas-core-service/pkg/models"
+	"time"
+)
+
+type CompanyService interface {
+	RegisterCompany(body *models.CompanyRegisterBody) (*models.CompanyInfo, error)
+	GetCompanyInfoByID(id string) (*models.CompanyInfo, error)
+	GetAllCompanies() ([]*models.CompanyInfo, error)
+}
+
+type CompanyServiceImpl struct {
+	companycollection *mongo.Collection
+	ctx               context.Context
+}
+
+func NewCompanyService(companycollection *mongo.Collection, ctx context.Context) CompanyService {
+	return &CompanyServiceImpl{
+		companycollection: companycollection,
+		ctx:               ctx,
+	}
+}
+
+func (c *CompanyServiceImpl) GetAllCompanies() ([]*models.CompanyInfo, error) {
+	var result []*models.CompanyInfo
+	cursor, err := c.companycollection.Find(c.ctx, bson.M{})
+	if err != nil {
+		return nil, err
+	}
+	if err = cursor.All(c.ctx, &result); err != nil {
+		for cursor.Next(c.ctx) {
+			var company models.CompanyInfo
+			err = cursor.Decode(&company)
+			if err != nil {
+				return nil, err
+			}
+			result = append(result, &company)
+		}
+	}
+	if err = cursor.Err(); err != nil {
+		return nil, err
+	}
+	cursor.Close(c.ctx)
+
+	if len(result) == 0 {
+		return nil, errors.New("no companies found")
+	}
+	return result, nil
+}
+func (c *CompanyServiceImpl) GetCompanyInfoByID(id string) (*models.CompanyInfo, error) {
+	var existingCompanyInfo models.CompanyInfo
+	err := c.companycollection.FindOne(c.ctx, bson.M{"company_id": id}).Decode(&existingCompanyInfo)
+	if err != nil {
+		if errors.Is(err, mongo.ErrNoDocuments) {
+			return nil, err
+		}
+	}
+	return &existingCompanyInfo, nil
+}
+func (c *CompanyServiceImpl) RegisterCompany(company *models.CompanyRegisterBody) (*models.CompanyInfo, error) {
+
+	var existingCompany *models.CompanyInfo
+
+	err := c.companycollection.FindOne(c.ctx, bson.M{"$or": []bson.M{{"username": company.Username}, {"email": bson.M{"$in": []string{company.Email}}}}}).Decode(&existingCompany)
+	if err == nil {
+		logging.Logger().Warn("Username or email already exists", err)
+		return nil, errors.New("username or email already exists")
+	} else if !errors.Is(err, mongo.ErrNoDocuments) {
+		logging.Logger().Error(err)
+		return nil, errors.New("InternalServerError")
+	}
+	companyInfo := &models.CompanyInfo{
+		CompanyID:   primitive.NewObjectID().Hex(),
+		CompanyName: company.CompanyName,
+		CreateAt:    time.Now().UTC().Round(time.Second),
+		UpdateAt:    time.Now().UTC().Round(time.Second),
+		Country:     company.Country,
+		Username:    company.Username,
+		Owner:       []string{company.Username},
+		Email:       company.Email,
+		FirstName:   company.FirstName,
+		LastName:    company.LastName,
+		Mobile:      company.Mobile,
+	}
+
+	_, err = c.companycollection.InsertOne(c.ctx, companyInfo)
+	return companyInfo, err
+}