示例#1
0
func restore(domain *db.Domain) {
	user_rr := getRecordReader(*backup_dir, "test-users.User.rec")
	station_rr := getRecordReader(*backup_dir, "test-stations.Station.rec")
	comment_rr := getRecordReader(*backup_dir, "test-comments.Comment.rec")
	contact_rr := getRecordReader(*backup_dir, "test-contacts.Contact.rec")

	userdb := domain.NewUserDB()
	if err := userdb.Create(); err != nil {
		log.Fatalf("Error creating user table: %s", err.Error())
	}
	stationdb := domain.NewStationDB()
	if err := stationdb.Create(); err != nil {
		log.Fatalf("Error creating station table: %s", err.Error())
	}
	commentdb := domain.NewCommentDB()
	if err := commentdb.Create(); err != nil {
		log.Fatalf("Error creating comment table: %s", err.Error())
	}
	contactdb := domain.NewContactDB()
	if err := contactdb.Create(); err != nil {
		log.Fatalf("Error creating contact table: %s", err.Error())
	}

	if err := RestoreUserTable(user_rr, userdb); err != nil {
		log.Fatalf("Error restoring user table: %s", err.Error())
	}
	if err := RestoreStationsTable(station_rr, stationdb); err != nil {
		log.Fatalf("Error restoring station table: %s", err.Error())
	}
	if err := RestoreCommentsTable(comment_rr, commentdb); err != nil {
		log.Fatalf("Error restoring comments table: %s", err.Error())
	}
	if err := RestoreContactsTable(contact_rr, contactdb); err != nil {
		log.Fatalf("Error restoring contacts table: %s", err.Error())
	}
}