SpringBoot使用protobuf格式的接口方式
<?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 https://maven.apache.org/xsd/maven-4.0.0.xsd'> <modelVersion>4.0.0</modelVersion> <parent><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-parent</artifactId><version>2.2.2.RELEASE</version><relativePath/> <!-- lookup parent from repository --> </parent> <groupId>com.example.protobuf</groupId> <artifactId>demo</artifactId> <version>0.0.1-SNAPSHOT</version> <name>demo</name> <description>Demo project for Spring Boot</description> <properties><java.version>1.8</java.version> </properties> <dependencies><dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter</artifactId></dependency><dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-test</artifactId> <scope>test</scope> <exclusions><exclusion> <groupId>org.junit.vintage</groupId> <artifactId>junit-vintage-engine</artifactId></exclusion> </exclusions></dependency><dependency> <groupId>com.google.protobuf</groupId> <artifactId>protobuf-java</artifactId> <version>3.11.0</version></dependency><dependency> <groupId>com.google.protobuf</groupId> <artifactId>protobuf-java-util</artifactId> <version>3.11.0</version></dependency><dependency> <groupId>com.googlecode.protobuf-java-format</groupId> <artifactId>protobuf-java-format</artifactId> <version>1.2</version></dependency><!-- 網(wǎng)絡(luò)請求依賴 --><dependency> <groupId>org.apache.httpcomponents</groupId> <artifactId>httpclient</artifactId> <version>4.5.2</version></dependency><dependency> <groupId>org.apache.httpcomponents</groupId> <artifactId>httpcore</artifactId> <version>4.4</version></dependency><dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter</artifactId></dependency><dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-web</artifactId></dependency><dependency> <groupId>org.apache.commons</groupId> <artifactId>commons-lang3</artifactId> <version>3.0</version></dependency><dependency> <groupId>commons-collections</groupId> <artifactId>commons-collections</artifactId> <version>3.0</version></dependency> </dependencies> <build><plugins> <plugin><groupId>org.springframework.boot</groupId><artifactId>spring-boot-maven-plugin</artifactId> </plugin></plugins> </build></project>編寫.proto文件,內(nèi)容如下:
user_login.proto:
syntax = 'proto3';option java_package = 'com.boomsecret.protobuf';option java_outer_classname = 'MessageUserLogin';message MessageUserLoginRequest { string username = 1; string password = 2;}message MessageUserLoginResponse { string access_token = 1; string username = 2;}生成java代碼:
protoc.exe --java_out=./ user_login.proto將生成的代碼移動到你的項目中合適位置:
package com.example.protobuf.demo.controller;import com.boomsecret.protobuf.MessageUserLogin;import org.apache.http.HttpResponse;import org.apache.http.client.methods.HttpPost;import org.springframework.stereotype.Controller;import org.springframework.web.bind.annotation.RequestBody;import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.ResponseBody;import util.HttpUtils;import java.net.URI;import java.util.UUID;@Controllerpublic class TestController { @RequestMapping(value = '/demo/test', produces = 'application/x-protobuf') @ResponseBody public MessageUserLogin.MessageUserLoginResponse getPersonProto(@RequestBody MessageUserLogin.MessageUserLoginRequest request) {MessageUserLogin.MessageUserLoginResponse.Builder builder = MessageUserLogin.MessageUserLoginResponse.newBuilder();builder.setAccessToken(UUID.randomUUID().toString()+'_res');builder.setUsername(request.getUsername()+'_res');return builder.build(); }}編寫測試類,通過HttpClient工具調(diào)用接口:
package com.example.protobuf.demo;import com.boomsecret.protobuf.MessageUserLogin;import org.apache.http.HttpResponse;import org.apache.http.client.methods.HttpPost;import org.junit.jupiter.api.Test;import org.springframework.boot.test.context.SpringBootTest;import util.HttpUtils;import java.net.URI;@SpringBootTestclass DemoApplicationTests { @Test void contextLoads() { } @Test public void test() {try { URI uri = new URI('http', null, '127.0.0.1', 8080, '/demo/test', '', null); HttpPost request = new HttpPost(uri); MessageUserLogin.MessageUserLoginRequest.Builder builder = MessageUserLogin.MessageUserLoginRequest.newBuilder(); builder.setUsername('tom'); builder.setPassword('123456'); HttpResponse response = HttpUtils.doPost(request, builder.build()); MessageUserLogin.MessageUserLoginResponse messageUserLoginResponse = MessageUserLogin.MessageUserLoginResponse.parseFrom(response.getEntity().getContent()); System.err.println(messageUserLoginResponse.getAccessToken());} catch (Exception e) {} }}HttpUtils內(nèi)容如下:
package util;import com.google.protobuf.GeneratedMessageV3;import com.googlecode.protobuf.format.JsonFormat;import org.apache.http.Header;import org.apache.http.HttpResponse;import org.apache.http.client.HttpClient;import org.apache.http.client.methods.HttpPost;import org.apache.http.entity.InputStreamEntity;import org.apache.http.impl.client.HttpClients;import java.io.ByteArrayInputStream;import java.io.IOException;/** * @author wangjinliang on 2018/10/18. */public class HttpUtils { public static HttpResponse doPost(HttpPost post, GeneratedMessageV3 message) throws IOException {HttpClient httpclient = HttpClients.createDefault();String requestUrl = post.getURI().toString();ByteArrayInputStream inputStream = new ByteArrayInputStream(message.toByteArray());InputStreamEntity inputStreamEntity = new InputStreamEntity(inputStream);post.setEntity(inputStreamEntity);post.addHeader('Content-Type', 'application/x-protobuf');for (Header header : post.getAllHeaders()) { System.out.println(header.getName() + ':' + header.getValue());}StringBuilder sb = new StringBuilder();sb.append('curl -XPOST ');for (Header header : post.getAllHeaders()) { sb.append(' -H '').append(header.getName()).append(':').append(header.getValue()).append(''');}String jsonBody = JsonFormat.printToString(message);jsonBody = jsonBody.replace(''', ''');sb.append(' -d '').append(jsonBody).append(''');sb.append(' ').append(requestUrl);System.out.println(sb.toString());return httpclient.execute(post); }}
以debug方式運(yùn)行SpringBoot項目,并在controller加斷點,然后運(yùn)行測試代碼:
可以看到請求過來的數(shù)據(jù)是正確的,放行后可以看到響應(yīng)數(shù)據(jù)也是正確的:
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持好吧啦網(wǎng)。
相關(guān)文章:
1. PHP防XSS 防SQL注入的代碼2. idea設(shè)置自動導(dǎo)入依賴的方法步驟3. 淺談SpringMVC jsp前臺獲取參數(shù)的方式 EL表達(dá)式4. python pymysql鏈接數(shù)據(jù)庫查詢結(jié)果轉(zhuǎn)為Dataframe實例5. ASP刪除img標(biāo)簽的style屬性只保留src的正則函數(shù)6. IDEA版最新MyBatis程序配置教程詳解7. 使用Python和百度語音識別生成視頻字幕的實現(xiàn)8. 教你如何寫出可維護(hù)的JS代碼9. idea不能自動補(bǔ)全yml配置文件的原因分析10. xml中的空格之完全解說
