mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-05 05:07:38 +08:00
144 lines
3.8 KiB
Go
144 lines
3.8 KiB
Go
// RAINBOND, Application Management Platform
|
|
// Copyright (C) 2014-2017 Goodrain Co., Ltd.
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version. For any non-GPL usage of Rainbond,
|
|
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
|
|
// must be obtained first.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package region
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pquerna/ffjson/ffjson"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
api_model "github.com/goodrain/rainbond/pkg/api/model"
|
|
)
|
|
|
|
//DefineSources DefineSources
|
|
func (t *Tenant) DefineSources(ss *api_model.SourceSpec) DefineSourcesInterface {
|
|
return &DefineSources{
|
|
tenant: t,
|
|
Model: Body{
|
|
SourceSpec: ss,
|
|
},
|
|
}
|
|
}
|
|
|
|
//DefineSources DefineSources
|
|
type DefineSources struct {
|
|
tenant *Tenant
|
|
Model Body
|
|
}
|
|
|
|
//Body Body
|
|
type Body struct {
|
|
SourceSpec *api_model.SourceSpec `json:"source_spec"`
|
|
}
|
|
|
|
//DefineSourcesInterface DefineSourcesInterface
|
|
type DefineSourcesInterface interface {
|
|
GetSource(sourceAlias string) ([]byte, error)
|
|
PostSource(sourceAlias string) error
|
|
PutSource(SourcesAlias string) error
|
|
DeleteSource(sourceAlias string) error
|
|
}
|
|
|
|
//GetSource GetSource
|
|
func (d *DefineSources) GetSource(sourceAlias string) ([]byte, error) {
|
|
resp, status, err := DoRequest(
|
|
fmt.Sprintf("/v2/tenants/%s/sources/%s/%s",
|
|
d.tenant.tenantID, d.Model.SourceSpec.Alias, d.Model.SourceSpec.SourceBody.EnvName),
|
|
"GET",
|
|
nil,
|
|
)
|
|
if err != nil {
|
|
logrus.Errorf("get define source %s error, %v", d.Model.SourceSpec.SourceBody.EnvName, err)
|
|
return nil, err
|
|
}
|
|
if status > 400 {
|
|
if status == 404 {
|
|
return nil, fmt.Errorf("source %s is not exist", d.Model.SourceSpec.SourceBody.EnvName)
|
|
}
|
|
return nil, fmt.Errorf("get define source %s failed", d.Model.SourceSpec.SourceBody.EnvName)
|
|
}
|
|
//valJ, err := simplejson.NewJson(resp)
|
|
return resp, nil
|
|
}
|
|
|
|
//PostSource PostSource
|
|
func (d *DefineSources) PostSource(sourceAlias string) error {
|
|
data, err := ffjson.Marshal(d.Model)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, status, err := DoRequest(
|
|
fmt.Sprintf("/v2/tenants/%s/sources/%s",
|
|
d.tenant.tenantID, d.Model.SourceSpec.Alias),
|
|
"POST",
|
|
data,
|
|
)
|
|
if err != nil {
|
|
logrus.Errorf("create define source error, %v", err)
|
|
return err
|
|
}
|
|
if status > 400 {
|
|
logrus.Errorf("create define source error")
|
|
return fmt.Errorf("cretae define source failed")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//PutSource PutSource
|
|
func (d *DefineSources) PutSource(sourceAlias string) error {
|
|
data, err := ffjson.Marshal(d.Model)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, status, err := DoRequest(
|
|
fmt.Sprintf("/v2/tenants/%s/sources/%s/%s",
|
|
d.tenant.tenantID, d.Model.SourceSpec.Alias, d.Model.SourceSpec.SourceBody.EnvName),
|
|
"PUT",
|
|
data,
|
|
)
|
|
if err != nil {
|
|
logrus.Errorf("update define source error, %v", err)
|
|
return err
|
|
}
|
|
if status > 400 {
|
|
logrus.Errorf("update define source error")
|
|
return fmt.Errorf("update define source failed")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//DeleteSource DeleteSource
|
|
func (d *DefineSources) DeleteSource(sourceAlias string) error {
|
|
_, status, err := DoRequest(
|
|
fmt.Sprintf("/v2/tenants/%s/sources/%s/%s",
|
|
d.tenant.tenantID, d.Model.SourceSpec.Alias, d.Model.SourceSpec.SourceBody.EnvName),
|
|
"DELETE",
|
|
nil,
|
|
)
|
|
if err != nil {
|
|
logrus.Errorf("delete define source error, %v", err)
|
|
return err
|
|
}
|
|
if status > 400 {
|
|
logrus.Errorf("delete define source %s error", d.Model.SourceSpec.SourceBody.EnvName)
|
|
}
|
|
return nil
|
|
}
|