core/oc-lib
6
0
Fork 0
Code Issues 6 Pull Requests Packages Projects Releases Wiki Activity

add purchase resource in model catalog

Browse Source
This commit is contained in:
mr 2025-02-10 13:04:13 +01:00
parent f663ec80f5
commit 4ad32401fd
4 changed files with 51 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
models/resources/compute.go
View File

@ -2,7 +2,6 @@ package resources
import (
"errors"
"fmt"
"strings"
"time"
@ -139,14 +138,21 @@ func (r *PricedComputeResource) GetType() tools.DataType {
}
func (r *PricedComputeResource) GetPrice() (float64, error) {
fmt.Println("GetPrice", r.UsageStart, r.UsageEnd)
if r.UsageStart == nil || r.UsageEnd == nil {
return 0, errors.New("usage start and end must be set")
now := time.Now()
if r.UsageStart == nil {
r.UsageStart = &now
}
if r.SelectedPricing == nil {
return 0, errors.New("selected pricing must be set")
if r.UsageEnd == nil {
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
r.UsageEnd = &add
}
pricing := *r.SelectedPricing
if pricing == nil {
if len(r.PricingProfiles) == 0 {
return 0, errors.New("pricing profile must be set")
}
pricing = r.PricingProfiles[0]
}
price := float64(0)
for _, l := range []map[string]float64{r.CPUsLocated, r.GPUsLocated} {
for model, amountOfData := range l {

15
models/resources/data.go
View File

@ -152,13 +152,24 @@ func (r *PricedDataResource) GetType() tools.DataType {
func (r *PricedDataResource) GetPrice() (float64, error) {
fmt.Println("GetPrice", r.UsageStart, r.UsageEnd)
if r.UsageStart == nil || r.UsageEnd == nil {
return 0, errors.New("usage start and end must be set")
now := time.Now()
if r.UsageStart == nil {
r.UsageStart = &now
}
if r.UsageEnd == nil {
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
r.UsageEnd = &add
}
if r.SelectedPricing == nil {
return 0, errors.New("selected pricing must be set")
}
pricing := *r.SelectedPricing
if pricing == nil {
if len(r.PricingProfiles) == 0 {
return 0, errors.New("pricing profile must be set")
}
pricing = r.PricingProfiles[0]
}
var err error
amountOfData := float64(1)
if pricing.GetOverrideStrategyValue() >= 0 {

31
models/resources/priced_resource.go
View File

@ -35,18 +35,6 @@ func (abs *PricedResource) GetCreatorID() string {
return abs.CreatorID
}
func (abs *PricedResource) SetStartUsage(start time.Time) {
if abs.UsageStart == nil {
abs.UsageStart = &start
}
}
func (abs *PricedResource) SetEndUsage(end time.Time) {
if abs.UsageEnd == nil {
abs.UsageEnd = &end
}
}
func (abs *PricedResource) IsPurchased() bool {
if abs.SelectedPricing == nil {
return false
@ -86,11 +74,20 @@ func (abs *PricedResource) GetExplicitDurationInS() float64 {
func (r *PricedResource) GetPrice() (float64, error) {
fmt.Println("GetPrice", r.UsageStart, r.UsageEnd)
if r.UsageStart == nil || r.UsageEnd == nil {
return 0, errors.New("usage start and end must be set")
now := time.Now()
if r.UsageStart == nil {
r.UsageStart = &now
}
if r.SelectedPricing == nil {
return 0, errors.New("selected pricing must be set")
if r.UsageEnd == nil {
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
r.UsageEnd = &add
}
return (*r.SelectedPricing).GetPrice(1, 0, *r.UsageStart, *r.UsageEnd)
pricing := *r.SelectedPricing
if pricing == nil {
if len(r.PricingProfiles) == 0 {
return 0, errors.New("pricing profile must be set")
}
pricing = r.PricingProfiles[0]
}
return pricing.GetPrice(1, 0, *r.UsageStart, *r.UsageEnd)
}

16
models/resources/storage.go
View File

@ -163,13 +163,21 @@ func (r *PricedStorageResource) GetType() tools.DataType {
func (r *PricedStorageResource) GetPrice() (float64, error) {
fmt.Println("GetPrice", r.UsageStart, r.UsageEnd)
if r.UsageStart == nil || r.UsageEnd == nil {
return 0, errors.New("usage start and end must be set")
now := time.Now()
if r.UsageStart == nil {
r.UsageStart = &now
}
if r.SelectedPricing == nil {
return 0, errors.New("selected pricing must be set")
if r.UsageEnd == nil {
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
r.UsageEnd = &add
}
pricing := *r.SelectedPricing
if pricing == nil {
if len(r.PricingProfiles) == 0 {
return 0, errors.New("pricing profile must be set")
}
pricing = r.PricingProfiles[0]
}
var err error
amountOfData := float64(1)
if pricing.GetOverrideStrategyValue() >= 0 {