bigquery

type Client interface {
	Init(ctx context.Context) (err error)
	CheckIfDatasetExists(ctx context.Context) (exists bool)
	CheckIfTableExists(ctx context.Context, table string) (exists bool)
	CreateTable(ctx context.Context, table string, typeForSchema interface{}, partitionField string, waitReady bool) (err error)
	UpdateTableSchema(ctx context.Context, table string, typeForSchema interface{}) (err error)
	InsertBuildEvent(ctx context.Context, event PipelineBuildEvent) (err error)
	InsertReleaseEvent(ctx context.Context, event PipelineReleaseEvent) (err error)
}

type Job struct {
	JobID  int `bigquery:"job_id"`
	Stages []struct {
		Name           string `bigquery:"name"`
		ContainerImage struct {
			Name         string        `bigquery:"name"`
			Tag          string        `bigquery:"tag"`
			IsPulled     bool          `bigquery:"is_pulled"`
			ImageSize    int           `bigquery:"image_size"`
			PullDuration time.Duration `bigquery:"pull_duration"`
			IsTrusted    bool          `bigquery:"is_trusted"`
		} `bigquery:"container_image"`

		RunDuration time.Duration `bigquery:"run_duration"`
		LogLines    []struct {
			Timestamp  time.Time `bigquery:"timestamp"`
			StreamType string    `bigquery:"stream_type"`
			Text       string    `bigquery:"text"`
		} `bigquery:"log_lines"`
	} `bigquery:"stages"`
	InsertedAt time.Time `bigquery:"inserted_at"`
}

type MockClient struct {
	// contains filtered or unexported fields
}

type MockClientMockRecorder struct {
	// contains filtered or unexported fields
}

type PipelineBuildEvent struct {
	BuildID      int    `bigquery:"build_id"`
	RepoSource   string `bigquery:"repo_source"`
	RepoOwner    string `bigquery:"repo_owner"`
	RepoName     string `bigquery:"repo_name"`
	RepoBranch   string `bigquery:"repo_branch"`
	RepoRevision string `bigquery:"repo_revision"`
	BuildVersion string `bigquery:"build_version"`
	BuildStatus  string `bigquery:"build_status"`

	Labels []struct {
		Key   string `bigquery:"key"`
		Value string `bigquery:"value"`
	} `bigquery:"labels"`

	InsertedAt time.Time `bigquery:"inserted_at"`
	UpdatedAt  time.Time `bigquery:"updated_at"`

	Commits []struct {
		Message string `bigquery:"message"`
		Author  struct {
			Email string `bigquery:"email"`
		} `bigquery:"author"`
	} `bigquery:"commits"`

	CPURequest     bigquery.NullFloat64 `bigquery:"cpu_request"`
	CPULimit       bigquery.NullFloat64 `bigquery:"cpu_limit"`
	CPUMaxUsage    bigquery.NullFloat64 `bigquery:"cpu_max_usage"`
	MemoryRequest  bigquery.NullFloat64 `bigquery:"memory_request"`
	MemoryLimit    bigquery.NullFloat64 `bigquery:"memory_limit"`
	MemoryMaxUsage bigquery.NullFloat64 `bigquery:"memory_max_usage"`

	TotalDuration time.Duration `bigquery:"duration"`
	TimeToRunning time.Duration `bigquery:"time_to_running"`

	Manifest string `bigquery:"manifest"`

	Jobs []Job `bigquery:"logs"`
}

type PipelineReleaseEvent struct {
	ReleaseID      int    `bigquery:"release_id"`
	RepoSource     string `bigquery:"repo_source"`
	RepoOwner      string `bigquery:"repo_owner"`
	RepoName       string `bigquery:"repo_name"`
	ReleaseTarget  string `bigquery:"release_target"`
	ReleaseVersion string `bigquery:"release_version"`
	ReleaseStatus  string `bigquery:"release_status"`

	Labels []struct {
		Key   string `bigquery:"key"`
		Value string `bigquery:"value"`
	} `bigquery:"labels"`

	InsertedAt time.Time `bigquery:"inserted_at"`
	UpdatedAt  time.Time `bigquery:"updated_at"`

	CPURequest     bigquery.NullFloat64 `bigquery:"cpu_request"`
	CPULimit       bigquery.NullFloat64 `bigquery:"cpu_limit"`
	CPUMaxUsage    bigquery.NullFloat64 `bigquery:"cpu_max_usage"`
	MemoryRequest  bigquery.NullFloat64 `bigquery:"memory_request"`
	MemoryLimit    bigquery.NullFloat64 `bigquery:"memory_limit"`
	MemoryMaxUsage bigquery.NullFloat64 `bigquery:"memory_max_usage"`

	TotalDuration time.Duration `bigquery:"duration"`
	TimeToRunning time.Duration `bigquery:"time_to_running"`

	Jobs []Job `bigquery:"logs"`
}