Browse Source
# Conflicts: # bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/constant/ServiceNameConstants.java # bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/domain/UserVip.java # bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/domain/UserVipRecord.javafeature-1.1
129 changed files with 2733 additions and 214 deletions
@ -0,0 +1,26 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<project xmlns="http://maven.apache.org/POM/4.0.0" |
|||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
|||
<parent> |
|||
<artifactId>bnyer-api</artifactId> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<version>1.0.0</version> |
|||
</parent> |
|||
<modelVersion>4.0.0</modelVersion> |
|||
|
|||
<artifactId>bnyer-api-file</artifactId> |
|||
|
|||
<description> |
|||
bnyer-api-file文件服务接口模块 |
|||
</description> |
|||
|
|||
<dependencies> |
|||
<!-- bnyer Common Core--> |
|||
<dependency> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<artifactId>bnyer-common-core</artifactId> |
|||
</dependency> |
|||
</dependencies> |
|||
|
|||
</project> |
|||
@ -0,0 +1,40 @@ |
|||
package com.bnyer.file.api; |
|||
|
|||
import com.bnyer.common.core.constant.ServiceNameConstants; |
|||
import com.bnyer.common.core.domain.R; |
|||
import com.bnyer.file.api.factory.RemoteFileFallbackFactory; |
|||
import org.springframework.cloud.openfeign.FeignClient; |
|||
import org.springframework.http.MediaType; |
|||
import org.springframework.web.bind.annotation.PostMapping; |
|||
import org.springframework.web.bind.annotation.RequestPart; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import java.util.List; |
|||
|
|||
/** |
|||
* 文件服务 |
|||
* |
|||
* @author penny |
|||
* @date 2023/04/15 11:03 |
|||
*/ |
|||
@FeignClient(contextId = "remoteFileService", value = ServiceNameConstants.FILE_SERVICE, fallbackFactory = RemoteFileFallbackFactory.class) |
|||
public interface RemoteFileService { |
|||
|
|||
|
|||
/** |
|||
* 上传文件到minio |
|||
* @param file 文件 |
|||
* @return - |
|||
*/ |
|||
@PostMapping(value = "/upload",consumes = MediaType.MULTIPART_FORM_DATA_VALUE) |
|||
R<String> uploadBanner(@RequestPart(name = "file") MultipartFile file); |
|||
|
|||
/** |
|||
* 批量上传文件到七牛云 |
|||
* @param files 文件 |
|||
* @return - |
|||
*/ |
|||
@PostMapping("/uploadBatch") |
|||
R<List<String>> uploadBatch(MultipartFile[] files); |
|||
|
|||
} |
|||
@ -0,0 +1,40 @@ |
|||
package com.bnyer.file.api.factory; |
|||
|
|||
import com.bnyer.common.core.domain.R; |
|||
import com.bnyer.file.api.RemoteFileService; |
|||
import org.slf4j.Logger; |
|||
import org.slf4j.LoggerFactory; |
|||
import org.springframework.cloud.openfeign.FallbackFactory; |
|||
import org.springframework.stereotype.Component; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import java.util.List; |
|||
|
|||
/** |
|||
* 文件服务降级处理 |
|||
* |
|||
* @author penny |
|||
*/ |
|||
@Component |
|||
public class RemoteFileFallbackFactory implements FallbackFactory<RemoteFileService> |
|||
{ |
|||
private static final Logger log = LoggerFactory.getLogger(RemoteFileFallbackFactory.class); |
|||
|
|||
|
|||
@Override |
|||
public RemoteFileService create(Throwable throwable) { |
|||
log.error("api文件服务调用失败:{}", throwable.getMessage()); |
|||
return new RemoteFileService() |
|||
{ |
|||
@Override |
|||
public R<String> uploadBanner(MultipartFile file) { |
|||
return R.fail("远程调用minio文件上传失败:" + throwable.getMessage()); |
|||
} |
|||
|
|||
@Override |
|||
public R<List<String>> uploadBatch(MultipartFile[] files) { |
|||
return R.fail("远程调用七牛云批量文件上传失败:" + throwable.getMessage()); |
|||
} |
|||
}; |
|||
} |
|||
} |
|||
@ -0,0 +1,2 @@ |
|||
org.springframework.boot.autoconfigure.EnableAutoConfiguration = com.bnyer.file.api.factory.RemoteFileFallbackFactory |
|||
|
|||
@ -0,0 +1,122 @@ |
|||
package com.bnyer.common.core.domain; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.annotation.TableField; |
|||
import com.baomidou.mybatisplus.annotation.TableId; |
|||
import com.baomidou.mybatisplus.annotation.TableName; |
|||
import com.fasterxml.jackson.annotation.JsonFormat; |
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import java.io.Serializable; |
|||
import java.util.Date; |
|||
import lombok.AllArgsConstructor; |
|||
import lombok.Getter; |
|||
import lombok.NoArgsConstructor; |
|||
import lombok.Setter; |
|||
import lombok.ToString; |
|||
|
|||
/** |
|||
* ai绘画内容表 |
|||
*/ |
|||
@ApiModel(value="com-bnyer-common-core-domain-AiPaint") |
|||
@Getter |
|||
@Setter |
|||
@ToString |
|||
@AllArgsConstructor |
|||
@NoArgsConstructor |
|||
@TableName(value = "img_ai_paint") |
|||
public class AiPaint implements Serializable { |
|||
/** |
|||
* 主键Id |
|||
*/ |
|||
@TableId(value = "id", type = IdType.ASSIGN_ID) |
|||
@ApiModelProperty(value="主键Id") |
|||
private Long id; |
|||
|
|||
/** |
|||
* 作品编号 |
|||
*/ |
|||
@TableField(value = "paint_id") |
|||
@ApiModelProperty(value="作品编号") |
|||
private String paintId; |
|||
|
|||
/** |
|||
* 绘图者id |
|||
*/ |
|||
@TableField(value = "painter_id") |
|||
@ApiModelProperty(value="绘图者id") |
|||
private Long painterId; |
|||
|
|||
/** |
|||
* 绘图者昵称 |
|||
*/ |
|||
@TableField(value = "painter_name") |
|||
@ApiModelProperty(value="绘图者昵称") |
|||
private String painterName; |
|||
|
|||
/** |
|||
* 图片 |
|||
*/ |
|||
@TableField(value = "img_url") |
|||
@ApiModelProperty(value="图片") |
|||
private String imgUrl; |
|||
|
|||
/** |
|||
* 关键词 |
|||
*/ |
|||
@TableField(value = "prompt") |
|||
@ApiModelProperty(value="关键词") |
|||
private String prompt; |
|||
|
|||
/** |
|||
* 模型名称 |
|||
*/ |
|||
@TableField(value = "model") |
|||
@ApiModelProperty(value="模型风格名称") |
|||
private String model; |
|||
|
|||
/** |
|||
* 风格名称 |
|||
*/ |
|||
@TableField(value = "style_name") |
|||
@ApiModelProperty(value="风格名称") |
|||
private String styleName; |
|||
|
|||
/** |
|||
* 图片高度 |
|||
*/ |
|||
@TableField(value = "height") |
|||
@ApiModelProperty(value="图片高度") |
|||
private String height; |
|||
|
|||
/** |
|||
* 图片宽度 |
|||
*/ |
|||
@TableField(value = "width") |
|||
@ApiModelProperty(value="图片宽度") |
|||
private String width; |
|||
|
|||
/** |
|||
* 是否显示 (0->隐藏；1->显示) |
|||
*/ |
|||
@TableField(value = "is_show") |
|||
@ApiModelProperty(value="是否显示 (0->隐藏；1->显示)") |
|||
private String isShow; |
|||
|
|||
/** |
|||
* 平台(0->Hub;1->抖音;2->快手;3->微信) |
|||
*/ |
|||
@TableField(value = "`source`") |
|||
@ApiModelProperty(value="平台(0->Hub;1->抖音;2->快手;3->微信)") |
|||
private String source; |
|||
|
|||
/** |
|||
* 创建时间 |
|||
*/ |
|||
@TableField(value = "create_time") |
|||
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") |
|||
@ApiModelProperty(value="创建时间") |
|||
private Date createTime; |
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
} |
|||
@ -0,0 +1,52 @@ |
|||
package com.bnyer.common.core.domain; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.annotation.TableField; |
|||
import com.baomidou.mybatisplus.annotation.TableId; |
|||
import com.baomidou.mybatisplus.annotation.TableName; |
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.*; |
|||
|
|||
/** |
|||
* 绘画-模型风格表 |
|||
*/ |
|||
@ApiModel(value="com-bnyer-common-core-domain-PaintStyle") |
|||
@Getter |
|||
@Setter |
|||
@ToString |
|||
@AllArgsConstructor |
|||
@NoArgsConstructor |
|||
@TableName(value = "img_paint_style") |
|||
public class PaintStyle extends BaseDomain { |
|||
|
|||
/** |
|||
* 模型风格名称 |
|||
*/ |
|||
@TableField(value = "`name`") |
|||
@ApiModelProperty(value="模型风格名称") |
|||
private String name; |
|||
|
|||
/** |
|||
* 模型名称 |
|||
*/ |
|||
@TableField(value = "model_name") |
|||
@ApiModelProperty(value="模型名称") |
|||
private String modelName; |
|||
|
|||
/** |
|||
* 模型风格图片 |
|||
*/ |
|||
@TableField(value = "img_url") |
|||
@ApiModelProperty(value="模型风格图片") |
|||
private String imgUrl; |
|||
|
|||
/** |
|||
* 是否热门(0->正常;1->热门) |
|||
*/ |
|||
@TableField(value = "is_hot") |
|||
@ApiModelProperty(value="是否热门(0->正常;1->热门)") |
|||
private String isHot; |
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
} |
|||
@ -0,0 +1,38 @@ |
|||
package com.bnyer.common.core.domain; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.annotation.TableField; |
|||
import com.baomidou.mybatisplus.annotation.TableId; |
|||
import com.baomidou.mybatisplus.annotation.TableName; |
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.*; |
|||
|
|||
/** |
|||
* 绘画-提示词表 |
|||
*/ |
|||
@ApiModel(value="com-bnyer-common-core-domain-Prompt") |
|||
@Getter |
|||
@Setter |
|||
@ToString |
|||
@AllArgsConstructor |
|||
@NoArgsConstructor |
|||
@TableName(value = "img_prompt") |
|||
public class Prompt extends BaseDomain { |
|||
|
|||
/** |
|||
* 提示词 |
|||
*/ |
|||
@TableField(value = "`text`") |
|||
@ApiModelProperty(value="提示词") |
|||
private String text; |
|||
|
|||
/** |
|||
* 类型(0->绘画;1->gpt) |
|||
*/ |
|||
@TableField(value = "`type`") |
|||
@ApiModelProperty(value="类型(0->绘画;1->gpt)") |
|||
private String type; |
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
package com.bnyer.common.core.dto; |
|||
|
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
import javax.validation.constraints.NotEmpty; |
|||
import javax.validation.constraints.NotNull; |
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("ai绘画前端分页接收类") |
|||
public class AiPaintPageDto extends BasePageDto { |
|||
|
|||
@NotEmpty(message = "平台渠道不能为空!") |
|||
@ApiModelProperty(value="平台渠道") |
|||
private String source; |
|||
|
|||
@NotNull(message = "绘图者id不能为空！") |
|||
@ApiModelProperty(value="绘图者id") |
|||
private Long painterId; |
|||
} |
|||
@ -0,0 +1,41 @@ |
|||
package com.bnyer.common.core.dto; |
|||
|
|||
import com.bnyer.common.core.domain.PaintStyle; |
|||
import com.bnyer.common.core.utils.bean.BeanUtils; |
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("绘画风格入参接收类") |
|||
public class PaintStyleDto implements Serializable { |
|||
|
|||
@ApiModelProperty(value="主键id") |
|||
private Long id; |
|||
|
|||
@ApiModelProperty(value="模型风格名称") |
|||
private String name; |
|||
|
|||
@ApiModelProperty(value="模型名称") |
|||
private String modelName; |
|||
|
|||
@ApiModelProperty(value="模型风格图片") |
|||
private String imgUrl; |
|||
|
|||
@ApiModelProperty(value="是否热门(0->正常;1->热门)") |
|||
private String isHot; |
|||
|
|||
@ApiModelProperty(value="排序") |
|||
private Integer sort; |
|||
|
|||
public PaintStyle extractParam(){ |
|||
PaintStyle paintStyle = new PaintStyle(); |
|||
BeanUtils.copyProperties(this,paintStyle); |
|||
return paintStyle; |
|||
} |
|||
} |
|||
@ -0,0 +1,22 @@ |
|||
package com.bnyer.common.core.dto; |
|||
|
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("绘画风格分页接收类") |
|||
public class PaintStylePageDto extends BasePageDto { |
|||
|
|||
@ApiModelProperty(value="模型风格名称") |
|||
private String name; |
|||
|
|||
@ApiModelProperty(value="是否热门(0->正常;1->热门)") |
|||
private String isHot; |
|||
|
|||
@ApiModelProperty(value="是否显示 (0->隐藏；1->显示)") |
|||
private String isShow; |
|||
} |
|||
@ -0,0 +1,32 @@ |
|||
package com.bnyer.common.core.dto; |
|||
|
|||
import com.bnyer.common.core.domain.Prompt; |
|||
import com.bnyer.common.core.utils.bean.BeanUtils; |
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("提示词入参接收类") |
|||
public class PromptDto implements Serializable { |
|||
|
|||
@ApiModelProperty(value="主键id") |
|||
private Long id; |
|||
|
|||
@ApiModelProperty(value="提示词") |
|||
private String text; |
|||
|
|||
@ApiModelProperty(value="类型(0->绘画;1->gpt)") |
|||
private String type; |
|||
|
|||
public Prompt extractParam(){ |
|||
Prompt prompt = new Prompt(); |
|||
BeanUtils.copyProperties(this,prompt); |
|||
return prompt; |
|||
} |
|||
} |
|||
@ -0,0 +1,22 @@ |
|||
package com.bnyer.common.core.dto; |
|||
|
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("提示词分页接收类") |
|||
public class PromptPageDto extends BasePageDto { |
|||
|
|||
@ApiModelProperty(value="提示词") |
|||
private String text; |
|||
|
|||
@ApiModelProperty(value="类型(0->绘画;1->gpt)") |
|||
private String type; |
|||
|
|||
@ApiModelProperty(value="是否显示 (0->隐藏；1->显示)") |
|||
private String isShow; |
|||
} |
|||
@ -0,0 +1,29 @@ |
|||
package com.bnyer.common.core.dto; |
|||
|
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
import javax.validation.constraints.NotEmpty; |
|||
import javax.validation.constraints.NotNull; |
|||
import java.io.Serializable; |
|||
import java.util.List; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("艺术家自定义标签接收类") |
|||
public class SignFrontDto implements Serializable { |
|||
|
|||
@ApiModelProperty(value="主键id") |
|||
private Long id; |
|||
|
|||
@NotEmpty(message = "标签名称必填!") |
|||
@ApiModelProperty(value="标签名称") |
|||
private String name; |
|||
|
|||
@NotNull(message = "父id必填!") |
|||
@ApiModelProperty(value="父id") |
|||
private Long parentId; |
|||
} |
|||
@ -0,0 +1,27 @@ |
|||
package com.bnyer.common.core.utils; |
|||
|
|||
import java.util.regex.Matcher; |
|||
import java.util.regex.Pattern; |
|||
|
|||
/** |
|||
* 汉译英工具类 |
|||
*/ |
|||
public class TranslateUtils { |
|||
|
|||
/** |
|||
* 字符串是否包含中文 |
|||
* |
|||
* @param str 待校验字符串 |
|||
* @return true 包含中文字符 false 不包含中文字符 |
|||
* @throws Exception |
|||
*/ |
|||
public static boolean isContainChinese(String str) throws Exception { |
|||
|
|||
if (StringUtils.isEmpty(str)) { |
|||
throw new Exception("文本内容为空!"); |
|||
} |
|||
Pattern p = Pattern.compile("[\u4E00-\u9FA5|\\！|\\，|\\。|\\（|\\）|\\《|\\》|\\“|\\”|\\？|\\：|\\；|\\【|\\】]"); |
|||
Matcher m = p.matcher(str); |
|||
return m.find(); |
|||
} |
|||
} |
|||
@ -0,0 +1,84 @@ |
|||
package com.bnyer.common.core.utils.file; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import java.io.ByteArrayInputStream; |
|||
import java.io.File; |
|||
import java.io.FileOutputStream; |
|||
import java.io.IOException; |
|||
import java.nio.charset.StandardCharsets; |
|||
import java.util.Base64; |
|||
|
|||
/** |
|||
* @author |
|||
* @date 2023/04/16 16:40 |
|||
* @description 转换base64为文件流 |
|||
*/ |
|||
public class Base64ToMultipartFileUtils implements MultipartFile { |
|||
|
|||
private final byte[] fileContent; |
|||
|
|||
private final String name; |
|||
private final String extension; |
|||
private final String contentType; |
|||
|
|||
private final String originalFilename; |
|||
|
|||
/** |
|||
* @param base64 |
|||
* @param dataUri 格式类似于: data:image/png;base64 |
|||
*/ |
|||
public Base64ToMultipartFileUtils(String base64, String dataUri, String name, String originalfilename) { |
|||
this.fileContent = Base64.getDecoder().decode(base64.getBytes(StandardCharsets.UTF_8)); |
|||
this.extension = dataUri.split(";")[0].split("/")[1]; |
|||
this.contentType = dataUri.split(";")[0].split(":")[1]; |
|||
this.originalFilename = originalfilename; |
|||
this.name = name; |
|||
} |
|||
|
|||
/** |
|||
* 【重要】必须与请求接收方参数名称一致，否则找不到参数 |
|||
* @return |
|||
*/ |
|||
@Override |
|||
public String getName() { |
|||
return this.name; |
|||
} |
|||
|
|||
@Override |
|||
public String getOriginalFilename() { |
|||
return originalFilename; |
|||
} |
|||
|
|||
@Override |
|||
public String getContentType() { |
|||
return contentType; |
|||
} |
|||
|
|||
@Override |
|||
public boolean isEmpty() { |
|||
return fileContent == null || fileContent.length == 0; |
|||
} |
|||
|
|||
@Override |
|||
public long getSize() { |
|||
return fileContent.length; |
|||
} |
|||
|
|||
@Override |
|||
public byte[] getBytes() throws IOException { |
|||
return fileContent; |
|||
} |
|||
|
|||
@Override |
|||
public ByteArrayInputStream getInputStream() throws IOException { |
|||
return new ByteArrayInputStream(fileContent); |
|||
} |
|||
|
|||
@Override |
|||
public void transferTo(File file) throws IOException, IllegalStateException { |
|||
try (FileOutputStream fos = new FileOutputStream(file)) { |
|||
fos.write(fileContent); |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,140 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<project xmlns="http://maven.apache.org/POM/4.0.0" |
|||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
|||
<parent> |
|||
<artifactId>bnyer-services</artifactId> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<version>1.0.0</version> |
|||
</parent> |
|||
<modelVersion>4.0.0</modelVersion> |
|||
|
|||
<artifactId>bnyer-ai</artifactId> |
|||
|
|||
<description> |
|||
bnyer-ai服务 |
|||
</description> |
|||
|
|||
<dependencies> |
|||
<!-- SpringCloud Alibaba Nacos --> |
|||
<dependency> |
|||
<groupId>com.alibaba.cloud</groupId> |
|||
<artifactId>spring-cloud-starter-alibaba-nacos-discovery</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- SpringCloud Alibaba Nacos Config --> |
|||
<dependency> |
|||
<groupId>com.alibaba.cloud</groupId> |
|||
<artifactId>spring-cloud-starter-alibaba-nacos-config</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- SpringCloud Alibaba Sentinel --> |
|||
<dependency> |
|||
<groupId>com.alibaba.cloud</groupId> |
|||
<artifactId>spring-cloud-starter-alibaba-sentinel</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- SpringBoot Actuator --> |
|||
<dependency> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-starter-actuator</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- Swagger UI --> |
|||
<dependency> |
|||
<groupId>io.springfox</groupId> |
|||
<artifactId>springfox-swagger-ui</artifactId> |
|||
<version>${swagger.fox.version}</version> |
|||
</dependency> |
|||
|
|||
<!-- Mysql Connector --> |
|||
<dependency> |
|||
<groupId>mysql</groupId> |
|||
<artifactId>mysql-connector-java</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- bnyer Common DataSource --> |
|||
<dependency> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<artifactId>bnyer-common-datasource</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- bnyer Common DataScope --> |
|||
<dependency> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<artifactId>bnyer-common-datascope</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- bnyer Common Log --> |
|||
<dependency> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<artifactId>bnyer-common-log</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- bnyer Common core --> |
|||
<dependency> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<artifactId>bnyer-common-core</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- bnyer Common Swagger --> |
|||
<dependency> |
|||
<groupId>com.dimensionalnode</groupId> |
|||
<artifactId>bnyer-common-swagger</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- mybatis-plus --> |
|||
<dependency> |
|||
<groupId>com.baomidou</groupId> |
|||
<artifactId>mybatis-plus-boot-starter</artifactId> |
|||
</dependency> |
|||
|
|||
<!--任务调度中心--> |
|||
<!-- <dependency>--> |
|||
<!-- <groupId>com.xuxueli</groupId>--> |
|||
<!-- <artifactId>xxl-job-core</artifactId>--> |
|||
<!-- <version>2.3.1</version>--> |
|||
<!-- </dependency>--> |
|||
|
|||
<!--微信小程序工具包--> |
|||
<!-- <dependency>--> |
|||
<!-- <groupId>com.github.binarywang</groupId>--> |
|||
<!-- <artifactId>weixin-java-miniapp</artifactId>--> |
|||
<!-- <version>4.2.0</version>--> |
|||
<!-- </dependency>--> |
|||
|
|||
<!--七牛云--> |
|||
<!-- <dependency>--> |
|||
<!-- <groupId>com.qiniu</groupId>--> |
|||
<!-- <artifactId>qiniu-java-sdk</artifactId>--> |
|||
<!-- <version>7.2.18</version>--> |
|||
<!-- </dependency>--> |
|||
|
|||
<!--chatgpt--> |
|||
<dependency> |
|||
<groupId>com.github.plexpt</groupId> |
|||
<artifactId>chatgpt</artifactId> |
|||
<version>4.0.5</version> |
|||
</dependency> |
|||
|
|||
</dependencies> |
|||
|
|||
<build> |
|||
<finalName>${project.artifactId}</finalName> |
|||
<plugins> |
|||
<plugin> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-maven-plugin</artifactId> |
|||
<version>2.6.2</version> |
|||
<executions> |
|||
<execution> |
|||
<goals> |
|||
<goal>repackage</goal> |
|||
</goals> |
|||
</execution> |
|||
</executions> |
|||
</plugin> |
|||
</plugins> |
|||
</build> |
|||
|
|||
</project> |
|||
@ -0,0 +1,33 @@ |
|||
package com.bnyer.ai; |
|||
|
|||
import com.bnyer.common.security.annotation.EnableCustomConfig; |
|||
import com.bnyer.common.security.annotation.EnableRyFeignClients; |
|||
import com.bnyer.common.swagger.annotation.EnableCustomSwagger2; |
|||
import org.springframework.boot.SpringApplication; |
|||
import org.springframework.boot.autoconfigure.SpringBootApplication; |
|||
import org.springframework.scheduling.annotation.EnableAsync; |
|||
|
|||
/** |
|||
* ai模块 |
|||
* |
|||
* @author penny |
|||
*/ |
|||
@EnableCustomConfig |
|||
@EnableCustomSwagger2 |
|||
@EnableRyFeignClients |
|||
@SpringBootApplication |
|||
@EnableAsync |
|||
public class BnyerAiApplication |
|||
{ |
|||
public static void main(String[] args) |
|||
{ |
|||
SpringApplication.run(BnyerAiApplication.class, args); |
|||
System.out.println("(♥◠‿◠)ﾉﾞ bnyerAi服务启动成功 ლ(´ڡ`ლ)ﾞ \n" + |
|||
"__________ \n" + |
|||
"\\______ \\ ____ ___.__. ___________ \n" + |
|||
" | | _// < | |/ __ \\_ __ \\\n" + |
|||
" | | \\ | \\___ \\ ___/| | \\/\n" + |
|||
" |______ /___| / ____|\\___ >__| \n" + |
|||
" \\/ \\/\\/ \\/ \n"); |
|||
} |
|||
} |
|||
@ -0,0 +1,32 @@ |
|||
package com.bnyer.ai.controller; |
|||
|
|||
import com.alibaba.fastjson.JSON; |
|||
import com.bnyer.ai.service.ChatGptService; |
|||
import com.bnyer.common.core.domain.Feedback; |
|||
import com.bnyer.common.core.dto.FeedBackDto; |
|||
import com.bnyer.common.core.web.domain.AjaxResult; |
|||
import io.swagger.annotations.Api; |
|||
import io.swagger.annotations.ApiOperation; |
|||
import io.swagger.annotations.ApiParam; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.validation.annotation.Validated; |
|||
import org.springframework.web.bind.annotation.PostMapping; |
|||
import org.springframework.web.bind.annotation.RequestBody; |
|||
import org.springframework.web.bind.annotation.RequestMapping; |
|||
import org.springframework.web.bind.annotation.RestController; |
|||
|
|||
@Api(value = "【AI】接口",tags = "【AI】接口") |
|||
@RestController |
|||
@RequestMapping("/ai") |
|||
public class ChatGptController { |
|||
|
|||
@Autowired |
|||
private ChatGptService chatGptService; |
|||
|
|||
@ApiOperation(value="chat") |
|||
@PostMapping(value = "/chatWithGpt") |
|||
public AjaxResult chatWithGpt(){ |
|||
chatGptService.talkWithChatGpt(); |
|||
return AjaxResult.success(); |
|||
} |
|||
} |
|||
@ -0,0 +1,6 @@ |
|||
package com.bnyer.ai.service; |
|||
|
|||
public interface ChatGptService { |
|||
|
|||
void talkWithChatGpt(); |
|||
} |
|||
@ -0,0 +1,42 @@ |
|||
package com.bnyer.ai.service.impl; |
|||
|
|||
import com.bnyer.ai.service.ChatGptService; |
|||
import com.plexpt.chatgpt.ChatGPT; |
|||
import com.plexpt.chatgpt.entity.chat.ChatCompletion; |
|||
import com.plexpt.chatgpt.entity.chat.ChatCompletionResponse; |
|||
import com.plexpt.chatgpt.entity.chat.Message; |
|||
import com.plexpt.chatgpt.util.Proxys; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.stereotype.Service; |
|||
|
|||
import java.net.Proxy; |
|||
import java.util.Arrays; |
|||
|
|||
@Service |
|||
@Slf4j |
|||
public class ChatGptServiceImpl implements ChatGptService { |
|||
@Override |
|||
public void talkWithChatGpt() { |
|||
Proxy proxy = Proxys.http("18.179.21.104", 21584); |
|||
ChatGPT chatGPT = ChatGPT.builder() |
|||
.apiKey("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") |
|||
.proxy(proxy) |
|||
.timeout(100000) |
|||
.apiHost("https://api.openai.com/") //反向代理地址
|
|||
.build() |
|||
.init(); |
|||
|
|||
Message system = Message.ofSystem("你现在是一个诗人，专门写七言绝句"); |
|||
Message message = Message.of("写一段七言绝句诗，题目是：火锅！"); |
|||
|
|||
ChatCompletion chatCompletion = ChatCompletion.builder() |
|||
.model(ChatCompletion.Model.GPT_3_5_TURBO.getName()) |
|||
.messages(Arrays.asList(system, message)) |
|||
.maxTokens(3000) |
|||
.temperature(0.9) |
|||
.build(); |
|||
ChatCompletionResponse response = chatGPT.chatCompletion(chatCompletion); |
|||
Message res = response.getChoices().get(0).getMessage(); |
|||
System.out.println(res); |
|||
} |
|||
} |
|||
@ -0,0 +1,22 @@ |
|||
spring: |
|||
cloud: |
|||
nacos: |
|||
discovery: |
|||
# 服务注册地址 |
|||
server-addr: http://117.50.61.7:8848 |
|||
# 命名空间地址 |
|||
namespace: c4f53d8c-0a91-4249-a804-f16f543ec3b0 |
|||
# 命名空间分组 |
|||
group: dev |
|||
config: |
|||
# 配置中心地址 |
|||
server-addr: http://117.50.61.7:8848 |
|||
# 配置文件格式 |
|||
file-extension: yml |
|||
# 命名空间地址 |
|||
namespace: c4f53d8c-0a91-4249-a804-f16f543ec3b0 |
|||
# 命名空间分组 |
|||
group: dev |
|||
# 共享配置 |
|||
shared-configs: |
|||
- application-${spring.profiles.active}.${spring.cloud.nacos.config.file-extension} |
|||
@ -0,0 +1,26 @@ |
|||
spring: |
|||
cloud: |
|||
nacos: |
|||
discovery: |
|||
# 解决部署在不同服务器访问不到的问题，需暴露外网ip |
|||
ip: 81.69.47.31 |
|||
# 部署在不同服务器上的指定端口 |
|||
port: 9102 |
|||
# 服务注册地址 |
|||
server-addr: http://175.24.122.142:8848 |
|||
# 命名空间地址 |
|||
namespace: abfe8ee6-161b-4f8f-b61f-51663bbfa4f9 |
|||
# 命名空间分组 |
|||
group: grey |
|||
config: |
|||
# 配置中心地址 |
|||
server-addr: http://175.24.122.142:8848 |
|||
# 配置文件格式 |
|||
file-extension: yml |
|||
# 命名空间地址 |
|||
namespace: abfe8ee6-161b-4f8f-b61f-51663bbfa4f9 |
|||
# 命名空间分组 |
|||
group: grey |
|||
# 共享配置 |
|||
shared-configs: |
|||
- application-${spring.profiles.active}.${spring.cloud.nacos.config.file-extension} |
|||
@ -0,0 +1,22 @@ |
|||
spring: |
|||
cloud: |
|||
nacos: |
|||
discovery: |
|||
# 服务注册地址 |
|||
server-addr: http://192.168.3.100:6001 |
|||
# 命名空间地址 |
|||
namespace: b133c9e5-9f8d-4ed4-9ebd-95557802889f |
|||
# 命名空间分组 |
|||
group: prod |
|||
config: |
|||
# 配置中心地址 |
|||
server-addr: http://192.168.3.100:6001 |
|||
# 配置文件格式 |
|||
file-extension: yml |
|||
# 命名空间地址 |
|||
namespace: b133c9e5-9f8d-4ed4-9ebd-95557802889f |
|||
# 命名空间分组 |
|||
group: prod |
|||
# 共享配置 |
|||
shared-configs: |
|||
- application-${spring.profiles.active}.${spring.cloud.nacos.config.file-extension} |
|||
@ -0,0 +1,22 @@ |
|||
spring: |
|||
cloud: |
|||
nacos: |
|||
discovery: |
|||
# 服务注册地址 |
|||
server-addr: http://117.50.61.7:8848 |
|||
# 命名空间地址 |
|||
namespace: 1bf94455-a046-41e3-b7e4-c12fd11c3690 |
|||
# 命名空间分组 |
|||
group: test |
|||
config: |
|||
# 配置中心地址 |
|||
server-addr: http://117.50.61.7:8848 |
|||
# 配置文件格式 |
|||
file-extension: yml |
|||
# 命名空间地址 |
|||
namespace: 1bf94455-a046-41e3-b7e4-c12fd11c3690 |
|||
# 命名空间分组 |
|||
group: test |
|||
# 共享配置 |
|||
shared-configs: |
|||
- application-${spring.profiles.active}.${spring.cloud.nacos.config.file-extension} |
|||
@ -0,0 +1,15 @@ |
|||
# Tomcat |
|||
server: |
|||
port: 9105 |
|||
|
|||
# Spring |
|||
spring: |
|||
main: |
|||
#解决循环依赖问题 |
|||
allow-circular-references: true |
|||
application: |
|||
# 应用名称 |
|||
name: bnyer-ai |
|||
profiles: |
|||
# 环境配置 |
|||
active: dev |
|||
@ -0,0 +1,74 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<configuration scan="true" scanPeriod="60 seconds" debug="false"> |
|||
<!-- 日志存放路径 --> |
|||
<property name="log.path" value="logs/bnyer-ai" /> |
|||
<!-- 日志输出格式 --> |
|||
<property name="log.pattern" value="%d{HH:mm:ss.SSS} [%thread] %-5level %logger{20} - [%method,%line] - %msg%n" /> |
|||
|
|||
<!-- 控制台输出 --> |
|||
<appender name="console" class="ch.qos.logback.core.ConsoleAppender"> |
|||
<encoder> |
|||
<pattern>${log.pattern}</pattern> |
|||
</encoder> |
|||
</appender> |
|||
|
|||
<!-- 系统日志输出 --> |
|||
<appender name="file_info" class="ch.qos.logback.core.rolling.RollingFileAppender"> |
|||
<file>${log.path}/info.log</file> |
|||
<!-- 循环政策：基于时间创建日志文件 --> |
|||
<rollingPolicy class="ch.qos.logback.core.rolling.TimeBasedRollingPolicy"> |
|||
<!-- 日志文件名格式 --> |
|||
<fileNamePattern>${log.path}/info.%d{yyyy-MM-dd}.log</fileNamePattern> |
|||
<!-- 日志最大的历史 60天 --> |
|||
<maxHistory>20</maxHistory> |
|||
</rollingPolicy> |
|||
<encoder> |
|||
<pattern>${log.pattern}</pattern> |
|||
</encoder> |
|||
<filter class="ch.qos.logback.classic.filter.LevelFilter"> |
|||
<!-- 过滤的级别 --> |
|||
<level>INFO</level> |
|||
<!-- 匹配时的操作：接收（记录） --> |
|||
<onMatch>ACCEPT</onMatch> |
|||
<!-- 不匹配时的操作：拒绝（不记录） --> |
|||
<onMismatch>DENY</onMismatch> |
|||
</filter> |
|||
</appender> |
|||
|
|||
<appender name="file_error" class="ch.qos.logback.core.rolling.RollingFileAppender"> |
|||
<file>${log.path}/error.log</file> |
|||
<!-- 循环政策：基于时间创建日志文件 --> |
|||
<rollingPolicy class="ch.qos.logback.core.rolling.TimeBasedRollingPolicy"> |
|||
<!-- 日志文件名格式 --> |
|||
<fileNamePattern>${log.path}/error.%d{yyyy-MM-dd}.log</fileNamePattern> |
|||
<!-- 日志最大的历史 60天 --> |
|||
<maxHistory>20</maxHistory> |
|||
</rollingPolicy> |
|||
<encoder> |
|||
<pattern>${log.pattern}</pattern> |
|||
</encoder> |
|||
<filter class="ch.qos.logback.classic.filter.LevelFilter"> |
|||
<!-- 过滤的级别 --> |
|||
<level>ERROR</level> |
|||
<!-- 匹配时的操作：接收（记录） --> |
|||
<onMatch>ACCEPT</onMatch> |
|||
<!-- 不匹配时的操作：拒绝（不记录） --> |
|||
<onMismatch>DENY</onMismatch> |
|||
</filter> |
|||
</appender> |
|||
|
|||
<!-- 系统模块日志级别控制 --> |
|||
<logger name="com.bnyer" level="info" /> |
|||
<!-- Spring日志级别控制 --> |
|||
<logger name="org.springframework" level="warn" /> |
|||
|
|||
<root level="info"> |
|||
<appender-ref ref="console" /> |
|||
</root> |
|||
|
|||
<!--系统操作日志--> |
|||
<root level="info"> |
|||
<appender-ref ref="file_info" /> |
|||
<appender-ref ref="file_error" /> |
|||
</root> |
|||
</configuration> |
|||
@ -1,7 +1,8 @@ |
|||
package com.bnyer.file.service; |
|||
package com.bnyer.file.service.impl; |
|||
|
|||
import com.bnyer.common.core.exception.ServiceException; |
|||
import com.bnyer.file.config.MinioConfig; |
|||
import com.bnyer.file.service.MinioService; |
|||
import com.bnyer.file.utils.FileUploadUtils; |
|||
import com.bnyer.file.utils.ImgUtil; |
|||
import io.minio.MinioClient; |
|||
@ -0,0 +1,29 @@ |
|||
package com.bnyer.img.mapper; |
|||
|
|||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
|||
import com.bnyer.common.core.domain.AiPaint; |
|||
import com.bnyer.common.core.dto.AiPaintPageDto; |
|||
import com.bnyer.img.vo.AiPaintVo; |
|||
import org.apache.ibatis.annotations.Mapper; |
|||
import org.apache.ibatis.annotations.Param; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Mapper |
|||
public interface AiPaintMapper extends BaseMapper<AiPaint> { |
|||
|
|||
/** |
|||
* 获取ai绘画分页 |
|||
* @param source 来源 |
|||
* @param painterId 绘画者id |
|||
* @return - |
|||
*/ |
|||
List<AiPaintVo> queryPage(@Param("source") String source,@Param("painterId") Long painterId); |
|||
|
|||
/** |
|||
* 获取ai绘画详情 |
|||
* @param id 主键id |
|||
* @return - |
|||
*/ |
|||
AiPaintVo queryDetails(Long id); |
|||
} |
|||
@ -0,0 +1,18 @@ |
|||
package com.bnyer.img.mapper; |
|||
|
|||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
|||
import com.bnyer.common.core.domain.PaintStyle; |
|||
import com.bnyer.img.vo.PaintStyleVo; |
|||
import org.apache.ibatis.annotations.Mapper; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Mapper |
|||
public interface PaintStyleMapper extends BaseMapper<PaintStyle> { |
|||
|
|||
/** |
|||
* 获取绘画风格列表 |
|||
* @return - |
|||
*/ |
|||
List<PaintStyleVo> queryList(); |
|||
} |
|||
@ -0,0 +1,18 @@ |
|||
package com.bnyer.img.mapper; |
|||
|
|||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
|||
import com.bnyer.common.core.domain.Prompt; |
|||
import com.bnyer.img.vo.PromptVo; |
|||
import org.apache.ibatis.annotations.Mapper; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Mapper |
|||
public interface PromptMapper extends BaseMapper<Prompt> { |
|||
/** |
|||
* 获取提示词列表 |
|||
* @param type 类型 |
|||
* @return - |
|||
*/ |
|||
List<PromptVo> queryList(String type); |
|||
} |
|||
@ -0,0 +1,40 @@ |
|||
package com.bnyer.img.service; |
|||
|
|||
import com.bnyer.common.core.domain.AiPaint; |
|||
import com.bnyer.common.core.dto.AiPaintPageDto; |
|||
import com.bnyer.img.vo.AiPaintVo; |
|||
|
|||
import java.util.List; |
|||
|
|||
public interface AiPaintService { |
|||
|
|||
/** |
|||
* 新增ai绘画 |
|||
* @param aiPaint ai绘画 |
|||
* @return - |
|||
*/ |
|||
int insert(AiPaint aiPaint); |
|||
|
|||
/** |
|||
* 批量删除ai绘画 |
|||
* @param ids id数组 |
|||
* @return - |
|||
*/ |
|||
int delete(List<Long> ids); |
|||
|
|||
/** |
|||
* 获取绘画者ai绘画分页 |
|||
* @param params 分页参数 |
|||
* @return - |
|||
*/ |
|||
List<AiPaintVo> queryPage(AiPaintPageDto params); |
|||
|
|||
/** |
|||
* 获取ai绘画详情 |
|||
* @param id 主键id |
|||
* @return - |
|||
*/ |
|||
AiPaintVo queryDetails(Long id); |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,25 @@ |
|||
package com.bnyer.img.service; |
|||
|
|||
import java.util.List; |
|||
|
|||
public interface HotSearchKeywordService { |
|||
|
|||
/** |
|||
* 获取最近一个月搜索次数最多的前10条记录 |
|||
* @return - |
|||
*/ |
|||
List<String> getHotKeywordList(); |
|||
|
|||
/** |
|||
* 点击增长热搜词热度次数 |
|||
* @param keyword 热搜词 |
|||
* @return - |
|||
*/ |
|||
void incrementHotKeywordScore(String keyword); |
|||
|
|||
/** |
|||
* 新增热搜词 |
|||
* @param keyword 热搜词 |
|||
*/ |
|||
void insertHotKeyword(String keyword); |
|||
} |
|||
@ -0,0 +1,14 @@ |
|||
package com.bnyer.img.service; |
|||
|
|||
import com.bnyer.img.vo.PaintStyleVo; |
|||
|
|||
import java.util.List; |
|||
|
|||
public interface PaintStyleService { |
|||
|
|||
/** |
|||
* 获取绘画风格列表 |
|||
* @return - |
|||
*/ |
|||
List<PaintStyleVo> queryPaintStyleList(); |
|||
} |
|||
@ -0,0 +1,14 @@ |
|||
package com.bnyer.img.service; |
|||
|
|||
import com.bnyer.img.vo.PromptVo; |
|||
|
|||
import java.util.List; |
|||
|
|||
public interface PromptService { |
|||
|
|||
/** |
|||
* 随机获取8条提示词列表 |
|||
* @return - |
|||
*/ |
|||
List<PromptVo> queryPromptList(); |
|||
} |
|||
@ -0,0 +1,44 @@ |
|||
package com.bnyer.img.service.impl; |
|||
|
|||
import cn.hutool.core.util.IdUtil; |
|||
import com.bnyer.common.core.domain.AiPaint; |
|||
import com.bnyer.common.core.dto.AiPaintPageDto; |
|||
import com.bnyer.common.core.utils.uuid.IdUtils; |
|||
import com.bnyer.img.mapper.AiPaintMapper; |
|||
import com.bnyer.img.service.AiPaintService; |
|||
import com.bnyer.img.vo.AiPaintVo; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.stereotype.Service; |
|||
import org.springframework.transaction.annotation.Transactional; |
|||
|
|||
import java.util.Date; |
|||
import java.util.List; |
|||
|
|||
@Service |
|||
public class AiPaintServiceImpl implements AiPaintService { |
|||
|
|||
@Autowired |
|||
private AiPaintMapper aiPaintMapper; |
|||
|
|||
@Override |
|||
@Transactional(rollbackFor = Exception.class) |
|||
public int insert(AiPaint aiPaint) { |
|||
return aiPaintMapper.insert(aiPaint); |
|||
} |
|||
|
|||
@Override |
|||
@Transactional(rollbackFor = Exception.class) |
|||
public int delete(List<Long> ids) { |
|||
return aiPaintMapper.deleteBatchIds(ids); |
|||
} |
|||
|
|||
@Override |
|||
public List<AiPaintVo> queryPage(AiPaintPageDto params) { |
|||
return aiPaintMapper.queryPage(params.getSource(),params.getPainterId()); |
|||
} |
|||
|
|||
@Override |
|||
public AiPaintVo queryDetails(Long id) { |
|||
return aiPaintMapper.queryDetails(id); |
|||
} |
|||
} |
|||
@ -0,0 +1,65 @@ |
|||
package com.bnyer.img.service.impl; |
|||
|
|||
import com.bnyer.common.core.constant.RedisKeyConstant; |
|||
import com.bnyer.common.redis.service.RedisService; |
|||
import com.bnyer.img.service.HotSearchKeywordService; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.data.redis.core.HashOperations; |
|||
import org.springframework.data.redis.core.ValueOperations; |
|||
import org.springframework.data.redis.core.ZSetOperations; |
|||
import org.springframework.stereotype.Service; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
import java.util.Set; |
|||
|
|||
@Service |
|||
@Slf4j |
|||
public class HotSearchKeywordServiceImpl implements HotSearchKeywordService { |
|||
|
|||
@Autowired |
|||
private RedisService redisService; |
|||
|
|||
|
|||
|
|||
@Override |
|||
public List<String> getHotKeywordList() { |
|||
Long now = System.currentTimeMillis(); |
|||
List<String> result = new ArrayList<>(); |
|||
ZSetOperations zSetOperations = redisService.redisTemplate.opsForZSet(); |
|||
HashOperations hashOperations = redisService.redisTemplate.opsForHash(); |
|||
Set<String> value = zSetOperations.reverseRangeByScore(RedisKeyConstant.HOT_KEY_WORD_KEY, 0, Double.MAX_VALUE); |
|||
//key不为空的时候 推荐相关的最热前十名
|
|||
for (String val : value) { |
|||
if (result.size() > 9) {//只返回最热的前十名
|
|||
break; |
|||
} |
|||
Long time = Long.valueOf((String) hashOperations.get(RedisKeyConstant.HOT_KEY_WORD_TIME_KEY, val)); |
|||
if ((now - time) < 2592000000L) {//返回最近一个月的数据
|
|||
result.add(val); |
|||
} else {//时间超过一个月没搜索就把这个词热度归0
|
|||
zSetOperations.add(RedisKeyConstant.HOT_KEY_WORD_KEY, val, 0); |
|||
} |
|||
} |
|||
return result; |
|||
} |
|||
|
|||
@Override |
|||
public void incrementHotKeywordScore(String keyword) { |
|||
Long now = System.currentTimeMillis(); |
|||
ZSetOperations zSetOperations = redisService.redisTemplate.opsForZSet(); |
|||
HashOperations hashOperations = redisService.redisTemplate.opsForHash(); |
|||
zSetOperations.incrementScore(RedisKeyConstant.HOT_KEY_WORD_KEY, keyword, 1); |
|||
hashOperations.put(RedisKeyConstant.HOT_KEY_WORD_TIME_KEY,keyword,String.valueOf(now)); |
|||
} |
|||
|
|||
@Override |
|||
public void insertHotKeyword(String keyword) { |
|||
Long now = System.currentTimeMillis(); |
|||
ZSetOperations zSetOperations = redisService.redisTemplate.opsForZSet(); |
|||
HashOperations hashOperations = redisService.redisTemplate.opsForHash(); |
|||
zSetOperations.incrementScore(RedisKeyConstant.HOT_KEY_WORD_KEY, keyword, 1); |
|||
hashOperations.put(RedisKeyConstant.HOT_KEY_WORD_TIME_KEY,keyword,String.valueOf(now)); |
|||
} |
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
package com.bnyer.img.service.impl; |
|||
|
|||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|||
import com.bnyer.common.core.domain.PaintStyle; |
|||
import com.bnyer.img.mapper.PaintStyleMapper; |
|||
import com.bnyer.img.service.PaintStyleService; |
|||
import com.bnyer.img.vo.PaintStyleVo; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.stereotype.Service; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Service |
|||
public class PaintStyleServiceImpl implements PaintStyleService { |
|||
|
|||
@Autowired |
|||
private PaintStyleMapper paintStyleMapper; |
|||
|
|||
@Override |
|||
public List<PaintStyleVo> queryPaintStyleList() { |
|||
return paintStyleMapper.queryList(); |
|||
} |
|||
} |
|||
@ -0,0 +1,40 @@ |
|||
package com.bnyer.img.service.impl; |
|||
|
|||
import com.bnyer.img.mapper.PromptMapper; |
|||
import com.bnyer.img.service.PromptService; |
|||
import com.bnyer.img.vo.PromptVo; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.stereotype.Service; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
@Service |
|||
public class PromptServiceimpl implements PromptService { |
|||
|
|||
@Autowired |
|||
private PromptMapper promptMapper; |
|||
|
|||
@Override |
|||
public List<PromptVo> queryPromptList() { |
|||
//type:0->绘画;1->gpt
|
|||
List<PromptVo> promptVos = promptMapper.queryList("0"); |
|||
List<PromptVo> news = new ArrayList<>(); |
|||
Map<Integer,String> map = new HashMap<>(); |
|||
//随机抽10条数据展示在页面上
|
|||
if (promptVos.size() <= 30) { |
|||
return promptVos; |
|||
}else{ |
|||
while (map.size() < 30) { |
|||
int random = (int)(Math.random() * promptVos.size()); |
|||
if (!map.containsKey(random)) { |
|||
map.put(random, ""); |
|||
news.add(promptVos.get(random)); |
|||
} |
|||
} |
|||
} |
|||
return news; |
|||
} |
|||
} |
|||
@ -0,0 +1,50 @@ |
|||
package com.bnyer.img.vo; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonFormat; |
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
import java.io.Serializable; |
|||
import java.util.Date; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("ai绘画响应体") |
|||
public class AiPaintVo implements Serializable { |
|||
|
|||
@ApiModelProperty(value="主键Id") |
|||
private Long id; |
|||
|
|||
@ApiModelProperty(value="作品编号") |
|||
private String paintId; |
|||
|
|||
@ApiModelProperty(value="绘图者昵称") |
|||
private String painterName; |
|||
|
|||
@ApiModelProperty(value="图片") |
|||
private String imgUrl; |
|||
|
|||
@ApiModelProperty(value="关键词") |
|||
private String prompt; |
|||
|
|||
@ApiModelProperty(value="风格名称") |
|||
private String styleName; |
|||
|
|||
@ApiModelProperty(value="图片高度") |
|||
private String height; |
|||
|
|||
@ApiModelProperty(value="图片宽度") |
|||
private String width; |
|||
|
|||
@ApiModelProperty(value="平台(0->Hub;1->抖音;2->快手;3->微信)") |
|||
private String source; |
|||
|
|||
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") |
|||
@ApiModelProperty(value="创建时间") |
|||
private Date createTime; |
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
} |
|||
@ -0,0 +1,31 @@ |
|||
package com.bnyer.img.vo; |
|||
|
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("绘画风格响应体") |
|||
public class PaintStyleVo implements Serializable { |
|||
@ApiModelProperty(value="主键id") |
|||
private Long id; |
|||
|
|||
@ApiModelProperty(value="模型风格名称") |
|||
private String name; |
|||
|
|||
@ApiModelProperty(value="模型名称") |
|||
private String modelName; |
|||
|
|||
@ApiModelProperty(value="模型风格图片") |
|||
private String imgUrl; |
|||
|
|||
@ApiModelProperty(value="是否热门(0->正常;1->热门)") |
|||
private String isHot; |
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
} |
|||
@ -0,0 +1,26 @@ |
|||
package com.bnyer.img.vo; |
|||
|
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
|
|||
@Getter |
|||
@Setter |
|||
@ApiModel("提示词响应体") |
|||
public class PromptVo implements Serializable { |
|||
|
|||
@ApiModelProperty(value="主键id") |
|||
private Long id; |
|||
|
|||
@ApiModelProperty(value="提示词") |
|||
private String text; |
|||
|
|||
@ApiModelProperty(value="类型(0->绘画;1->gpt)") |
|||
private String type; |
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
} |
|||
@ -0,0 +1,40 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> |
|||
<mapper namespace="com.bnyer.img.mapper.AiPaintMapper"> |
|||
<resultMap id="BaseResultMap" type="com.bnyer.common.core.domain.AiPaint"> |
|||
<!--@mbg.generated--> |
|||
<!--@Table img_ai_paint--> |
|||
<id column="id" jdbcType="BIGINT" property="id" /> |
|||
<result column="paint_id" jdbcType="VARCHAR" property="paintId" /> |
|||
<result column="painter_id" jdbcType="BIGINT" property="painterId" /> |
|||
<result column="painter_name" jdbcType="VARCHAR" property="painterName" /> |
|||
<result column="img_url" jdbcType="VARCHAR" property="imgUrl" /> |
|||
<result column="prompt" jdbcType="LONGVARCHAR" property="prompt" /> |
|||
<result column="model" jdbcType="VARCHAR" property="model" /> |
|||
<result column="style_name" jdbcType="VARCHAR" property="styleName" /> |
|||
<result column="height" jdbcType="VARCHAR" property="height" /> |
|||
<result column="width" jdbcType="VARCHAR" property="width" /> |
|||
<result column="is_show" jdbcType="CHAR" property="isShow" /> |
|||
<result column="source" jdbcType="CHAR" property="source" /> |
|||
<result column="create_time" jdbcType="TIMESTAMP" property="createTime" /> |
|||
</resultMap> |
|||
<sql id="Base_Column_List"> |
|||
<!--@mbg.generated--> |
|||
id, paint_id, painter_id, painter_name, img_url, prompt, model,style_name, height, width, is_show, |
|||
`source`, create_time |
|||
</sql> |
|||
|
|||
<select id="queryPage" resultType="com.bnyer.img.vo.AiPaintVo"> |
|||
select |
|||
id, paint_id,painter_name,img_url, prompt, style_name,height, width,create_time,source |
|||
from img_ai_paint |
|||
where is_show = '1' and source = #{source} and painter_id = #{painterId} |
|||
order by create_time desc |
|||
</select> |
|||
|
|||
<select id="queryDetails" resultType="com.bnyer.img.vo.AiPaintVo"> |
|||
select |
|||
id, paint_id,painter_name,img_url, prompt, style_name,height, width,create_time,source |
|||
from img_ai_paint where id = #{id} |
|||
</select> |
|||
</mapper> |
|||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue