Added search params for get contract
This commit is contained in:
69
database/contract/contract.go
Normal file
69
database/contract/contract.go
Normal file
@@ -0,0 +1,69 @@
|
||||
package contract
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"log"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/jinzhu/gorm"
|
||||
"gitlab.com/pactual1/backend/models"
|
||||
"gitlab.com/pactual1/backend/shared"
|
||||
)
|
||||
|
||||
func GetContracts(status string, companyName string, startTime time.Time, deviceIDs []int64,limit, offset int) ([]models.Contract, int64, int, error) {
|
||||
// Create a slice to hold the contracts
|
||||
var contracts []models.Contract
|
||||
|
||||
// Initialize the query
|
||||
db := shared.GetDb().Where("status = ?", status)
|
||||
countDb := db
|
||||
|
||||
if companyName != "" {
|
||||
db = db.Joins("left join companies on companies.id = contracts.buyer_id").Where("companies.name LIKE ?", "%"+companyName+"%")
|
||||
countDb = countDb.Joins("left join companies on companies.id = contracts.buyer_id").Where("companies.name LIKE ?", "%"+companyName+"%")
|
||||
}
|
||||
|
||||
if !startTime.IsZero() {
|
||||
db = db.Where("start_time >= ?", startTime)
|
||||
countDb = countDb.Where("start_time >= ?", startTime)
|
||||
}
|
||||
if len(deviceIDs) > 0 {
|
||||
db = db.Where("device_ids && ARRAY[?]::int8[]", deviceIDs)
|
||||
countDb = countDb.Where("device_ids && ARRAY[?]::int8[]", deviceIDs)
|
||||
}
|
||||
|
||||
// Count the total number of contracts
|
||||
var total int64
|
||||
if err := countDb.Model(&models.Contract{}).Count(&total).Error; err != nil {
|
||||
log.Printf("GetLatestContracts Error: Database error: %v", err)
|
||||
return contracts, total, http.StatusInternalServerError, err
|
||||
}
|
||||
|
||||
// Fetch the latest contracts from the database with LIMIT and OFFSET
|
||||
if err := db.Order("created_at desc").Limit(limit).Offset(offset).Find(&contracts).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
log.Printf("GetLatestContracts Error: No contracts found: %v", err)
|
||||
return contracts, total, http.StatusNotFound, err
|
||||
} else {
|
||||
log.Printf("GetLatestContracts Error: Database error: %v", err)
|
||||
return contracts, total, http.StatusInternalServerError, err
|
||||
}
|
||||
}
|
||||
return contracts, total, http.StatusOK, nil
|
||||
}
|
||||
|
||||
|
||||
|
||||
func GetContractByID(contractID uint64) (models.Contract, int ,error) {
|
||||
|
||||
// Fetch the contract creation date based on contractID
|
||||
var contract models.Contract
|
||||
if err := shared.GetDb().Where("id = ?", contractID).First(&contract).Error; err != nil {
|
||||
log.Printf("GetDeviceData Error: Could not fetch contract: %v", err)
|
||||
return contract, http.StatusInternalServerError, err
|
||||
}
|
||||
|
||||
return contract , http.StatusOK, nil
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user