读取文件Nickname

service/trade-service-source/trade-service-source-core/src/main/java/com/hchbox/controller/ShopController.java

@@ -0,0 +1,37 @@
+package com.hchbox.controller;
+
+import com.hchbox.entity.master.Shop;
+import com.hchbox.model.param.CommonResult;
+import com.hchbox.service.ShopService;
+import com.taobao.api.domain.Item;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import lombok.AllArgsConstructor;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.List;
+
+/**
+ * @author YQY
+ * @date : 2021/11/23
+ */
+@RestController
+@AllArgsConstructor
+@Api(tags = "店铺操作")
+@RequestMapping("/source/shop")
+public class ShopController {
+
+    private final ShopService shopService;
+
+    @GetMapping("/readFile")
+    @ApiOperation("读取文件")
+    public void readFile() {
+        shopService.readFile();
+    }
+
+    @GetMapping("/queryNick")
+    @ApiOperation("查询nick_name")
+    public CommonResult<List<Shop>> queryNick() {
+        return new CommonResult<List<Shop>>().success(shopService.queryNick());
+    }
+}

service/trade-service-source/trade-service-source-dao/src/main/java/com/hchbox/entity/master/Shop.java

@@ -0,0 +1,35 @@
+package com.hchbox.entity.master;
+
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.Getter;
+import lombok.Setter;
+
+import javax.persistence.*;
+import java.io.Serializable;
+
+/**
+ * @author YQY
+ * @date : 2021/11/23
+ */
+@Data
+@Entity
+@Table(name = "nick_name")
+public class Shop implements Serializable {
+
+    private static final long serialVersionUID = -5051120357215189176L;
+
+    @Id
+    @Column(name = "id")
+    @ApiModelProperty(value = "id")
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @Column(name = "nk_name")
+    @ApiModelProperty(value = "店铺名称")
+    private String nkName;
+
+    @Column(name = "status")
+    @ApiModelProperty(value = "状态")
+    private Integer status;
+}

service/trade-service-source/trade-service-source-dao/src/main/java/com/hchbox/repository/master/ShopRepository.java

@@ -0,0 +1,21 @@
+package com.hchbox.repository.master;
+
+import com.hchbox.entity.master.Shop;
+import com.hchbox.model.param.CommonResult;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.stereotype.Repository;
+
+import java.util.List;
+
+/**
+ * @author YQY
+ * @date : 2021/11/23
+ */
+@Repository
+public interface ShopRepository extends JpaRepository<Shop, Long>, JpaSpecificationExecutor<Shop> {
+
+    @Query(value = "select * from nick_name where status = 0 limit 0 , ?1" ,nativeQuery = true)
+    List<Shop> findNice(Integer num);
+}

service/trade-service-source/trade-service-source-taobao/src/main/java/com/hchbox/service/ShopService.java

@@ -0,0 +1,25 @@
+package com.hchbox.service;
+
+import com.hchbox.entity.master.Shop;
+import com.hchbox.model.param.CommonResult;
+
+import java.util.List;
+
+/**
+ * @author YQY
+ * @date : 2021/11/23
+ */
+public interface ShopService {
+
+    /**
+     * 读文件
+     */
+    void readFile();
+
+    /**
+     * 查询nick_name
+     *
+     * @return {@link CommonResult}<{@link List}<{@link Shop}>>
+     */
+    List<Shop> queryNick();
+}

service/trade-service-source/trade-service-source-taobao/src/main/java/com/hchbox/service/impl/ShopServiceImpl.java

@@ -0,0 +1,59 @@
+package com.hchbox.service.impl;
+
+import cn.hutool.core.io.file.FileReader;
+import com.hchbox.entity.master.Shop;
+import com.hchbox.model.param.CommonResult;
+import com.hchbox.repository.master.ShopRepository;
+import com.hchbox.service.ShopService;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+/**
+ * @author YQY
+ * @date : 2021/11/23
+ */
+@Slf4j
+@Service
+@RequiredArgsConstructor
+public class ShopServiceImpl implements ShopService {
+
+    private final ShopRepository shopRepository;
+    private static Integer NUM = 30;
+
+    @Override
+    public void readFile() {
+        FileReader fileReader = new FileReader("/usr/local/webapp/店铺名称统计.txt");
+        String[] nickNames = fileReader.readString().split(",");
+        if (nickNames.length > 0) {
+            Set<String> set = new HashSet();
+            for (int i = 0; i < nickNames.length; i++) {
+                set.add(nickNames[i]);
+            }
+            set.forEach(c -> {
+                Shop shop = new Shop();
+                shop.setNkName(c);
+                shop.setStatus(0);
+                shopRepository.save(shop);
+            });
+        }
+    }
+
+    @Override
+    public List<Shop> queryNick() {
+        List<Shop> nice = shopRepository.findNice(NUM);
+        if (nice.size() > 0) {
+            for (Shop shop : nice) {
+                Shop sp = new Shop();
+                sp.setId(shop.getId());
+                sp.setStatus(1);
+                shopRepository.save(sp);
+            }
+        }
+        return nice;
+    }
+}