-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Matt Spilchen
committed
Oct 13, 2023
1 parent
fa4852c
commit 491554f
Showing
5 changed files
with
276 additions
and
60 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,164 @@ | ||
| /* | ||
| (c) Copyright [2023] Open Text. | ||
| 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 vclusterops | ||
|
|
||
| import ( | ||
| "fmt" | ||
|
|
||
| "github.com/vertica/vcluster/vclusterops/util" | ||
| "github.com/vertica/vcluster/vclusterops/vlog" | ||
| ) | ||
|
|
||
| type httpsPollSubscriptionStateOp struct { | ||
| OpBase | ||
| OpHTTPSBase | ||
| timeout int | ||
| } | ||
|
|
||
| func makeHTTPSPollSubscriptionStateOp(log vlog.Printer, hosts []string, | ||
| useHTTPPassword bool, userName string, httpsPassword *string) (httpsPollSubscriptionStateOp, error) { | ||
| op := httpsPollSubscriptionStateOp{} | ||
| op.name = "HTTPSPollSubscriptionStateOp" | ||
| op.log = log.WithName(op.name) | ||
| op.hosts = hosts | ||
| op.useHTTPPassword = useHTTPPassword | ||
| op.timeout = StartupPollingTimeout | ||
|
|
||
| err := util.ValidateUsernameAndPassword(op.name, useHTTPPassword, userName) | ||
| if err != nil { | ||
| return op, err | ||
| } | ||
| op.userName = userName | ||
| op.httpsPassword = httpsPassword | ||
|
|
||
| return op, nil | ||
| } | ||
|
|
||
| func (op *httpsPollSubscriptionStateOp) getPollingTimeout() int { | ||
| return op.timeout | ||
| } | ||
|
|
||
| func (op *httpsPollSubscriptionStateOp) setupClusterHTTPRequest(hosts []string) error { | ||
| op.clusterHTTPRequest = ClusterHTTPRequest{} | ||
| op.clusterHTTPRequest.RequestCollection = make(map[string]HostHTTPRequest) | ||
| op.setVersionToSemVar() | ||
|
|
||
| for _, host := range hosts { | ||
| httpRequest := HostHTTPRequest{} | ||
| httpRequest.Method = GetMethod | ||
| httpRequest.Timeout = httpRequestTimeoutSeconds | ||
| httpRequest.BuildHTTPSEndpoint("subscriptions") | ||
| if op.useHTTPPassword { | ||
| httpRequest.Password = op.httpsPassword | ||
| httpRequest.Username = op.userName | ||
| } | ||
|
|
||
| op.clusterHTTPRequest.RequestCollection[host] = httpRequest | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (op *httpsPollSubscriptionStateOp) prepare(execContext *OpEngineExecContext) error { | ||
| execContext.dispatcher.Setup(op.hosts) | ||
|
|
||
| return op.setupClusterHTTPRequest(op.hosts) | ||
| } | ||
|
|
||
| func (op *httpsPollSubscriptionStateOp) execute(execContext *OpEngineExecContext) error { | ||
| if err := op.runExecute(execContext); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return op.processResult(execContext) | ||
| } | ||
|
|
||
| func (op *httpsPollSubscriptionStateOp) finalize(_ *OpEngineExecContext) error { | ||
| return nil | ||
| } | ||
|
|
||
| // The content of SubscriptionMap should look like | ||
| /* "subscription_list": [ | ||
| { | ||
| "node_name": "v_practice_db_node0001", | ||
| "shard_name": "replica", | ||
| "subscription_state": "ACTIVE", | ||
| "is_primary": true | ||
| }, | ||
| { | ||
| "node_name": "v_practice_db_node0001", | ||
| "shard_name": "segment0001", | ||
| "subscription_state": "ACTIVE", | ||
| "is_primary": true | ||
| }, | ||
| ... | ||
| ] | ||
| */ | ||
| type SubscriptionList struct { | ||
| SubscriptionList []SubscriptionInfo `json:"subscription_list"` | ||
| } | ||
|
|
||
| type SubscriptionInfo struct { | ||
| Nodename string `json:"node_name"` | ||
| ShardName string `json:"shard_name"` | ||
| SubscriptionState string `json:"subscription_state"` | ||
| IsPrimary bool `json:"is_primary"` | ||
| } | ||
|
|
||
| func (op *httpsPollSubscriptionStateOp) processResult(execContext *OpEngineExecContext) error { | ||
| err := pollState(op, execContext) | ||
| if err != nil { | ||
| return fmt.Errorf("not all subscriptions are ACTIVE, %w", err) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (op *httpsPollSubscriptionStateOp) shouldStopPolling() (bool, error) { | ||
| var subscriptionList SubscriptionList | ||
|
|
||
| for host, result := range op.clusterHTTPRequest.ResultCollection { | ||
| op.logResponse(host, result) | ||
|
|
||
| if result.IsPasswordAndCertificateError() { | ||
| return true, fmt.Errorf("[%s] wrong password/certificate for https service on host %s", | ||
| op.name, host) | ||
| } | ||
|
|
||
| if result.isPassing() { | ||
| err := op.parseAndCheckResponse(host, result.content, &subscriptionList) | ||
| if err != nil { | ||
| op.log.PrintError("[%s] fail to parse result on host %s, details: %s", | ||
| op.name, host, err) | ||
| return true, err | ||
| } | ||
|
|
||
| // check whether all subscriptions are ACTIVE | ||
| for _, s := range subscriptionList.SubscriptionList { | ||
| if s.SubscriptionState != "ACTIVE" { | ||
| return false, nil | ||
| } | ||
| } | ||
|
|
||
| op.log.PrintInfo("All subscriptions are ACTIVE\n") | ||
| return true, nil | ||
| } | ||
| } | ||
|
|
||
| // this could happen if ResultCollection is empty | ||
| op.log.PrintError("[%s] empty result received from the provided hosts %v", op.name, op.hosts) | ||
| return false, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.