test/elasticsearchStudy
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

实现删除和更新数据接口

Browse Source
This commit is contained in:
F嘉阳
2018-05-19 22:44:45 +08:00
parent 92b5d778d2
commit 7f533ff401
4 changed files with 151 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/top/fjy8018/elasticsearch/Controller/AddBookController.java
View File

@@ -32,6 +32,7 @@ public class AddBookController {
/**
* 插入数据
*
* @param title
* @param author
* @param wordCount

48
src/main/java/top/fjy8018/elasticsearch/Controller/DeleteBookController.java Normal file
View File

@@ -0,0 +1,48 @@
package top.fjy8018.elasticsearch.Controller;
import lombok.extern.slf4j.Slf4j;
import org.elasticsearch.action.delete.DeleteResponse;
import org.elasticsearch.client.transport.TransportClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
/**
* @author F嘉阳
* @date 2018-05-19 22:03
*/
@RequestMapping("/delete/book")
@RestController
@Slf4j
public class DeleteBookController {
private static String INDEX = "book";
@Autowired
private TransportClient client;
@DeleteMapping("/novel")
public ResponseEntity delete(@RequestParam(name = "id") String id) {
if (id == null) {
log.error("【删除失败】：传入ID为空");
return new ResponseEntity(HttpStatus.NOT_FOUND);
}
DeleteResponse response = this.client.prepareDelete(INDEX, "novel", id).get();
log.info("【response】" + response.toString());
if (response.getId().equals(id)) {
return new ResponseEntity(HttpStatus.OK);
}
log.error("【删除失败】ID:" + id + response.toString());
return new ResponseEntity(HttpStatus.INTERNAL_SERVER_ERROR);
}
}

1
src/main/java/top/fjy8018/elasticsearch/Controller/GetBookController.java
View File

@@ -25,6 +25,7 @@ public class GetBookController {
/**
* 根据ID查询结果
*
* @param id
* @return
*/

88
src/main/java/top/fjy8018/elasticsearch/Controller/UpdateBookController.java Normal file
View File

@@ -0,0 +1,88 @@
package top.fjy8018.elasticsearch.Controller;
import lombok.extern.slf4j.Slf4j;
import org.elasticsearch.action.update.UpdateRequest;
import org.elasticsearch.action.update.UpdateResponse;
import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.common.xcontent.XContentFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.format.annotation.DateTimeFormat;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.ExecutionException;
/**
* @author F嘉阳
* @date 2018-05-19 22:22
*/
@RequestMapping("/update/book")
@RestController
@Slf4j
public class UpdateBookController {
private static String INDEX = "book";
@Autowired
private TransportClient client;
@PutMapping("/novel")
public ResponseEntity update(@RequestParam(name = "id") String id,
@RequestParam(name = "title", required = false) String title,
@RequestParam(name = "author", required = false) String author,
@RequestParam(name = "word_count", required = false) int wordCount,
@RequestParam(name = "publish_date", required = false)
@DateTimeFormat(pattern = "yyyy-MM-dd") Date publishDate) {
String type = "novel";
// 构造更新请求
UpdateRequest request = new UpdateRequest(INDEX, "novel", id);
UpdateResponse response = new UpdateResponse();
try {
// 构造更新请求体
XContentBuilder content = XContentFactory.jsonBuilder().startObject().field("type", type);
if (title != null) {
content.field("title", title);
}
if (author != null) {
content.field("author", author);
}
if (wordCount != 0) {
content.field("word_count", wordCount);
}
if (publishDate != null) {
// 格式化日期
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
String dateStr = sdf.format(publishDate);
content.field("publish_date", dateStr);
}
content.endObject();
request.doc(content);
response = this.client.update(request).get();
} catch (Exception e) {
e.printStackTrace();
log.error("【title】" + title + "【author】" + author + "【wordCount】" + wordCount + "【publishDate】" + publishDate);
return new ResponseEntity(HttpStatus.INTERNAL_SERVER_ERROR);
}
if (response.getId().equals(id)) {
return new ResponseEntity(HttpStatus.OK);
}
log.error("【response】" + response.toString() + "【title】" + title + "【author】" + author + "【wordCount】" + wordCount + "【publishDate】" + publishDate);
return new ResponseEntity(HttpStatus.INTERNAL_SERVER_ERROR);
}
}