Skip to content

Commit

Permalink
New datasources for datafactory
Browse files Browse the repository at this point in the history
* `azurerm_data_factory_trigger_schedules`: Retrieve a list of all trigger schedules for a given Data Factory
* `azurerm_data_factory_trigger_schedule`: Retrieve a rigger schedule for a given Data Factory by name
  • Loading branch information
favoretti committed Jan 19, 2024
1 parent db18694 commit b7a4641
Show file tree
Hide file tree
Showing 7 changed files with 624 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,318 @@
// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0

package datafactory

import (
"context"
"fmt"
"time"

"github.com/Azure/azure-sdk-for-go/services/datafactory/mgmt/2018-06-01/datafactory" // nolint: staticcheck
"github.com/hashicorp/go-azure-sdk/resource-manager/datafactory/2018-06-01/factories"
"github.com/hashicorp/terraform-provider-azurerm/internal/sdk"
"github.com/hashicorp/terraform-provider-azurerm/internal/services/datafactory/parse"
"github.com/hashicorp/terraform-provider-azurerm/internal/services/datafactory/validate"
"github.com/hashicorp/terraform-provider-azurerm/internal/tf/pluginsdk"
"github.com/hashicorp/terraform-provider-azurerm/utils"
)

type TriggerScheduleDataSource struct{}

type TriggerScheduleDataSourceModel struct {
Name string `tfschema:"name"`
DataFactoryID string `tfschema:"data_factory_id"`
Description string `tfschema:"description"`
Schedule []TriggerScheduleSchedule `tfschema:"schedule"`
StartTime string `tfschema:"start_time"`
EndTime string `tfschema:"end_time"`
TimeZone string `tfschema:"time_zone"`
Frequency string `tfschema:"frequency"`
Interval int `tfschema:"interval"`
Activated bool `tfschema:"activated"`
PipelineName string `tfschema:"pipeline_name"`
Annotations []string `tfschema:"annotations"`
}

type TriggerScheduleSchedule struct {
DaysOfMonth []int `tfschema:"days_of_month"`
DaysOfWeek []string `tfschema:"days_of_week"`
Hours []int `tfschema:"hours"`
Minutes []int `tfschema:"minutes"`
Monthly []TriggerScheduleScheduleMonthly `tfschema:"monthly"`
}

type TriggerScheduleScheduleMonthly struct {
Weekday string `tfschema:"weekday"`
Week int `tfschema:"week"`
}

var _ sdk.DataSource = TriggerScheduleDataSource{}

func (d TriggerScheduleDataSource) ModelObject() interface{} {
return &TriggerScheduleDataSourceModel{}
}

func (d TriggerScheduleDataSource) ResourceType() string {
return "azurerm_data_factory_trigger_schedule"
}

func (d TriggerScheduleDataSource) Arguments() map[string]*pluginsdk.Schema {
return map[string]*pluginsdk.Schema{
"name": {
Type: pluginsdk.TypeString,
Required: true,
ValidateFunc: validate.DataFactoryPipelineAndTriggerName(),
},

"data_factory_id": {
Type: pluginsdk.TypeString,
Required: true,
ValidateFunc: factories.ValidateFactoryID,
},
}
}

func (d TriggerScheduleDataSource) Attributes() map[string]*pluginsdk.Schema {
return map[string]*pluginsdk.Schema{
"description": {
Type: pluginsdk.TypeString,
Computed: true,
},

"schedule": {
Type: pluginsdk.TypeList,
Computed: true,
Elem: &pluginsdk.Resource{
Schema: map[string]*pluginsdk.Schema{
"days_of_month": {
Type: pluginsdk.TypeList,
Computed: true,
Elem: &pluginsdk.Schema{
Type: pluginsdk.TypeInt,
},
},

"days_of_week": {
Type: pluginsdk.TypeList,
Computed: true,
Elem: &pluginsdk.Schema{
Type: pluginsdk.TypeString,
},
},

"hours": {
Type: pluginsdk.TypeList,
Computed: true,
Elem: &pluginsdk.Schema{
Type: pluginsdk.TypeInt,
},
},

"minutes": {
Type: pluginsdk.TypeList,
Computed: true,
Elem: &pluginsdk.Schema{
Type: pluginsdk.TypeInt,
},
},

"monthly": {
Type: pluginsdk.TypeList,
Computed: true,
Elem: &pluginsdk.Resource{
Schema: map[string]*pluginsdk.Schema{
"weekday": {
Type: pluginsdk.TypeString,
Computed: true,
},

"week": {
Type: pluginsdk.TypeInt,
Computed: true,
},
},
},
},
},
},
},

"start_time": {
Type: pluginsdk.TypeString,
Computed: true,
},

"end_time": {
Type: pluginsdk.TypeString,
Computed: true,
},

"time_zone": {
Type: pluginsdk.TypeString,
Computed: true,
},

"frequency": {
Type: pluginsdk.TypeString,
Computed: true,
},

"interval": {
Type: pluginsdk.TypeInt,
Computed: true,
},

"activated": {
Type: pluginsdk.TypeBool,
Computed: true,
},

"pipeline_name": {
Type: pluginsdk.TypeString,
Computed: true,
},

"annotations": {
Type: pluginsdk.TypeList,
Computed: true,
Elem: &pluginsdk.Schema{
Type: pluginsdk.TypeString,
},
},
}
}

