| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 executor |
| |
| import ( |
| "encoding/json" |
| "fmt" |
| "strconv" |
| |
| "github.com/apache/incubator-pegasus/admin-cli/util" |
| "github.com/apache/incubator-pegasus/go-client/session" |
| ) |
| |
| type TableDataVersion struct { |
| DataVersion string `json:"data_version"` |
| } |
| |
| func QueryTableVersion(client *Client, table string) error { |
| version, err := QueryReplicaDataVersion(client, table) |
| if err != nil { |
| return nil |
| } |
| |
| // formats into JSON |
| outputBytes, _ := json.MarshalIndent(version, "", " ") |
| fmt.Fprintln(client, string(outputBytes)) |
| return nil |
| } |
| |
| func QueryReplicaDataVersion(client *Client, table string) (*TableDataVersion, error) { |
| nodes := client.Nodes.GetAllNodes(session.NodeTypeReplica) |
| resp, err := client.Meta.QueryConfig(table) |
| if err != nil { |
| return nil, err |
| } |
| |
| args := util.Arguments{ |
| Name: "app_id", |
| Value: strconv.Itoa(int(resp.AppID)), |
| } |
| results := util.BatchCallHTTP(nodes, getTableDataVersion, args) |
| |
| var finalVersion TableDataVersion |
| versions := make(map[string]TableDataVersion) |
| for _, result := range results { |
| if result.Err != nil { |
| return nil, result.Err |
| } |
| err := json.Unmarshal([]byte(result.Resp), &versions) |
| if err != nil { |
| return nil, err |
| } |
| |
| for _, version := range versions { |
| if finalVersion.DataVersion == "" { |
| finalVersion = version |
| } else { |
| if version.DataVersion == finalVersion.DataVersion { |
| continue |
| } else { |
| return nil, fmt.Errorf("replica versions are not consistent") |
| } |
| } |
| } |
| } |
| return &finalVersion, nil |
| } |
| |
| func getTableDataVersion(addr string, args util.Arguments) (string, error) { |
| url := fmt.Sprintf("http://%s/replica/data_version?%s=%s", addr, args.Name, args.Value) |
| return util.CallHTTPGet(url) |
| } |