data_source_obmcs_database_db_system_shape.go

// Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.

package main

import (
	"time"

	"github.com/hashicorp/terraform/helper/schema"
	"github.com/oracle/bmcs-go-sdk"

	"github.com/oracle/terraform-provider-oci/options"

	"github.com/oracle/terraform-provider-oci/crud"
)

func DBSystemShapeDatasource() *schema.Resource {
	return &schema.Resource{
		Read: readDBSystemShapes,
		Schema: map[string]*schema.Schema{
			"db_system_shapes": {
				Type:     schema.TypeList,
				Computed: true,
				Elem: &schema.Resource{
					Schema: map[string]*schema.Schema{
						"available_core_count": {
							Type:     schema.TypeInt,
							Computed: true,
						},
						"name": {
							Type:     schema.TypeString,
							Computed: true,
						},
						"shape": {
							Type:     schema.TypeString,
							Computed: true,
						},
					},
				},
			},
			"availability_domain": {
				Type:     schema.TypeString,
				Required: true,
			},
			"compartment_id": {
				Type:     schema.TypeString,
				Required: true,
			},
			"limit": {
				Type:     schema.TypeInt,
				Optional: true,
			},
			"page": {
				Type:     schema.TypeString,
				Optional: true,
			},
		},
	}
}

func readDBSystemShapes(d *schema.ResourceData, m interface{}) (e error) {
	client := m.(*OracleClients)
	reader := &DBSystemShapeDatasourceCrud{}
	reader.D = d
	reader.Client = client.client
	return crud.ReadResource(reader)
}

type DBSystemShapeDatasourceCrud struct {
	crud.BaseCrud
	Res *baremetal.ListDBSystemShapes
}

func (s *DBSystemShapeDatasourceCrud) Get() (e error) {
	availabilityDomain := s.D.Get("availability_domain").(string)
	compartmentID := s.D.Get("compartment_id").(string)

	opts := &baremetal.ListOptions{}
	options.SetPageOptions(s.D, &opts.PageListOptions)
	options.SetLimitOptions(s.D, &opts.LimitListOptions)

	s.Res = &baremetal.ListDBSystemShapes{}

	for {
		var list *baremetal.ListDBSystemShapes
		if list, e = s.Client.ListDBSystemShapes(
			availabilityDomain, compartmentID, opts,
		); e != nil {
			break
		}

		s.Res.DBSystemShapes = append(s.Res.DBSystemShapes, list.DBSystemShapes...)

		if hasNextPage := options.SetNextPageOption(list.NextPage, &opts.PageListOptions); !hasNextPage {
			break
		}
	}

	return
}

func (s *DBSystemShapeDatasourceCrud) SetData() {
	if s.Res != nil {
		// Important, if you don't have an ID, make one up for your datasource
		// or things will end in tears
		s.D.SetId(time.Now().UTC().String())
		resources := []map[string]interface{}{}
		for _, v := range s.Res.DBSystemShapes {
			res := map[string]interface{}{
				"available_core_count": v.AvailableCoreCount,
				"name":                 v.Name,
				"shape":                v.Shape,
			}
			resources = append(resources, res)
		}
		s.D.Set("db_system_shapes", resources)
	}
	return
}