Function getWriteTimestamp

dgraph/cmd/bulk/loader.go:223–248

(zero *grpc.ClientConn, dg *dgo.Dgraph)

}

func getWriteTimestamp(zero *grpc.ClientConn, dg *dgo.Dgraph) uint64 {
	if zero != nil {
		client := pb.NewZeroClient(zero)
		for {
			ctx, cancel := context.WithTimeout(context.Background(), time.Second)
			ts, err := client.Timestamps(ctx, &pb.Num{Val: 1})
			cancel()
			if err == nil {
				return ts.GetStartId()
			}
			fmt.Printf("Error communicating with dgraph zero, retrying: %v", err)
			time.Sleep(time.Second)
		}
	}

	for {
		ctx, cancel := context.WithTimeout(context.Background(), time.Second)
		_, ts, err := dg.AllocateTimestamps(ctx, 1)
		cancel()
		if err == nil {
			return ts
		}
		fmt.Printf("Error communicating with dgraph alpha, retrying: %v", err)
		time.Sleep(time.Second)
	}
}

// leaseNamespace is called at the end of map phase. It leases the namespace ids till the maximum
// seen namespace id.