diff --git a/src/main/java/com/ai/config/MilvusConfig.java b/src/main/java/com/ai/config/MilvusConfig.java new file mode 100644 index 0000000..0f694b3 --- /dev/null +++ b/src/main/java/com/ai/config/MilvusConfig.java @@ -0,0 +1,20 @@ +package com.ai.config; + +import io.milvus.client.MilvusClient; +import io.milvus.client.MilvusServiceClient; +import io.milvus.param.ConnectParam; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; + +@Configuration +public class MilvusConfig { + + @Bean + public MilvusClient milvusClient() { + ConnectParam connectParam = ConnectParam.newBuilder() + .withHost("localhost") + .withPort(19530) + .build(); + return new MilvusServiceClient(connectParam); + } +} diff --git a/src/main/java/com/ai/controller/MilvusController.java b/src/main/java/com/ai/controller/MilvusController.java new file mode 100644 index 0000000..99c8324 --- /dev/null +++ b/src/main/java/com/ai/controller/MilvusController.java @@ -0,0 +1,20 @@ +package com.ai.controller; + +import com.ai.service.SimilarService; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +@RestController +@RequestMapping("/milvus") +public class MilvusController { + + @Autowired + private SimilarService similarService; + + @RequestMapping("/create") + public String create() { + similarService.createCollection(); + return "milvus collection create"; + } +} diff --git a/src/main/java/com/ai/model/Product.java b/src/main/java/com/ai/model/Product.java new file mode 100644 index 0000000..3f99543 --- /dev/null +++ b/src/main/java/com/ai/model/Product.java @@ -0,0 +1,15 @@ +package com.ai.model; + +import lombok.Data; + +@Data +public class Product { + + private Long spuId; + + private String spuName; + + private String description; + + private String type; +} diff --git a/src/main/java/com/ai/service/SimilarService.java b/src/main/java/com/ai/service/SimilarService.java new file mode 100644 index 0000000..98e7f3c --- /dev/null +++ b/src/main/java/com/ai/service/SimilarService.java @@ -0,0 +1,38 @@ +package com.ai.service; + +import io.milvus.client.MilvusClient; +import io.milvus.grpc.DataType; +import io.milvus.param.collection.CreateCollectionParam; +import io.milvus.param.collection.FieldType; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; + +@Service +public class SimilarService { + + @Autowired + private MilvusClient milvusClient; + + public void createCollection(){ + //定义字段 + FieldType spuId = FieldType.newBuilder() + .withName("spu_id") + .withDataType(DataType.Int64) + .withPrimaryKey(true) + .build(); + + FieldType vectorField = FieldType.newBuilder() + .withName("type_vector") + .withDataType(DataType.FloatVector) + .withDimension(1536) + .build(); + + CreateCollectionParam createParam = CreateCollectionParam.newBuilder() + .withCollectionName("similar_collection") + .addFieldType(spuId) + .addFieldType(vectorField) + .build(); + + milvusClient.createCollection(createParam); + } +}