func (d TriggerScheduleDataSource) Read() sdk.ResourceFunc {
return sdk.ResourceFunc{
Timeout: 5 * time.Minute,
Func: func(ctx context.Context, metadata sdk.ResourceMetaData) error {
var model TriggerScheduleDataSourceModel
if err := metadata.Decode(&model); err != nil {
return err
}

subscriptionId := metadata.Client.Account.SubscriptionId
client := metadata.Client.DataFactory.TriggersClient

dataFactoryId, err := factories.ParseFactoryID(model.DataFactoryID)
if err != nil {
return err
}

id := parse.NewTriggerID(subscriptionId, dataFactoryId.ResourceGroupName, dataFactoryId.FactoryName, model.Name)
if err != nil {
return err
}

existing, err := client.Get(ctx, id.ResourceGroup, id.FactoryName, id.Name, "")
if err != nil {
if utils.ResponseWasNotFound(existing.Response) {
return fmt.Errorf("Trigger %s not found", id)
}
return fmt.Errorf("retreiving Trigger %s: %+v", id, err)
}

metadata.SetID(id)

model.Name = *existing.Name
model.DataFactoryID = dataFactoryId.ID()

scheduleTriggerProps, ok := existing.Properties.AsScheduleTrigger()
if !ok {
return fmt.Errorf("classifying %s: Expected: %q Received: %q", id.ID(), datafactory.TypeBasicTriggerTypeScheduleTrigger, *existing.Type)
}

if scheduleTriggerProps != nil {
model.Activated = scheduleTriggerProps.RuntimeState == datafactory.TriggerRuntimeStateStarted

if recurrence := scheduleTriggerProps.Recurrence; recurrence != nil {
if v := recurrence.StartTime; v != nil {
model.StartTime = v.Format(time.RFC3339)
}
if v := recurrence.EndTime; v != nil {
model.EndTime = v.Format(time.RFC3339)
}
model.Frequency = string(recurrence.Frequency)
model.Interval = int(*recurrence.Interval)
model.TimeZone = *recurrence.TimeZone

if schedule := recurrence.Schedule; schedule != nil {
model.Schedule = flattenDataFactoryScheduleModel(schedule)
}
}

if pipelines := scheduleTriggerProps.Pipelines; pipelines != nil {
if len(*pipelines) > 0 {
pipeline := *pipelines
if reference := pipeline[0].PipelineReference; reference != nil {
model.PipelineName = *reference.ReferenceName
}
}
}

model.Annotations = flattenDataFactoryAnnotations(scheduleTriggerProps.Annotations)

if scheduleTriggerProps.Description != nil {
model.Description = *scheduleTriggerProps.Description
}

}
if err := metadata.Encode(&model); err != nil {
return fmt.Errorf("encoding: %+v", err)
}

return metadata.Encode(&model)
},
}
}

func flattenDataFactoryScheduleModel(schedule *datafactory.RecurrenceSchedule) []TriggerScheduleSchedule {
if schedule == nil {
return []TriggerScheduleSchedule{}
}

result := TriggerScheduleSchedule{}
results := []TriggerScheduleSchedule{}

if schedule.Hours != nil {
for _, v := range *schedule.Hours {
result.Hours = append(result.Hours, int(v))
}
}

if schedule.Minutes != nil {
for _, v := range *schedule.Minutes {
result.Minutes = append(result.Minutes, int(v))
}
}

if schedule.MonthDays != nil {
for _, v := range *schedule.MonthDays {
result.DaysOfMonth = append(result.DaysOfMonth, int(v))
}
}

if schedule.WeekDays != nil {
weekDays := make([]string, 0)
for _, v := range *schedule.WeekDays {
weekDays = append(weekDays, string(v))
}
result.DaysOfWeek = weekDays
}

if schedule.MonthlyOccurrences != nil {
var monthlyOccurrences []TriggerScheduleScheduleMonthly
for _, v := range *schedule.MonthlyOccurrences {
occurrence := TriggerScheduleScheduleMonthly{}
occurrence.Weekday = string(v.Day)
if v.Occurrence != nil {
occurrence.Week = int(*v.Occurrence)
}
monthlyOccurrences = append(monthlyOccurrences, occurrence)
}
result.Monthly = monthlyOccurrences
}
results = append(results, result)
return results
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0

package datafactory_test

import (
"fmt"
"testing"

"github.com/hashicorp/terraform-provider-azurerm/internal/acceptance"
"github.com/hashicorp/terraform-provider-azurerm/internal/acceptance/check"
)

type DataFactoryTriggerScheduleDataSource struct{}

func TestAccDataFactoryTriggerScheduleDataSource_basic(t *testing.T) {
data := acceptance.BuildTestData(t, "data.azurerm_data_factory_trigger_schedule", "test")
r := DataFactoryTriggerScheduleDataSource{}

data.DataSourceTest(t, []acceptance.TestStep{
{
Config: r.basic(data),
Check: acceptance.ComposeTestCheckFunc(
check.That(data.ResourceName).Key("name").Exists(),
),
},
})
}

func (DataFactoryTriggerScheduleDataSource) basic(data acceptance.TestData) string {
return fmt.Sprintf(`
%s
data "azurerm_data_factory_trigger_schedule" "test" {
name = azurerm_data_factory_trigger_schedule.test.name
data_factory_id = azurerm_data_factory.test.id
}
`, TriggerScheduleResource{}.basic(data))
}
Loading

0 comments on commit b7a4641

Please sign in to comment.