utils.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. // Copyright 2019 Yunion
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package apsara
  15. import (
  16. "yunion.io/x/jsonutils"
  17. "yunion.io/x/log"
  18. )
  19. type jsonRequestFunc func(action string, params map[string]string) (jsonutils.JSONObject, error)
  20. func unmarshalResult(resp jsonutils.JSONObject, respErr error, resultKey []string, result interface{}) error {
  21. if respErr != nil {
  22. return respErr
  23. }
  24. if result == nil {
  25. return nil
  26. }
  27. if resultKey != nil && len(resultKey) > 0 {
  28. respErr = resp.Unmarshal(result, resultKey...)
  29. } else {
  30. respErr = resp.Unmarshal(result)
  31. }
  32. if respErr != nil {
  33. log.Errorf("unmarshal json error %s", respErr)
  34. }
  35. return nil
  36. }
  37. // 执行操作
  38. func DoAction(client jsonRequestFunc, action string, params map[string]string, resultKey []string, result interface{}) error {
  39. resp, err := client(action, params)
  40. return unmarshalResult(resp, err, resultKey, result)
  41. }