opensource/cloudpods
3
0
Fork 0
mirror of https://github.com/yunionio/cloudpods.git synced 2026-09-24 16:03:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(region): guest: add {open,close,list}-forward API

Browse Source
This commit is contained in:
Yousong Zhou
2020-12-15 22:12:23 +08:00
parent 8f54866272
commit 1427d1b3a6
1 changed files with 78 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/compute/models/guest_actions.go
+78
View File
@@ -51,6 +51,7 @@ import (
"yunion.io/x/onecloud/pkg/cloudcommon/policy"
"yunion.io/x/onecloud/pkg/cloudcommon/userdata"
"yunion.io/x/onecloud/pkg/cloudprovider"
guestdriver_types "yunion.io/x/onecloud/pkg/compute/guestdrivers/types"
"yunion.io/x/onecloud/pkg/compute/options"
"yunion.io/x/onecloud/pkg/httperrors"
"yunion.io/x/onecloud/pkg/mcclient"
@@ -5126,3 +5127,80 @@ func (self *SGuest) PerformRemoteUpdate(ctx context.Context, userCred mcclient.T
}
return nil, nil
}
func (self *SGuest) AllowPerformOpenForward(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
return self.IsOwner(userCred) || db.IsAdminAllowPerform(userCred, self, "open-forward")
}
func (self *SGuest) PerformOpenForward(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
req, err := guestdriver_types.NewOpenForwardRequestFromJSON(data)
if err != nil {
return nil, err
}
for _, nicDesc := range self.fetchNICShortDesc(ctx) {
if nicDesc.VpcId != api.DEFAULT_VPC_ID {
req.Addr = nicDesc.IpAddr
req.NetworkId = nicDesc.NetworkId
}
}
if req.NetworkId == "" {
return nil, httperrors.NewInputParameterError("guest has no vpc ip")
}
resp, err := self.GetDriver().RequestOpenForward(ctx, userCred, self, req)
if err != nil {
return nil, httperrors.NewGeneralError(err)
}
return resp.JSON(), nil
}
func (self *SGuest) AllowPerformCloseForward(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
return self.IsOwner(userCred) || db.IsAdminAllowPerform(userCred, self, "close-forward")
}
func (self *SGuest) PerformCloseForward(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
req, err := guestdriver_types.NewCloseForwardRequestFromJSON(data)
if err != nil {
return nil, err
}
for _, nicDesc := range self.fetchNICShortDesc(ctx) {
if nicDesc.VpcId != api.DEFAULT_VPC_ID {
req.NetworkId = nicDesc.NetworkId
}
}
if req.NetworkId == "" {
return nil, httperrors.NewInputParameterError("guest has no vpc ip")
}
resp, err := self.GetDriver().RequestCloseForward(ctx, userCred, self, req)
if err != nil {
return nil, httperrors.NewGeneralError(err)
}
return resp.JSON(), nil
}
func (self *SGuest) AllowPerformListForward(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
return self.IsOwner(userCred) || db.IsAdminAllowPerform(userCred, self, "list-forward")
}
func (self *SGuest) PerformListForward(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
req, err := guestdriver_types.NewListForwardRequestFromJSON(data)
if err != nil {
return nil, err
}
for _, nicDesc := range self.fetchNICShortDesc(ctx) {
if nicDesc.VpcId != api.DEFAULT_VPC_ID {
req.Addr = nicDesc.IpAddr
req.NetworkId = nicDesc.NetworkId
}
}
if req.NetworkId == "" {
return nil, httperrors.NewInputParameterError("guest has no vpc ip")
}
resp, err := self.GetDriver().RequestListForward(ctx, userCred, self, req)
if err != nil {
return nil, httperrors.NewGeneralError(err)
}
return resp.JSON(), nil
}