forked from rook/rook
108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
/*
|
|
Copyright 2016 The Rook Authors. All rights reserved.
|
|
|
|
Licensed 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 object
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/rook/rook/cmd/rookctl/pool"
|
|
"github.com/rook/rook/cmd/rookctl/rook"
|
|
"github.com/rook/rook/pkg/model"
|
|
"github.com/rook/rook/pkg/rook/client"
|
|
"github.com/rook/rook/pkg/util/flags"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const (
|
|
initObjectStoreTimeout = 60
|
|
)
|
|
|
|
var (
|
|
store model.ObjectStore
|
|
dataConfig pool.Config
|
|
metadataConfig pool.Config
|
|
certificateFile string
|
|
createCmd = &cobra.Command{
|
|
Use: "create",
|
|
Short: "Creates a new object storage instance in the cluster",
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
createCmd.Flags().StringVarP(&store.Name, "name", "n", "default", "The name of the object store instance")
|
|
createCmd.Flags().Int32VarP(&store.Port, "port", "p", model.RGWPort, "The port on which to expose the object store")
|
|
createCmd.Flags().Int32Var(&store.RGWReplicas, "rgw-replicas", 1, "The number of RGW services for load balancing")
|
|
createCmd.Flags().StringVar(&certificateFile, "certificate", "", "Path to the ssl cert file (pem format)")
|
|
pool.AddPoolFlags(createCmd, "data-", &dataConfig)
|
|
pool.AddPoolFlags(createCmd, "metadata-", &metadataConfig)
|
|
|
|
createCmd.RunE = createObjectStoreEntry
|
|
}
|
|
|
|
func createObjectStoreEntry(cmd *cobra.Command, args []string) error {
|
|
rook.SetupLogging()
|
|
if err := flags.VerifyRequiredFlags(cmd, []string{"name", "data-type", "metadata-type"}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if certificateFile != "" {
|
|
cert, err := ioutil.ReadFile(certificateFile)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "failed to read certificate", err)
|
|
os.Exit(1)
|
|
}
|
|
store.Certificate = string(cert)
|
|
}
|
|
|
|
dataPool, err := pool.ConfigToModel(dataConfig)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "failed to read data settings", err)
|
|
os.Exit(1)
|
|
}
|
|
store.DataConfig = *dataPool
|
|
metadataPool, err := pool.ConfigToModel(metadataConfig)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "failed to read metadata settings", err)
|
|
os.Exit(1)
|
|
}
|
|
store.MetadataConfig = *metadataPool
|
|
|
|
c := rook.NewRookNetworkRestClientWithTimeout(initObjectStoreTimeout * time.Second)
|
|
err = createObjectStore(c)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func createObjectStore(c client.RookRestClient) error {
|
|
|
|
_, err := c.CreateObjectStore(store)
|
|
|
|
// HTTP 202 Accepted is expected
|
|
if err != nil && !client.IsHttpAccepted(err) {
|
|
return fmt.Errorf("failed to create new object store: %+v", err)
|
|
}
|
|
|
|
fmt.Println("succeeded starting creation of object store")
|
|
return nil
|
|
}
|