import sys

import traceback

from datahub import DataHub

from datahub.exceptions import ResourceExistException

from datahub.models import FieldType, RecordSchema, TupleRecord, BlobRecord, CursorType, RecordType

access_id =

access_key =

endpoint = 'https://dh-cn-shanghai.aliyuncs.com'

dh = DataHub(access_id, access_key, endpoint)

##写入

project_name=

topic_name =

try:

    # block等待所有shard状态ready

    dh.wait_shards_ready(project_name, topic_name)

    print("shards all ready!!!")

    print("=======================================\n\n")

    topic_result = dh.get_topic(project_name, topic_name)

    print(topic_result)

    if topic_result.record_type != RecordType.TUPLE:

        print("topic type illegal!")

        sys.exit(-1)

    print("topic type normal")

    print("=======================================\n\n")

    record_schema = topic_result.record_schema

    records0 = []

    record0 = TupleRecord(schema=record_schema, values=['1', '2yc1', '30.01', '4True', '5455869335000000','6','1455869335000000'])

    record0.shard_id = '0'

    record0.put_attribute('AK', '47')

    records0.append(record0)

    for i in range (1,10000):

        record2 = TupleRecord(schema=record_schema)

        record2.set_value(0, str(i))

        record2.set_value(1, str(i)+'yc3')

        record2.set_value(2,  str(i+1.1))

        record2.set_value(3, str(i))

        record2.set_value(4, '1455869335000011')

        record2.set_value(5, '20180913_1115')

        record2.set_value(6, int(time.time())*1000000)

        record2.attributes = {'key': 'value'}

        record2.partition_key = 'EVENT_TIME'

        records0.append(record2)

    put_result = dh.put_records(project_name, topic_name, records0)

    print(put_result)

    print("put tuple %d records, failed count: %d" %(len(records0), put_result.failed_record_count))

    # failed_record_count如果大于0最好对failed record再进行重试

    print('结束')

    print

    print("=======================================\n\n")

except DatahubException as e:

    print(e)

    sys.exit(-1)