一种基于Retrofit 1.x的简单Mock机制

背景

客户端开发过程中难免会遇到服务器接口尚未开发完成、服务器正在部署、测试服务器挂了等等情况导致接口无法访问,影响客户端的调试。如果你们的APP的开发流程是某一个版本的客户端和服务端同步开发的话,这种情况会更频繁。这时候我们就需要一个 Mock service 来为服务器接口请求做一个“假冒”的响应了。

本文介绍我们项目基于 Retrofit 1.9.0 的一直简单的 Mock 实现。Retrofit 2.x 已经出来很长时间了,而且 2.x 有 內建的 Mock 机制,所以对于 2.x,本文就没有直接意义了。俺们为什么没有升级 2.x,因为正如选择一个第三方库是一个需要十分慎重的决定一样,升级一个第三方库也需要慎重,尤其是大版本升级。Retrofit 2.x 相对于1.x 有很大的变化,在俺们的项目目前用 1.9.0 没有出过什么问题的情况下,不想冒险去升级2.x 。所以基于 1.9.0 实现了这很简单的 Mock,只在 Retrofit 1.9.0 测试过,但是相信 Retrofit 1.x 都能使用。

发现突破口

关于 Retrofit 的使用,前面的 这篇文章 已经有介绍。
Retrofit 1.x 需要先 new RestAdapter.Builder(),并且需要调用这个 Builder 实例的 setClient(final Client client) 方法来设置 Client (这里这个 Client 的类型是 retrofit.client.Client)。一般来说,也可以不调用 setClient()RestAdapterensureSaneDefaults() 方法保证了有默认值可以使用,具体可查看源码。查看源码可以发现 retrofit.client.ClientResponse execute(Request request) 方法就是执行 Request 并得到 Response 的地方。那么我们就可以在这里入手做些文章。

实现

用装饰者模式,实现 Client 接口,把本来的 Client 包装一下,来给 execute() 方法增加 mock 的职能。

class MyClient implements Client {

    private Client mClient;
    private MockServer mMockServer = null;

    /**
     * @param endPoint 服务器HOST
     * @param realClient 不用mock时,真正的client
     * @param interfaceClass 声明API方法的interface
     */
    public MyClient(String endPoint, Client realClient, Class interfaceClass) {
        if (realClient == null) {
            throw new IllegalArgumentException("real client must not be null!");
        }

        this.mClient = realClient;
        if (Constants.MOCK_RESP_ENABLED) {
            this.mMockServer = new MockServer(endPoint, interfaceClass);
        }
    }

    @Override
    public Response execute(Request request) throws IOException {
        String url = request.getUrl();
        // 由于执行到execute()方法时,如果是GET,request的url已经带了参数,
        // 所以我们需要处理url,取出一个API本身的path
        String path = null;
        try {
            int indexOfQuery = url.indexOf('?');
            if (indexOfQuery > 0) {
                path = url.substring(0, indexOfQuery);
            } else {
                path = url;
            }
        } catch (Exception ignored) {
        }

        Response mockResponse = null;
        // 配置一个是否启用mock的开关,最好配置在build.gradle里,在上线前关闭
        if (Constants.MOCK_RESP_ENABLED) {
            // 根据path来区分是哪个API
            mockResponse = mMockServer.makeMockResponse(url, path);
        }

        // 如果这个API配置了mock,则使用mock,否则请求真实服务器接口
        if (mockResponse != null) {
            return mockResponse;
        } else {
            return mClient.execute(request);
        }
    }
}

接下来,MockServer 类的实现思路就很简单了, makeMockResponse() 方法针对不同的API(通过 path 参数判断)返回相应的假数据即可。用一个 HashMap<String, String> 来存储path 对应 mock response 的键值对表,对相应的API根据key查询。

但是为了与 Retrofit 本身使用注解的风格相统一,我们也可以使用注解来标注一个接口方法的Mock。如:

interface MyApiService {

    // ...

    @MyMock(MockResponses.USER_INFO)
    @GET("/api/userInfo")
    void getUserInfo(Callback<String> cb);

    // ...
}
interface MockResponses {

    // ...
    
    String USER_INFO = "{\"status\":0,\"message\":\"\",\"data\":{\"nickname\":\"Donald Trump\",\"age\":70}}";
    
    // ...
}

自定义注解 MyMock

package me.tangni.xxx;

import java.lang.annotation.Documented;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;

import static java.lang.annotation.ElementType.METHOD;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

/**
 * @author tangni
 */

@Documented
@Target(METHOD)
@Retention(RUNTIME)
public @interface MyMock {
    String value();
}

