This repository has been archived by the owner on Oct 20, 2022. It is now read-only.
/
parametercontext.go
107 lines (89 loc) · 4.05 KB
/
parametercontext.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
// Copyright 2020 Orange SA
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.package apis
package nificlient
import (
"strconv"
"github.com/antihax/optional"
nigoapi "github.com/erdrix/nigoapi/pkg/nifi"
)
func (n *nifiClient) GetParameterContext(id string) (*nigoapi.ParameterContextEntity, error) {
// Get nigoapi client, favoring the one associated to the coordinator node.
client, context := n.privilegeCoordinatorClient()
if client == nil {
log.Error(ErrNoNodeClientsAvailable, "Error during creating node client")
return nil, ErrNoNodeClientsAvailable
}
// Request on Nifi Rest API to get the parameter context informations
pcEntity, rsp, body, err := client.ParameterContextsApi.GetParameterContext(context, id)
if err := errorGetOperation(rsp, body, err); err != nil {
return nil, err
}
return &pcEntity, nil
}
func (n *nifiClient) CreateParameterContext(entity nigoapi.ParameterContextEntity) (*nigoapi.ParameterContextEntity, error) {
// Get nigoapi client, favoring the one associated to the coordinator node.
client, context := n.privilegeCoordinatorClient()
if client == nil {
log.Error(ErrNoNodeClientsAvailable, "Error during creating node client")
return nil, ErrNoNodeClientsAvailable
}
// Request on Nifi Rest API to create the parameter context
pcEntity, rsp, body, err := client.ParameterContextsApi.CreateParameterContext(context, entity)
if err := errorCreateOperation(rsp, body, err); err != nil {
return nil, err
}
return &pcEntity, nil
}
func (n *nifiClient) RemoveParameterContext(entity nigoapi.ParameterContextEntity) error {
// Get nigoapi client, favoring the one associated to the coordinator node.
client, context := n.privilegeCoordinatorClient()
if client == nil {
log.Error(ErrNoNodeClientsAvailable, "Error during creating node client")
return ErrNoNodeClientsAvailable
}
// Request on Nifi Rest API to remove the parameter context
_, rsp, body, err := client.ParameterContextsApi.DeleteParameterContext(context, entity.Id,
&nigoapi.ParameterContextsApiDeleteParameterContextOpts{
Version: optional.NewString(strconv.FormatInt(*entity.Revision.Version, 10)),
})
return errorDeleteOperation(rsp, body, err)
}
func (n *nifiClient) CreateParameterContextUpdateRequest(contextId string, entity nigoapi.ParameterContextEntity) (*nigoapi.ParameterContextUpdateRequestEntity, error) {
// Get nigoapi client, favoring the one associated to the coordinator node.
client, context := n.privilegeCoordinatorClient()
if client == nil {
log.Error(ErrNoNodeClientsAvailable, "Error during creating node client")
return nil, ErrNoNodeClientsAvailable
}
// Request on Nifi Rest API to create the parameter context update request
request, rsp, body, err := client.ParameterContextsApi.SubmitParameterContextUpdate(context, contextId, entity)
if err := errorUpdateOperation(rsp, body, err); err != nil {
return nil, err
}
return &request, nil
}
func (n *nifiClient) GetParameterContextUpdateRequest(contextId, id string) (*nigoapi.ParameterContextUpdateRequestEntity, error) {
// Get nigoapi client, favoring the one associated to the coordinator node.
client, context := n.privilegeCoordinatorClient()
if client == nil {
log.Error(ErrNoNodeClientsAvailable, "Error during creating node client")
return nil, ErrNoNodeClientsAvailable
}
// Request on Nifi Rest API to get the parameter context update request information
request, rsp, body, err := client.ParameterContextsApi.GetParameterContextUpdate(context, contextId, id)
if err := errorGetOperation(rsp, body, err); err != nil {
return nil, err
}
return &request, nil
}