diff --git a/service/trade-service-source/trade-service-source-core/src/main/java/com/hchbox/controller/ShopController.java b/service/trade-service-source/trade-service-source-core/src/main/java/com/hchbox/controller/ShopController.java new file mode 100644 index 0000000..f5bd13c --- /dev/null +++ b/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> queryNick() { + return new CommonResult>().success(shopService.queryNick()); + } +} diff --git a/service/trade-service-source/trade-service-source-dao/src/main/java/com/hchbox/entity/master/Shop.java b/service/trade-service-source/trade-service-source-dao/src/main/java/com/hchbox/entity/master/Shop.java new file mode 100644 index 0000000..c0758fa --- /dev/null +++ b/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; +} diff --git a/service/trade-service-source/trade-service-source-dao/src/main/java/com/hchbox/repository/master/ShopRepository.java b/service/trade-service-source/trade-service-source-dao/src/main/java/com/hchbox/repository/master/ShopRepository.java new file mode 100644 index 0000000..52c3e96 --- /dev/null +++ b/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, JpaSpecificationExecutor { + + @Query(value = "select * from nick_name where status = 0 limit 0 , ?1" ,nativeQuery = true) + List findNice(Integer num); +} diff --git a/service/trade-service-source/trade-service-source-taobao/src/main/java/com/hchbox/service/ShopService.java b/service/trade-service-source/trade-service-source-taobao/src/main/java/com/hchbox/service/ShopService.java new file mode 100644 index 0000000..335ebe4 --- /dev/null +++ b/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 queryNick(); +} diff --git a/service/trade-service-source/trade-service-source-taobao/src/main/java/com/hchbox/service/impl/ShopServiceImpl.java b/service/trade-service-source/trade-service-source-taobao/src/main/java/com/hchbox/service/impl/ShopServiceImpl.java new file mode 100644 index 0000000..384037b --- /dev/null +++ b/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 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 queryNick() { + List 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; + } +}