WebFlux Web CRUD 实践

Author: liqiangqiang

springboot-webflux-2-restful/src/main/java/org/spring/springboot/controller/CityRestController.java

@@ -0,0 +1,43 @@
+package org.spring.springboot.dao;
+
+import org.spring.springboot.domain.City;
+import org.springframework.stereotype.Repository;
+
+import java.util.Collection;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+import java.util.concurrent.atomic.AtomicLong;
+
+@Repository
+public class CityRepository {
+
+    private ConcurrentMap<Long, City> repository = new ConcurrentHashMap<>();
+
+    private static final AtomicLong idGenerator = new AtomicLong(0);
+
+    public Long save(City city) {
+        Long id = idGenerator.incrementAndGet();
+        city.setId(id);
+        repository.put(id, city);
+        return id;
+    }
+
+    public Collection<City> findAll() {
+        return repository.values();
+    }
+
+
+    public City findCityById(Long id) {
+        return repository.get(id);
+    }
+
+    public Long updateCity(City city) {
+        repository.put(city.getId(), city);
+        return city.getId();
+    }
+
+    public Long deleteCity(Long id) {
+        repository.remove(id);
+        return id;
+    }
+}

springboot-webflux-2-restful/src/main/java/org/spring/springboot/dao/CityRepository.java

@@ -0,0 +1,42 @@
+package org.spring.springboot.handler;
+
+import org.spring.springboot.dao.CityRepository;
+import org.spring.springboot.domain.City;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+@Component
+public class CityHandler {
+
+    private final CityRepository cityRepository;
+
+    @Autowired
+    public CityHandler(CityRepository cityRepository) {
+        this.cityRepository = cityRepository;
+    }
+
+    public Mono<Long> save(City city) {
+        return Mono.create(cityMonoSink -> cityMonoSink.success(cityRepository.save(city)));
+    }
+
+    public Mono<City> findCityById(Long id) {
+        return Mono.create(cityMonoSink -> cityMonoSink.success(cityRepository.findCityById(id)));
+    }
+
+    public Flux<City> findAllCity() {
+        return Flux.create(cityFluxSink -> {
+            cityRepository.findAll().forEach(city -> cityFluxSink.next(city));
+            cityFluxSink.complete();
+        });
+    }
+
+    public Mono<Long> modifyCity(City city) {
+        return Mono.create(cityMonoSink -> cityMonoSink.success(cityRepository.updateCity(city)));
+    }
+
+    public Mono<Long> deleteCity(Long id) {
+        return Mono.create(cityMonoSink -> cityMonoSink.success(cityRepository.deleteCity(id)));
+    }
+}

springboot-webflux-2-restful/src/main/java/org/spring/springboot/handler/CityHandler.java

@@ -0,0 +1,41 @@
+package org.spring.springboot.webflux.controller;
+
+import org.spring.springboot.domain.City;
+import org.spring.springboot.handler.CityHandler;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+@RestController
+@RequestMapping(value = "/city")
+public class CityWebFluxController {
+
+    @Autowired
+    private CityHandler cityHandler;
+
+    @GetMapping(value = "/{id}")
+    public Mono<City> findCityById(@PathVariable("id") Long id) {
+        return cityHandler.findCityById(id);
+    }
+
+    @GetMapping()
+    public Flux<City> findAllCity() {
+        return cityHandler.findAllCity();
+    }
+
+    @PostMapping()
+    public Mono<Long> saveCity(@RequestBody City city) {
+        return cityHandler.save(city);
+    }
+
+    @PutMapping()
+    public Mono<Long> modifyCity(@RequestBody City city) {
+        return cityHandler.modifyCity(city);
+    }
+
+    @DeleteMapping(value = "/{id}")
+    public Mono<Long> deleteCity(@PathVariable("id") Long id) {
+        return cityHandler.deleteCity(id);
+    }
+}