MockServer 中,构造方法需要传入声明接口方法的 interface 的类型,本例中就是 MyApiService.class,然后通过反射,遍历其中的所有声明的方法,并取出每个方法的 @GET@POST 注解 (暂时只处理这两个)中的API的path,用作存储 Mock response 的 key。

public class MockServer {

    private HashMap<String, String> mockResps;

    public MockServer(String endPoint, Class interfaceClass) {
        if (Constants.MOCK_RESP_ENABLED) {
            mockResps = generateMockRespMap(endPoint, interfaceClass);
        }
    }

    /**
     *
     * @param reqUrl 请求的url
     * @param path 接口的path
     * @return 如果该接口指定了mock response,则返回mock response,否则返回null。
     */
    public Response makeMockResponse(String reqUrl, String path) {
        if (mockResps != null && mockResps.containsKey(path)) {
            String mockResp = mockResps.get(path);
            return generateMockResponse(reqUrl, mockResp, System.currentTimeMillis());
        } else {
            return null;
        }
    }

    private static HashMap<String, String> generateMockRespMap(String endPoint, Class interfaceClass) {
        HashMap<String, String> map = new HashMap<>();
        Method[] methods = interfaceClass.getDeclaredMethods();
        if (methods != null && methods.length > 0) {
            for (Method method : methods) {
                MyMock mockRespAnno = method.getAnnotation(MyMock.class);
                if (mockRespAnno != null) {
                    String path = null;
                    // TODO: 这里先只处理 GET 和 POST
                    Annotation methodAnno = method.getAnnotation(GET.class);
                    if (methodAnno == null) {
                        methodAnno = method.getAnnotation(POST.class);
                        if (methodAnno != null) {
                            path = ((POST)methodAnno).value();
                        }
                    } else {
                        path = ((GET)methodAnno).value();
                    }

                    if (!TextUtils.isEmpty(path)) {
                        String mockResp = mockRespAnno.value();
                        map.put(endPoint + path, mockResp);
                    }
                }
            }
        }
        return map;
    }

    private static Response generateMockResponse(String url, final String mockResp, long sentMillis) {
        final int contentLen = mockResp != null ? mockResp.length() : 0;

        List<Header> headers = new ArrayList<>();
        headers.add(new Header("Server", "MOCK-SERVER"));
        headers.add(new Header("Content-Type", "application/json;charset=UTF-8"));
        headers.add(new Header("OkHttp-Selected-Protocol", "http/1.1"));
        headers.add(new Header("OkHttp-Sent-Millis", String.valueOf(sentMillis)));
        headers.add(new Header("OkHttp-Received-Millis", String.valueOf(System.currentTimeMillis())));

        TypedInput body = new TypedInput() {
            @Override
            public String mimeType() {
                return "application/json;charset=UTF-8";
            }

            @Override
            public long length() {
                return contentLen;
            }

            @Override
            public InputStream in() throws IOException {
                String content = mockResp;
                if (content == null) {
                    content = "";
                }
                return new ByteArrayInputStream(content.getBytes("UTF-8"));
            }
        };

        return new Response(url, 200, "MOCK-OK", headers, body);
    }
}

最后,使用:

RestAdapter restAdapter = new RestAdapter.Builder()
        .setLogLevel(Constants.DEBUG ? RestAdapter.LogLevel.FULL : RestAdapter.LogLevel.NONE)
        .setRequestInterceptor(/*...*/)
        .setConverter(new MyCustomJsonConverter())
        .setClient(new MyClient(Constants.SERVER_HOST, new MyRealClient(), MyApiService.class))
        .setEndpoint(Constants.SERVER_HOST)
        .build();

mApiService = restAdapter.create(MyApiService.class);
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,117评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,328评论 1 293
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,839评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,007评论 0 206
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,384评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,629评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,880评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,593评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,313评论 1 243
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,575评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,066评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,392评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,052评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,082评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,844评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,662评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,575评论 2 270

推荐阅读更多精彩内容

  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 170,569评论 25 707
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,100评论 18 139
  • 国家电网公司企业标准(Q/GDW)- 面向对象的用电信息数据交换协议 - 报批稿:20170802 前言: 排版 ...
    庭说阅读 10,507评论 6 13
  • —1— 小时候性格咋咋呼呼 在班里有一种“我就是老大”的感觉 朋友挺多 但无法交心 有一段时间我和燕子、曼曼关系很...
    一号公路上阅读 262评论 0 0
  • 很多时候我们都拼命的想要别人别的更好,想要给他更好,却忘记了本质这种东西。本质使然,岂是你我之力可以扭转的
    Qiao雪莲阅读 147评论 0 0