我正在从GCP中的云函数触发数据流作业。
嵌入云函数的代码
import apache_beam as beam
import argparse
PROJECT = 'projectName'
BUCKET='bucketName'
SCHEMA = 'sr:INTEGER,abv:FLOAT,id:INTEGER,name:STRING,style:STRING,ounces:FLOAT'
DATAFLOW_JOB_NAME = 'jobName'
def discard_incomplete(data):
"""Filters out records that don't have an information."""
return len(data['abv']) > 0 and len(data['id']) > 0 and len(data['name']) > 0 and len(data['style']) > 0
def convert_types(data):
"""Converts string values to their appropriate type."""
data['abv'] = float(data['abv']) if 'abv' in data else None
data['id'] = int(data['id']) if 'id' in data else None
data['name'] = str(data['name']) if 'name' in data else None
data['style'] = str(data['style']) if 'style' in data else None
data['ounces'] = float(data['ounces']) if 'ounces' in data else None
return data
def del_unwanted_cols(data):
"""Delete the unwanted columns"""
del data['ibu']
del data['brewery_id']
return data
def execute(event, context):
argv = [
'--project={0}'.format(PROJECT),
'--job_name={0}'.format(DATAFLOW_JOB_NAME),
'--staging_location=gs://{0}/staging/'.format(BUCKET),
'--temp_location=gs://{0}/staging/'.format(BUCKET),
'--region=us-central1',
'--runner=DataflowRunner'
]
p = beam.Pipeline(argv=argv)
input = 'gs://{0}/beers.csv'.format(BUCKET)
(p | 'ReadData' >> beam.io.ReadFromText(input, skip_header_lines =1)
| 'SplitData' >> beam.Map(lambda x: x.split(','))
| 'FormatToDict' >> beam.Map(lambda x: {"sr": x[0], "abv": x[1], "ibu": x[2], "id": x[3], "name": x[4], "style": x[5], "brewery_id": x[6], "ounces": x[7]})
| 'DeleteIncompleteData' >> beam.Filter(discard_incomplete)
| 'ChangeDataType' >> beam.Map(convert_types)
| 'DeleteUnwantedData' >> beam.Map(del_unwanted_cols)
| 'WriteToBigQuery' >> beam.io.WriteToBigQuery(
'{0}:sandeep_beer_test.beer_data'.format(PROJECT),
schema=SCHEMA,
write_disposition=beam.io.BigQueryDisposition.WRITE_APPEND))
p.run()当云函数被执行时,数据流作业确实会被触发,但是作业总是失败。当我检查作业日志时,我看到以下错误消息- ModuleNotFoundError: No module named 'google.cloud.functions'
requirements.txt
apache-beam[gcp]
如果我在安装apache之后直接从云shell运行,云函数中嵌入的python代码工作得很好。
请分享您的输入,如何通过数据流错误的缺失模块。
谢谢
桑迪普
发布于 2021-02-04 05:51:52
这很可能是因为您要通过--save_main_session。
https://stackoverflow.com/questions/66038332
复制相似问题