-
Notifications
You must be signed in to change notification settings - Fork 1
/
api_fs_model.go
132 lines (114 loc) · 3.62 KB
/
api_fs_model.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package api
import (
"context"
"strconv"
paifeaturestore "github.com/alibabacloud-go/paifeaturestore-20230621/v3/client"
"github.com/aliyun/aliyun-pai-featurestore-go-sdk/v2/constants"
"github.com/antihax/optional"
)
// Linger please
var (
_ context.Context
)
type FsModelApiService service
/*
FsModelApiService Get Model By ID
- @param ctx context.Context - for authentication, logging, cancellation, deadlines, tracing, etc. Passed from http.Request or context.Background().
- @param modelId
@return InlineResponse20086
*/
func (a *FsModelApiService) GetModelByID(modelId string) (GetModelResponse, error) {
var (
localVarReturnValue GetModelResponse
)
response, err := a.client.GetModelFeature(&a.client.instanceId, &modelId)
if err != nil {
return localVarReturnValue, err
}
mid, _ := strconv.Atoi(modelId)
model := Model{
ModelId: mid,
ProjectName: *response.Body.ProjectName,
Name: *response.Body.Name,
}
if id, err := strconv.Atoi(*response.Body.ProjectId); err == nil {
model.ProjectId = id
}
var features []*ModelFeatures
for _, featureItem := range response.Body.Features {
feature := ModelFeatures{
FeatureViewName: *featureItem.FeatureViewName,
Name: *featureItem.Name,
}
if featureItem.AliasName != nil && *featureItem.AliasName != "" && *featureItem.AliasName != feature.Name {
feature.AliasName = *featureItem.AliasName
}
if id, err := strconv.Atoi(*featureItem.FeatureViewId); err == nil {
feature.FeatureViewId = id
}
switch *featureItem.Type {
case "INT32":
feature.Type = int32(constants.FS_INT32)
case "INT64":
feature.Type = int32(constants.FS_INT64)
case "FLOAT":
feature.Type = int32(constants.FS_FLOAT)
case "DOUBLE":
feature.Type = int32(constants.FS_DOUBLE)
case "BOOLEAN":
feature.Type = int32(constants.FS_BOOLEAN)
case "TIMESTAMP":
feature.Type = int32(constants.FS_TIMESTAMP)
default:
feature.Type = int32(constants.FS_STRING)
}
features = append(features, &feature)
}
model.Features = features
localVarReturnValue.Model = &model
return localVarReturnValue, nil
}
/*
FsModelApiService List Models
* @param ctx context.Context - for authentication, logging, cancellation, deadlines, tracing, etc. Passed from http.Request or context.Background().
* @param optional nil or *FsModelApiListModelsOpts - Optional Parameters:
* @param "Pagesize" (optional.Int32) -
* @param "Pagenumber" (optional.Int32) -
* @param "ProjectId" (optional.Int32) -
@return InlineResponse20085
*/
type FsModelApiListModelsOpts struct {
Pagesize optional.Int32
Pagenumber optional.Int32
ProjectId optional.Int32
}
func (a *FsModelApiService) ListModels(pagesize, pagenumber int, projectId string) (ListModelsResponse, error) {
var (
localVarReturnValue ListModelsResponse
)
request := paifeaturestore.ListModelFeaturesRequest{}
request.SetPageSize(int32(pagesize))
request.SetPageNumber(int32(pagenumber))
request.SetProjectId(projectId)
response, err := a.client.ListModelFeatures(&a.client.instanceId, &request)
if err != nil {
return localVarReturnValue, err
}
localVarReturnValue.TotalCount = int(*response.Body.TotalCount)
var models []*Model
for _, modelFeature := range response.Body.ModelFeatures {
if id, err := strconv.Atoi(*modelFeature.ModelFeatureId); err == nil {
model := Model{
ModelId: id,
Name: *modelFeature.Name,
ProjectName: *modelFeature.ProjectName,
}
if id, err := strconv.Atoi(*modelFeature.ProjectId); err == nil {
model.ProjectId = id
}
models = append(models, &model)
}
}
localVarReturnValue.Models = models
return localVarReturnValue, nil
}