table_schema.go

package plugin

import (
	"github.com/turbot/steampipe-plugin-sdk/grpc/proto"
)

func (t Table) GetSchema() *proto.TableSchema {
	schema := &proto.TableSchema{
		Columns:     make([]*proto.ColumnDefinition, len(t.Columns)),
		Description: t.Description,
	}

	// column schema
	for i, column := range t.Columns {
		schema.Columns[i] = &proto.ColumnDefinition{
			Name:        column.Name,
			Type:        column.Type,
			Description: column.Description,
		}
	}
	// key columns
	if t.Get != nil && len(t.Get.KeyColumns) > 0 {
		schema.GetCallKeyColumnList = t.Get.KeyColumns.ToProtobuf()
	}
	if t.List != nil {
		if len(t.List.KeyColumns) > 0 {
			schema.ListCallKeyColumnList = t.List.KeyColumns.ToProtobuf()
		}

	}

	return schema
}