mirror of
https://github.com/paralus/paralus.git
synced 2026-05-07 08:56:42 +00:00
c66bdc25cd
* restructure rcloud-base as a single base controller * updated master.rest * moved sentry from internal to pkg as it is used by relay * removing unused rpc and it's dependencies * Fix usermgmt tests * Don't redefine variables in rest file Co-authored-by: Abin Simon <abin.simon@rafay.co>
107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/base64"
|
|
|
|
"github.com/RafaySystems/rcloud-base/internal/cluster/fixtures"
|
|
"github.com/RafaySystems/rcloud-base/pkg/common"
|
|
"github.com/RafaySystems/rcloud-base/pkg/query"
|
|
"github.com/RafaySystems/rcloud-base/pkg/service"
|
|
rpcv3 "github.com/RafaySystems/rcloud-base/proto/rpc/scheduler"
|
|
commonv3 "github.com/RafaySystems/rcloud-base/proto/types/commonpb/v3"
|
|
infrapbv3 "github.com/RafaySystems/rcloud-base/proto/types/infrapb/v3"
|
|
)
|
|
|
|
type clusterServer struct {
|
|
service.ClusterService
|
|
downloadData common.DownloadData
|
|
}
|
|
|
|
// NewClusterServer returns new cluster server implementation
|
|
func NewClusterServer(es service.ClusterService, data *common.DownloadData) rpcv3.ClusterServer {
|
|
return &clusterServer{
|
|
ClusterService: es,
|
|
downloadData: *data,
|
|
}
|
|
}
|
|
|
|
func (s *clusterServer) CreateCluster(ctx context.Context, e *infrapbv3.Cluster) (*infrapbv3.Cluster, error) {
|
|
edge, err := s.Create(ctx, e)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return edge, nil
|
|
}
|
|
|
|
func (s *clusterServer) GetClusters(ctx context.Context, e *commonv3.QueryOptions) (*infrapbv3.ClusterList, error) {
|
|
clusters, err := s.List(ctx, query.WithOptions(e))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return clusters, nil
|
|
}
|
|
|
|
func (s *clusterServer) GetCluster(ctx context.Context, e *infrapbv3.Cluster) (*infrapbv3.Cluster, error) {
|
|
cluster, err := s.Select(ctx, e, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return cluster, nil
|
|
}
|
|
|
|
func (s *clusterServer) DeleteCluster(ctx context.Context, e *infrapbv3.Cluster) (*rpcv3.DeleteClusterResponse, error) {
|
|
err := s.Delete(ctx, e)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &rpcv3.DeleteClusterResponse{}, nil
|
|
}
|
|
|
|
func (s *clusterServer) UpdateCluster(ctx context.Context, e *infrapbv3.Cluster) (*infrapbv3.Cluster, error) {
|
|
edge, err := s.Update(ctx, e)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return edge, nil
|
|
}
|
|
|
|
func (s *clusterServer) DownloadCluster(ctx context.Context, cluster *infrapbv3.Cluster) (*commonv3.HttpBody, error) {
|
|
c, err := s.Select(ctx, cluster, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bb := new(bytes.Buffer)
|
|
|
|
if c.Spec.ProxyConfig != nil {
|
|
if c.Spec.ProxyConfig.BootstrapCA != "" {
|
|
c.Spec.ProxyConfig.BootstrapCA = base64.StdEncoding.EncodeToString([]byte(c.Spec.ProxyConfig.BootstrapCA))
|
|
}
|
|
}
|
|
|
|
err = fixtures.DownloadTemplate.Execute(bb, struct {
|
|
DownloadData common.DownloadData
|
|
Cluster *infrapbv3.Cluster
|
|
}{
|
|
s.downloadData, c,
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &commonv3.HttpBody{
|
|
ContentType: "application/x-rafay-yaml",
|
|
Data: bb.Bytes(),
|
|
}, nil
|
|
}
|
|
|
|
func (s *clusterServer) UpdateClusterStatus(ctx context.Context, cluster *infrapbv3.Cluster) (*infrapbv3.Cluster, error) {
|
|
err := s.UpdateClusterConditionStatus(ctx, cluster)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return cluster, nil
|
|
}
|