example.UserInitializer.java Source code

Introduction

Here is the source code for example.UserInitializer.java

Source

/*
 * Copyright 2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package example;

import static org.springframework.util.StringUtils.*;

import example.User.Address;
import example.User.Picture;
import lombok.RequiredArgsConstructor;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

import org.springframework.batch.item.ExecutionContext;
import org.springframework.batch.item.file.FlatFileItemReader;
import org.springframework.batch.item.file.mapping.DefaultLineMapper;
import org.springframework.batch.item.file.separator.DefaultRecordSeparatorPolicy;
import org.springframework.batch.item.file.transform.DelimitedLineTokenizer;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.UrlResource;
import org.springframework.util.StringUtils;
import org.springframework.web.util.UriTemplate;

/**
 * Initialize {@link UserRepository} with sample data.
 * 
 * @author Christoph Strobl
 * @author Oliver Gierke
 */
@RequiredArgsConstructor
public class UserInitializer {

    private static final UriTemplate REMOTE_TEMPLATE = new UriTemplate(
            "https://randomuser.me/api/?results={numberOfUsers}&format=csv&nat=US");

    private final UserRepository repository;

    public void initLocally() throws Exception {

        List<User> users = readUsers(new ClassPathResource("randomuser.me.csv"));

        repository.deleteAll();
        repository.save(users);
    }

    public void initRemote(int numberOfUsers) throws Exception {

        List<User> users = readUsers(new UrlResource(REMOTE_TEMPLATE.expand(numberOfUsers)));

        repository.deleteAll();
        repository.save(users);
    }

    private static List<User> readUsers(Resource resource) throws Exception {

        Scanner scanner = new Scanner(resource.getInputStream());
        String line = scanner.nextLine();
        scanner.close();

        FlatFileItemReader<User> reader = new FlatFileItemReader<User>();
        reader.setResource(resource);

        DelimitedLineTokenizer tokenizer = new DelimitedLineTokenizer();
        tokenizer.setNames(line.split(","));
        tokenizer.setStrict(false);

        DefaultLineMapper<User> lineMapper = new DefaultLineMapper<User>();
        lineMapper.setFieldSetMapper(fields -> {

            User user = new User();

            user.setEmail(fields.readString("email"));
            user.setFirstname(capitalize(fields.readString("first")));
            user.setLastname(capitalize(fields.readString("last")));
            user.setNationality(fields.readString("nationality"));

            String city = Arrays.stream(fields.readString("city").split(" "))//
                    .map(StringUtils::capitalize)//
                    .collect(Collectors.joining(" "));
            String street = Arrays.stream(fields.readString("street").split(" "))//
                    .map(StringUtils::capitalize)//
                    .collect(Collectors.joining(" "));

            try {
                user.setAddress(new Address(city, street, fields.readString("zip")));
            } catch (IllegalArgumentException e) {
                user.setAddress(new Address(city, street, fields.readString("postcode")));
            }

            user.setPicture(new Picture(fields.readString("large"), fields.readString("medium"),
                    fields.readString("thumbnail")));
            user.setUsername(fields.readString("username"));
            user.setPassword(fields.readString("password"));

            return user;
        });

        lineMapper.setLineTokenizer(tokenizer);

        reader.setLineMapper(lineMapper);
        reader.setRecordSeparatorPolicy(new DefaultRecordSeparatorPolicy());
        reader.setLinesToSkip(1);
        reader.open(new ExecutionContext());

        List<User> users = new ArrayList<>();
        User user = null;

        do {

            user = reader.read();

            if (user != null) {
                users.add(user);
            }

        } while (user != null);

        return users;
    }
}